feat: add clsx dependency, enhance logging in agent logic, and improve agent logs component

application/agents/base.py

@@ -255,7 +255,7 @@ class BaseAgent(ABC):
             model=self.gpt_model, messages=messages, tools=self.tools
         )
         if log_context:
-            data = build_stack_data(self.llm)
+            data = build_stack_data(self.llm, exclude_attributes=["client"])
             log_context.stacks.append({"component": "llm", "data": data})
         return resp
 
@@ -271,6 +271,6 @@ class BaseAgent(ABC):
             self, resp, tools_dict, messages, attachments
         )
         if log_context:
-            data = build_stack_data(self.llm_handler)
+            data = build_stack_data(self.llm_handler, exclude_attributes=["tool_calls"])
             log_context.stacks.append({"component": "llm_handler", "data": data})
         return resp

application/agents/classic_agent.py

@@ -48,15 +48,13 @@ class ClassicAgent(BaseAgent):
         ):
             yield {"answer": resp.message.content}
         else:
-            # completion = self.llm.gen_stream(
-            #     model=self.gpt_model, messages=messages, tools=self.tools
-            # )
-            # log type of resp
-            logger.info(f"Response type: {type(resp)}")
-            logger.info(f"Response: {resp}")
             for line in resp:
                 if isinstance(line, str):
                     yield {"answer": line}
 
+        log_context.stacks.append(
+            {"component": "agent", "data": {"tool_calls": self.tool_calls.copy()}}
+        )
+
         yield {"sources": retrieved_data}
         yield {"tool_calls": self.tool_calls.copy()}

application/agents/react_agent.py

@@ -82,6 +82,10 @@ class ReActAgent(BaseAgent):
                 if isinstance(line, str):
                     self.observations.append(line)
 
+        log_context.stacks.append(
+            {"component": "agent", "data": {"tool_calls": self.tool_calls.copy()}}
+        )
+
         yield {"sources": retrieved_data}
         yield {"tool_calls": self.tool_calls.copy()}
 

application/logging.py

@@ -29,6 +29,8 @@ def build_stack_data(
     exclude_attributes: List[str] = None,
     custom_data: Dict = None,
 ) -> Dict:
+    if obj is None:
+        raise ValueError("The 'obj' parameter cannot be None")
     data = {}
     if include_attributes is None:
         include_attributes = []
@@ -56,8 +58,8 @@ def build_stack_data(
                         data[attr_name] = [str(item) for item in attr_value]
                 elif isinstance(attr_value, dict):
                     data[attr_name] = {k: str(v) for k, v in attr_value.items()}
-                else:
-                    data[attr_name] = str(attr_value)
+        except AttributeError as e:
+            logging.warning(f"AttributeError while accessing {attr_name}: {e}")
         except AttributeError:
             pass
     if custom_data:

application/worker.py

@@ -179,6 +179,7 @@ def run_agent_logic(agent_config, input_data):
             "tool_calls": tool_calls,
             "thought": thought,
         }
+        logging.info(f"Agent response: {result}")
         return result
     except Exception as e:
         logging.error(f"Error in run_agent_logic: {e}", exc_info=True)