Add debug logging when creating a new agent

Signed-off-by: Eran Cohen <[email protected]>

Author: eranco74

src/app/endpoints/query.py

@@ -86,8 +86,11 @@ def get_agent(  # pylint: disable=too-many-arguments,too-many-positional-argumen
     if conversation_id is not None:
         agent = _agent_cache.get(conversation_id)
         if agent:
-            logger.debug("Reusing existing agent with key: %s", conversation_id)
+            logger.debug(
+                "Reusing existing agent with conversation_id: %s", conversation_id
+            )
             return agent, conversation_id
+        logger.debug("No existing agent found for conversation_id: %s", conversation_id)
 
     logger.debug("Creating new agent")
     # TODO(lucasagomes): move to ReActAgent
@@ -101,6 +104,7 @@ def get_agent(  # pylint: disable=too-many-arguments,too-many-positional-argumen
         enable_session_persistence=True,
     )
     conversation_id = agent.create_session(get_suid())
+    logger.debug("Created new agent and conversation_id: %s", conversation_id)
     _agent_cache[conversation_id] = agent
     conversation_id_to_agent_id[conversation_id] = agent.agent_id
 

src/app/endpoints/streaming_query.py

@@ -64,8 +64,11 @@ async def get_agent(
     if conversation_id is not None:
         agent = _agent_cache.get(conversation_id)
         if agent:
-            logger.debug("Reusing existing agent with key: %s", conversation_id)
+            logger.debug(
+                "Reusing existing agent with conversation_id: %s", conversation_id
+            )
             return agent, conversation_id
+        logger.debug("No existing agent found for conversation_id: %s", conversation_id)
 
     logger.debug("Creating new agent")
     agent = AsyncAgent(
@@ -78,6 +81,7 @@ async def get_agent(
         enable_session_persistence=True,
     )
     conversation_id = await agent.create_session(get_suid())
+    logger.debug("Created new agent and conversation_id: %s", conversation_id)
     _agent_cache[conversation_id] = agent
     conversation_id_to_agent_id[conversation_id] = agent.agent_id
     return agent, conversation_id