style: remove unnecessary StrOutputParser

CogitoNTNU · Oct 29, 2024 · a77417f · a77417f
1 parent 1b6befe
commit a77417f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/noder.py b/core/noder.py
@@ -51,7 +51,7 @@ def tool_decider(state: GraphState):
         """,
     )
 
-    chain = prompt | ToolsAgent.agent | StrOutputParser()
+    chain = prompt | ToolsAgent.agent
     response = chain.invoke({"messages": state["messages"], "data": state.get("data", {})})
     return {"messages": [response]}