Merge pull request #1755 from siiddhantt/feat/agent-menu

feat: agent webhook and minor fixes

Author: dartpain

application/agents/base.py

@@ -256,7 +256,7 @@ def _llm_gen(self, messages: List[Dict], log_context: Optional[LogContext] = Non
             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
 
@@ -272,6 +272,6 @@ def _llm_handler(
             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 @@ def _gen_inner(
         ):
             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 @@ def _gen_inner(
                 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()}