feat: add cornerstone

traderai/server.py

@@ -21,6 +21,7 @@ from pydantic import BaseModel
 from traderai.agent import OllamaAgent, OllamaUnavailable
 from traderai.config import save_settings, settings_payload
 from traderai.config import get_settings
+from traderai.cornerstone_client import CornerstoneClient
 from traderai.memory import DEFAULT_THREAD_ID, MemoryStore
 from traderai.scheduler import WakeScheduler
 from traderai.scmdb_client import SCMDBClient
@@ -77,7 +78,15 @@ def create_app() -> FastAPI:
     scheduler = WakeScheduler(memory)
     uex = UEXClient(settings.uex_base_url, settings.uex_secret_key, settings.uex_bearer_token)
     scmdb = SCMDBClient(settings.scmdb_base_url)
-    tools = ToolRegistry(uex, settings.require_write_approval, memory=memory, scheduler=scheduler, scmdb=scmdb)
+    cornerstone = CornerstoneClient(settings.cornerstone_base_url)
+    tools = ToolRegistry(
+        uex,
+        settings.require_write_approval,
+        memory=memory,
+        scheduler=scheduler,
+        scmdb=scmdb,
+        cornerstone=cornerstone,
+    )
     agent = OllamaAgent(
         settings.ollama_base_url,
         settings.ollama_model,