feat: integrate stock selector into strategy engine scheduler

2 files changed, 72 insertions, 0 deletions

diff --git a/services/strategy-engine/src/strategy_engine/config.py b/services/strategy-engine/src/strategy_engine/config.py
index 9fd9c49..2a9cb43 100644
--- a/services/strategy-engine/src/strategy_engine/config.py
+++ b/services/strategy-engine/src/strategy_engine/config.py
@@ -7,3 +7,9 @@ class StrategyConfig(Settings):
     symbols: list[str] = ["AAPL", "MSFT", "GOOGL", "AMZN", "TSLA"]
     timeframes: list[str] = ["1m"]
     strategy_params: dict = {}
+    selector_candidates_time: str = "15:00"
+    selector_filter_time: str = "15:15"
+    selector_final_time: str = "15:30"
+    selector_max_picks: int = 3
+    anthropic_api_key: str = ""
+    anthropic_model: str = "claude-sonnet-4-20250514"
diff --git a/services/strategy-engine/src/strategy_engine/main.py b/services/strategy-engine/src/strategy_engine/main.py
index 30de528..e9c96b2 100644
--- a/services/strategy-engine/src/strategy_engine/main.py
+++ b/services/strategy-engine/src/strategy_engine/main.py
@@ -1,17 +1,23 @@
 """Strategy Engine Service entry point."""
 
 import asyncio
+from datetime import datetime
 from pathlib import Path
+import zoneinfo
 
+from shared.alpaca import AlpacaClient
 from shared.broker import RedisBroker
+from shared.db import Database
 from shared.healthcheck import HealthCheckServer
 from shared.logging import setup_logging
 from shared.metrics import ServiceMetrics
 from shared.notifier import TelegramNotifier
+from shared.sentiment_models import MarketSentiment
 
 from strategy_engine.config import StrategyConfig
 from strategy_engine.engine import StrategyEngine
 from strategy_engine.plugin_loader import load_strategies
+from strategy_engine.stock_selector import StockSelector
 
 # The strategies directory lives alongside the installed package
 STRATEGIES_DIR = Path(__file__).parent.parent.parent.parent / "strategies"
@@ -30,6 +36,40 @@ async def process_symbol(engine: StrategyEngine, stream: str, log) -> None:
         last_id = await engine.process_once(stream, last_id)
 
 
+async def run_stock_selector(
+    selector: StockSelector,
+    notifier: TelegramNotifier,
+    db: Database,
+    config: StrategyConfig,
+    log,
+) -> None:
+    """Run the stock selector once per day at the configured time."""
+    et = zoneinfo.ZoneInfo("America/New_York")
+
+    while True:
+        now_et = datetime.now(et)
+        target_hour, target_min = map(int, config.selector_final_time.split(":"))
+
+        if now_et.hour == target_hour and now_et.minute == target_min:
+            log.info("stock_selector_running")
+            try:
+                selections = await selector.select()
+                if selections:
+                    ms_data = await db.get_latest_market_sentiment()
+                    ms = None
+                    if ms_data:
+                        ms = MarketSentiment(**ms_data)
+                    await notifier.send_stock_selection(selections, ms)
+                    log.info("stock_selector_complete", picks=[s.symbol for s in selections])
+                else:
+                    log.info("stock_selector_no_picks")
+            except Exception as exc:
+                log.error("stock_selector_error", error=str(exc))
+            await asyncio.sleep(120)  # Sleep past this minute
+        else:
+            await asyncio.sleep(30)
+
+
 async def run() -> None:
     config = StrategyConfig()
     log = setup_logging("strategy-engine", config.log_level, config.log_format)
@@ -41,6 +81,16 @@ async def run() -> None:
     )
 
     broker = RedisBroker(config.redis_url)
+
+    db = Database(config.database_url)
+    await db.connect()
+
+    alpaca = AlpacaClient(
+        api_key=config.alpaca_api_key,
+        api_secret=config.alpaca_api_secret,
+        paper=config.alpaca_paper,
+    )
+
     strategies = load_strategies(STRATEGIES_DIR)
 
     for strategy in strategies:
@@ -67,6 +117,20 @@ async def run() -> None:
             task = asyncio.create_task(process_symbol(engine, stream, log))
             tasks.append(task)
 
+        if config.anthropic_api_key:
+            selector = StockSelector(
+                db=db,
+                broker=broker,
+                alpaca=alpaca,
+                anthropic_api_key=config.anthropic_api_key,
+                anthropic_model=config.anthropic_model,
+                max_picks=config.selector_max_picks,
+            )
+            tasks.append(asyncio.create_task(
+                run_stock_selector(selector, notifier, db, config, log)
+            ))
+            log.info("stock_selector_enabled", time=config.selector_final_time)
+
         await asyncio.gather(*tasks)
     except Exception as exc:
         log.error("fatal_error", error=str(exc))
@@ -78,6 +142,8 @@ async def run() -> None:
         metrics.service_up.labels(service="strategy-engine").set(0)
         await notifier.close()
         await broker.close()
+        await alpaca.close()
+        await db.close()
 
 
 def main() -> None: