From 55e4fe93c0997db04cdd6fe55ec511cac93d00f4 Mon Sep 17 00:00:00 2001 From: italojohnny Date: Sun, 23 Jun 2024 20:54:08 -0300 Subject: [PATCH] fix code --- .../components/agents/ToolCallingAgent.py | 16 +++------------- 1 file changed, 3 insertions(+), 13 deletions(-) diff --git a/src/backend/base/langflow/components/agents/ToolCallingAgent.py b/src/backend/base/langflow/components/agents/ToolCallingAgent.py index a9403e861..318c5c622 100644 --- a/src/backend/base/langflow/components/agents/ToolCallingAgent.py +++ b/src/backend/base/langflow/components/agents/ToolCallingAgent.py @@ -1,27 +1,17 @@ -from typing import List, Optional - +from typing import List, Dict from langchain.agents.tool_calling_agent.base import create_tool_calling_agent from langchain_core.prompts import ChatPromptTemplate -<<<<<<< Updated upstream - -from langflow.custom import Component -from langflow.io import BoolInput, HandleInput, MessageTextInput, Output -from langflow.schema import Data -======= from langchain.agents import AgentExecutor from langchain_core.messages import BaseMessage ->>>>>>> Stashed changes from langflow.schema.message import Message from langflow.custom import Component -from langflow.io import HandleInput, TextInput, BoolInput, Output +from langflow.io import HandleInput, BoolInput, Output, MessageTextInput from langflow.schema import Data class ToolCallingAgentComponent(Component): display_name: str = "Tool Calling Agent" - description: str = ( - "Agent that uses tools. Only models that are compatible with function calling are supported." - ) + description: str = "Agent that uses tools. Only models that are compatible with function calling are supported." icon = "LangChain" beta = True