mirror of
https://github.com/Xe138/AI-Trader.git
synced 2026-04-09 12:17:24 -04:00
Compare commits
1 Commits
v0.3.0-alp
...
v0.3.0-alp
| Author | SHA1 | Date | |
|---|---|---|---|
| e968434062 |
@@ -221,7 +221,7 @@ class BaseAgent:
|
|||||||
|
|
||||||
print(f"✅ Agent {self.signature} initialization completed")
|
print(f"✅ Agent {self.signature} initialization completed")
|
||||||
|
|
||||||
def set_context(self, context_injector: "ContextInjector") -> None:
|
async def set_context(self, context_injector: "ContextInjector") -> None:
|
||||||
"""
|
"""
|
||||||
Inject ContextInjector after initialization.
|
Inject ContextInjector after initialization.
|
||||||
|
|
||||||
@@ -241,6 +241,9 @@ class BaseAgent:
|
|||||||
tool_interceptors=[context_injector]
|
tool_interceptors=[context_injector]
|
||||||
)
|
)
|
||||||
|
|
||||||
|
# CRITICAL: Reload tools from new client so they use the interceptor
|
||||||
|
self.tools = await self.client.get_tools()
|
||||||
|
|
||||||
print(f"✅ Context injected: signature={context_injector.signature}, "
|
print(f"✅ Context injected: signature={context_injector.signature}, "
|
||||||
f"date={context_injector.today_date}, job_id={context_injector.job_id}, "
|
f"date={context_injector.today_date}, job_id={context_injector.job_id}, "
|
||||||
f"session_id={context_injector.session_id}")
|
f"session_id={context_injector.session_id}")
|
||||||
|
|||||||
@@ -140,7 +140,7 @@ class ModelDayExecutor:
|
|||||||
job_id=self.job_id,
|
job_id=self.job_id,
|
||||||
session_id=session_id
|
session_id=session_id
|
||||||
)
|
)
|
||||||
agent.set_context(context_injector)
|
await agent.set_context(context_injector)
|
||||||
|
|
||||||
# Run trading session
|
# Run trading session
|
||||||
logger.info(f"Running trading session for {self.model_sig} on {self.date}")
|
logger.info(f"Running trading session for {self.model_sig} on {self.date}")
|
||||||
@@ -155,10 +155,13 @@ class ModelDayExecutor:
|
|||||||
# Update session summary
|
# Update session summary
|
||||||
await self._update_session_summary(cursor, session_id, conversation, agent)
|
await self._update_session_summary(cursor, session_id, conversation, agent)
|
||||||
|
|
||||||
# Store positions (pass session_id)
|
# Commit and close connection before _write_results_to_db opens a new one
|
||||||
self._write_results_to_db(agent, session_id)
|
|
||||||
|
|
||||||
conn.commit()
|
conn.commit()
|
||||||
|
conn.close()
|
||||||
|
conn = None # Mark as closed
|
||||||
|
|
||||||
|
# Store positions (pass session_id) - this opens its own connection
|
||||||
|
self._write_results_to_db(agent, session_id)
|
||||||
|
|
||||||
# Update status to completed
|
# Update status to completed
|
||||||
self.job_manager.update_job_detail_status(
|
self.job_manager.update_job_detail_status(
|
||||||
|
|||||||
Reference in New Issue
Block a user