content-find-agent-new/content_agent/business_modules/source_seed/__init__.py

from __future__ import annotations

from typing import Any

from content_agent.business_modules.source_seed.source_context import (
    build_pattern_seed_pack,
    load_source_context,
)
from content_agent.interfaces import RuntimeFileStore


def run(
    run_id: str,
    policy_run_id: str,
    source: str | dict[str, Any] | None,
    runtime: RuntimeFileStore,
) -> dict[str, Any]:
    source_context = load_source_context(run_id, source)
    pattern_seed_pack = build_pattern_seed_pack(run_id, policy_run_id, source_context)
    runtime.write_json(run_id, "source_context.json", source_context)
    runtime.write_json(run_id, "pattern_seed_pack.json", pattern_seed_pack)
    return {
        "source_context": source_context,
        "pattern_seed_pack": pattern_seed_pack,
    }