fix xml agent tool input (langchain-ai#181)

ohyoxo · Feb 10, 2024 · 5adef76 · 5adef76
1 parent f76f547
commit 5adef76
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/backend/app/agent_types/xml_agent.py b/backend/app/agent_types/xml_agent.py
@@ -98,7 +98,7 @@ async def call_tool(messages):
         # We construct an ToolInvocation from the function_call
         tool, tool_input = last_message.content.split("</tool>")
         _tool = tool.split("<tool>")[1]
-        if "<tool_input>" not in _tool:
+        if "<tool_input>" not in tool_input:
             _tool_input = ""
         else:
             _tool_input = tool_input.split("<tool_input>")[1]