JSON serializable

MetaGLM · Jul 17, 2024 · 29efaea · 29efaea
1 parent 5ac6485
commit 29efaea
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/langchain_glm/callbacks/agent_callback_handler.py b/langchain_glm/callbacks/agent_callback_handler.py
@@ -131,7 +131,7 @@ async def on_tool_start(
 
     async def on_tool_end(
         self,
-        output: str,
+        output: Any,
         *,
         run_id: UUID,
         parent_run_id: Optional[UUID] = None,
@@ -143,7 +143,7 @@ async def on_tool_end(
             "run_id": str(run_id),
             "status": AgentStatus.tool_end,
             "tool": kwargs["name"],
-            "tool_output": output,
+            "tool_output": str(output),
         }
         self.queue.put_nowait(dumps(data))