content-find-agent-new/content_agent/interfaces.py

from __future__ import annotations

from pathlib import Path
from typing import Any, Protocol


class RuntimeFileStore(Protocol):
    def prepare_run(self, trace_id: str) -> Path: ...
    def run_dir(self, trace_id: str) -> Path: ...
    def write_json(self, trace_id: str, filename: str, data: dict[str, Any]) -> Path: ...
    def append_jsonl(self, trace_id: str, filename: str, rows: list[dict[str, Any]]) -> Path: ...
    def read_json(self, trace_id: str, filename: str) -> dict[str, Any]: ...
    def read_jsonl(self, trace_id: str, filename: str) -> list[dict[str, Any]]: ...
    def file_status(self, trace_id: str) -> dict[str, bool]: ...


class PlatformSearchClient(Protocol):
    def search(self, query: dict[str, Any]) -> list[dict[str, Any]]: ...


class PolicyBundleStore(Protocol):
    def load_policy_bundle(self, policy_bundle_version: str) -> dict[str, Any]: ...