From 343d829c25daf869f05ec1996f1fa05ccf4cc196 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Thu, 10 Aug 2023 18:15:41 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(OpenAIConversationalAgent.py?= =?UTF-8?q?):=20fix=20typo=20in=20display=20name=20of=20Conversational=20A?= =?UTF-8?q?gent=20class=20=E2=9C=A8=20feat(OpenAIConversationalAgent.py):?= =?UTF-8?q?=20add=20support=20for=20OpenAI=20API=20key=20and=20base=20URL?= =?UTF-8?q?=20in=20build=20method=20to=20enable=20API=20function=20calling?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../components/agents/OpenAIConversationalAgent.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/src/backend/langflow/components/agents/OpenAIConversationalAgent.py b/src/backend/langflow/components/agents/OpenAIConversationalAgent.py index 5d089ab3b..e2b876978 100644 --- a/src/backend/langflow/components/agents/OpenAIConversationalAgent.py +++ b/src/backend/langflow/components/agents/OpenAIConversationalAgent.py @@ -15,7 +15,7 @@ from langchain.agents.agent_toolkits.conversational_retrieval.openai_functions i class ConversationalAgent(CustomComponent): - display_name: str = "OpenaAI Conversational Agent" + display_name: str = "OpenAI Conversational Agent" description: str = "Conversational Agent that can use OpenAI's function calling API" def build_config(self): @@ -41,12 +41,18 @@ class ConversationalAgent(CustomComponent): def build( self, model_name: str, + openai_api_key: str, + openai_api_base: str, tools: Tool, memory: Optional[BaseMemory] = None, system_message: Optional[SystemMessagePromptTemplate] = None, max_token_limit: int = 2000, ) -> AgentExecutor: - llm = ChatOpenAI(model=model_name) + llm = ChatOpenAI( + model=model_name, + openai_api_key=openai_api_key, + openai_api_base=openai_api_base, + ) if not memory: memory_key = "chat_history" memory = ConversationTokenBufferMemory(