Initialize agent in get_agent function to ensure proper setup before processing requests.

jrobertboos · jrobertboos · commit 06626789d7e9 · 2025-08-05T11:24:16.000-04:00
diff --git a/src/app/endpoints/streaming_query.py b/src/app/endpoints/streaming_query.py
@@ -73,6 +73,8 @@ async def get_agent(
         enable_session_persistence=True,
     )
 
+    await agent.initialize()
+
     if existing_agent_id and conversation_id:
         orphan_agent_id = agent.agent_id
         agent._agent_id = conversation_id  # type: ignore[assignment]  # pylint: disable=protected-access

Original file line number	Diff line number	Diff line change
`@@ -73,6 +73,8 @@ async def get_agent(`
`73`	`73`	`enable_session_persistence=True,`
`74`	`74`	`)`
`75`	`75`
	`76`	`+ await agent.initialize()`
	`77`	`+`
`76`	`78`	`if existing_agent_id and conversation_id:`
`77`	`79`	`orphan_agent_id = agent.agent_id`
`78`	`80`	`agent._agent_id = conversation_id # type: ignore[assignment] # pylint: disable=protected-access`