From 47940c8aa5949eff975daa93cc890bfa448779fe Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 24 Jun 2024 09:39:38 -0300 Subject: [PATCH] refactor: Update ToolCallingAgentComponent input_dict type to use Dict[str, str] for values --- src/backend/base/langflow/components/agents/ToolCallingAgent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/backend/base/langflow/components/agents/ToolCallingAgent.py b/src/backend/base/langflow/components/agents/ToolCallingAgent.py index e2b643f1c..017003aaa 100644 --- a/src/backend/base/langflow/components/agents/ToolCallingAgent.py +++ b/src/backend/base/langflow/components/agents/ToolCallingAgent.py @@ -87,7 +87,7 @@ class ToolCallingAgentComponent(Component): verbose=True, handle_parsing_errors=self.handle_parsing_errors, ) - input_dict: dict[str, str | list[BaseMessage]] = {"input": self.input_value} + input_dict: dict[str, str | list[Dict[str, str]]] = {"input": self.input_value} if hasattr(self, "memory") and self.memory: input_dict["chat_history"] = self.convert_chat_history(self.memory) result = await runnable.ainvoke(input_dict)