From d1eed09c015e328483ada190e7633f0a76621627 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Sun, 23 Jun 2024 16:00:14 -0300 Subject: [PATCH] Fix linting errors in ToolCallingAgentComponent --- .../langflow/components/agents/ToolCallingAgent.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/src/backend/base/langflow/components/agents/ToolCallingAgent.py b/src/backend/base/langflow/components/agents/ToolCallingAgent.py index a8e4d3dd3..3fceb5631 100644 --- a/src/backend/base/langflow/components/agents/ToolCallingAgent.py +++ b/src/backend/base/langflow/components/agents/ToolCallingAgent.py @@ -1,13 +1,13 @@ -from typing import List +from typing import List, cast +from langchain.agents import AgentExecutor, BaseSingleActionAgent from langchain.agents.tool_calling_agent.base import create_tool_calling_agent from langchain_core.prompts import ChatPromptTemplate -from langchain.agents import AgentExecutor -from langchain_core.messages import BaseMessage -from langflow.schema.message import Message + from langflow.custom import Component -from langflow.io import HandleInput, TextInput, BoolInput, Output +from langflow.io import BoolInput, HandleInput, Output, TextInput from langflow.schema import Data +from langflow.schema.message import Message class ToolCallingAgentComponent(Component): @@ -80,12 +80,12 @@ class ToolCallingAgentComponent(Component): agent = create_tool_calling_agent(self.llm, self.tools, prompt) runnable = AgentExecutor.from_agent_and_tools( - agent=agent, + agent=cast(BaseSingleActionAgent, agent), tools=self.tools, 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)