add more logs for tool calling

Author: ewilliams-cloudera

llm-service/app/services/query/agents/crewai_querier.py

@@ -140,6 +140,9 @@ def assemble_crew(
         "You know when to use tools and when to answer directly.",
         llm=crewai_llm,
         verbose=True,
+        step_callback=lambda output: step_callback(
+            output, "Tool Result", crew_events_queue
+        ),
     )
 
     # Define tasks for the researcher agents

llm-service/app/services/query/crew_events.py

@@ -42,6 +42,7 @@
 
 from crewai import TaskOutput
 from crewai.agents.parser import AgentFinish
+from crewai.tools.tool_types import ToolResult
 
 from pydantic import BaseModel
 
@@ -72,3 +73,13 @@ def step_callback(output: Any, agent: str, crew_events_queue: Queue[CrewEvent])
                 timestamp=time.time(),
             )
         )
+
+    if isinstance(output, ToolResult):
+        crew_events_queue.put(
+            CrewEvent(
+                type="tool_result",
+                name=agent,
+                data=output.result,
+                timestamp=time.time(),
+            )
+        )