From 0cf0bed97e6162a8a7a2c782a1116de0dc11274b Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Fri, 21 Jun 2024 12:36:12 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20(ToolCallingAgent.py):=20fix=20s?= =?UTF-8?q?yntax=20error=20in=20type=20hint=20for=20dictionary=20in=20conv?= =?UTF-8?q?ert=5Fchat=5Fhistory=20method?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../base/langflow/components/agents/ToolCallingAgent.py | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/src/backend/base/langflow/components/agents/ToolCallingAgent.py b/src/backend/base/langflow/components/agents/ToolCallingAgent.py index a2c0746c8..5e82fb15c 100644 --- a/src/backend/base/langflow/components/agents/ToolCallingAgent.py +++ b/src/backend/base/langflow/components/agents/ToolCallingAgent.py @@ -1,13 +1,12 @@ -from typing import List, Optional +from typing import List 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, HumanMessage, AIMessage +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.field_typing import LanguageModel, Text, Tool from langflow.schema import Data @@ -92,7 +91,7 @@ class ToolCallingAgentComponent(Component): return Message(text=result_string) - def convert_chat_history(self, chat_history: List[Data]) -> List[Dict[str, str]]: + def convert_chat_history(self, chat_history: List[Data]) -> List[dict[str, str]]: messages = [] for item in chat_history: role = "user" if item.sender == "User" else "assistant"