prediction_engine/strategy_codegen/__init__.py

"""Strategy code generator — agent-readable conditional execution plans.

Re-exports from submodules:
  execution_plan: ExecutionPlan, ExecutionPlanBuilder, PlanExecutionTracker
  strategy_generator: StrategyGenerator, OutputFormat, PeriodicScheduler
  combo_plans: ConditionalPlan, SwingDetector, ComboStrategyBuilder
"""

from prediction_engine.strategy_codegen.execution_plan import (
    ActionType,
    CombineLogic,
    ConditionalAction,
    ConditionGroup,
    ExecutionPlan,
    ExecutionPlanBuilder,
    PlanExecutionLog,
    PlanExecutionTracker,
    TriggerCondition,
    TriggerType,
    generate_arb_plan,
    generate_event_plan,
    generate_momentum_plan,
)
from prediction_engine.strategy_codegen.strategy_generator import (
    CausalEdge,
    GeneratedStrategy,
    MarketAnalysis,
    OutputFormat,
    PeriodicScheduler,
    ScheduledCheck,
    StrategyGenerator,
    StrategySpec,
    create_strategy_generator,
)
from prediction_engine.strategy_codegen.combo_plans import (
    ComboActionType,
    ComboLeg,
    ComboScheduledCheck,
    ComboStrategyBuilder,
    ConditionalPlan,
    ConditionNode,
    ConditionOp,
    SwingDetector,
)
from prediction_engine.strategy_codegen.guaranteed_finder import (
    AGGRESSIVE_CONSTRAINTS,
    CONSERVATIVE_CONSTRAINTS,
    SCALPER_CONSTRAINTS,
    Constraint,
    ConstraintDomain,
    ConstraintKind,
    ConstraintSolver,
    MarketSnapshot,
    Opportunity,
    OpportunityRanker,
    OpportunityScanner,
    OpportunityScannerBuilder,
    OpportunityType,
    create_scanner,
)

from prediction_engine.strategy_codegen.strategy_ontology import (
    ArbitrageOpportunity,
    ConstraintOp as OntologyConstraintOp,
    ConstraintSet,
    ConstraintType,
    ExecutionQueue,
    GuaranteedMoneyDetector,
    MonteCarloSimulator,
    PlaybookBuilder,
    PlaybookDAG,
    PlaybookEdge,
    PlaybookNode,
    PlaybookNodeType,
    QLearnerRanker,
    SearchStrategy,
    StrategyOntology,
    create_strategy_ontology,
)

__all__ = [
    # execution_plan
    "ActionType", "CombineLogic", "ConditionalAction", "ConditionGroup",
    "ExecutionPlan", "ExecutionPlanBuilder", "PlanExecutionLog",
    "PlanExecutionTracker", "TriggerCondition", "TriggerType",
    "generate_arb_plan", "generate_event_plan", "generate_momentum_plan",
    # strategy_generator
    "CausalEdge", "GeneratedStrategy", "MarketAnalysis", "OutputFormat",
    "PeriodicScheduler", "ScheduledCheck", "StrategyGenerator",
    "StrategySpec", "create_strategy_generator",
    # combo_plans
    "ComboActionType", "ComboLeg", "ComboScheduledCheck",
    "ComboStrategyBuilder", "ConditionalPlan", "ConditionNode",
    "ConditionOp", "SwingDetector",
    # guaranteed_finder
    "AGGRESSIVE_CONSTRAINTS", "CONSERVATIVE_CONSTRAINTS", "SCALPER_CONSTRAINTS",
    "Constraint", "ConstraintDomain", "ConstraintKind", "ConstraintSolver",
    "MarketSnapshot", "Opportunity", "OpportunityRanker", "OpportunityScanner",
    "OpportunityScannerBuilder", "OpportunityType", "create_scanner",
    # strategy_ontology
    "StrategyOntology", "PlaybookDAG", "PlaybookBuilder", "PlaybookNode",
    "PlaybookEdge", "PlaybookNodeType", "ConstraintSet", "ConstraintType",
    "ConstraintOp", "QLearnerRanker", "GuaranteedMoneyDetector",
    "ArbitrageOpportunity", "MonteCarloSimulator", "ExecutionQueue",
    "SearchStrategy", "create_strategy_ontology",
]