diff --git a/pebble/orchestrator/__init__.py b/pebble/orchestrator/__init__.py new file mode 100644 index 00000000..86aa161b --- /dev/null +++ b/pebble/orchestrator/__init__.py @@ -0,0 +1,30 @@ +"""Pebble research orchestrator. + +This package replaces the former single-file `pebble/orchestrator.py`. All +existing behavior lives unchanged in `_pipeline.py`; this `__init__.py` +re-exports the symbols that callers outside the package import today, so +existing import sites keep working without modification. + +Future work (the L2 swarm) will add sibling modules (`schemas.py`, +`planner.py`, `renderer.py`, ...) inside this package. +""" + +from ._pipeline import ( + ProspectBudgetTracker, + activate_foragers, + quorum_verify_claims, + research_single_prospect, + score_source_richness, + synthesize_profile, + verify_urls, +) + +__all__ = [ + "ProspectBudgetTracker", + "activate_foragers", + "quorum_verify_claims", + "research_single_prospect", + "score_source_richness", + "synthesize_profile", + "verify_urls", +] diff --git a/pebble/orchestrator.py b/pebble/orchestrator/_pipeline.py similarity index 99% rename from pebble/orchestrator.py rename to pebble/orchestrator/_pipeline.py index 3a1ca398..f1508bed 100644 --- a/pebble/orchestrator.py +++ b/pebble/orchestrator/_pipeline.py @@ -24,10 +24,10 @@ from dataclasses import dataclass, field from typing import Callable -from .model_client import ModelClient -from .harness import WorkerHarness, HarnessConfig, AgentOutcome, TaskSpec, harness_config_for_agent -from .storage.db import log_harness_outcome, get_source_reliability, save_profile, save_source_scores, save_session -from .data_sources import ( +from ..model_client import ModelClient +from ..harness import WorkerHarness, HarnessConfig, AgentOutcome, TaskSpec, harness_config_for_agent +from ..storage.db import log_harness_outcome, get_source_reliability, save_profile, save_source_scores, save_session +from ..data_sources import ( fetch_organization, search_organizations, fetch_company, @@ -37,8 +37,8 @@ fetch_full_profile, search_officers, ) -from .data_sources.sec import search_cik -from .claim_templates import ( +from ..data_sources.sec import search_cik +from ..claim_templates import ( claims_from_fec, claims_from_usaspending, claims_from_opencorporates,