jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Update typing import in

Browse source 
OpenAIConversationalAgent.py
This commit is contained in:
Gabriel Luiz Freitas Almeida 2023-11-23 10:03:12 -03:00
commit 0541078e43
1 changed files with 3 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

6
src/backend/langflow/components/agents/OpenAIConversationalAgent.py
View file

@ -1,4 +1,4 @@
from typing import Optional from typing import List, Optional
from langchain.agents.agent import AgentExecutor from langchain.agents.agent import AgentExecutor
from langchain.agents.agent_toolkits.conversational_retrieval.openai_functions import _get_default_system_message from langchain.agents.agent_toolkits.conversational_retrieval.openai_functions import _get_default_system_message
@ -25,7 +25,7 @@ class ConversationalAgent(CustomComponent):
"gpt-4-32k", "gpt-4-32k",
] ]
return { return {
"tools": {"is_list": True, "display_name": "Tools"}, "tools": {"display_name": "Tools"},
"memory": {"display_name": "Memory"}, "memory": {"display_name": "Memory"},
"system_message": {"display_name": "System Message"}, "system_message": {"display_name": "System Message"},
"max_token_limit": {"display_name": "Max Token Limit"}, "max_token_limit": {"display_name": "Max Token Limit"},
@ -41,7 +41,7 @@ class ConversationalAgent(CustomComponent):
self, self,
model_name: str, model_name: str,
openai_api_key: str, openai_api_key: str,
tools: Tool, tools: List[Tool],
openai_api_base: Optional[str] = None, openai_api_base: Optional[str] = None,
memory: Optional[BaseMemory] = None, memory: Optional[BaseMemory] = None,
system_message: Optional[SystemMessagePromptTemplate] = None, system_message: Optional[SystemMessagePromptTemplate] = None,