Merge pull request #276 from omertuc/logging2

Add debug logging when creating a new agent

Author: tisnik

src/app/endpoints/query.py

@@ -85,8 +85,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
@@ -100,6 +103,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

@@ -63,8 +63,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(
@@ -77,6 +80,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