add logs

Author: wushixiong

omnibox_wizard/wizard/grimoire/agent/agent.py

@@ -637,6 +637,7 @@ async def astream(
         :param agent_request: The request containing the user's query and tools to be used.
         :return: An async iterable of ChatResponse objects.
         """
+        agent_request.
         with tracer.start_as_current_span("agent.astream") as span:
             span.set_attributes(
                 {
@@ -661,7 +662,12 @@ async def astream(
                             "search", get_merged_description(all_tools)
                         )
                     ]
-
+                span.set_attributes(
+                    {
+                        "all_tools": f"{self.all_tools}",
+                        "custom_tool_call": f"{self.custom_tool_call}"
+                    }
+                )
                 assert all_tools, "all_tools must not be empty"
 
                 if self.custom_tool_call: