From 33ef3b0a7e1a9bdb90019726e71d1c61c5dae338 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 9 Aug 2023 21:58:25 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=90=9B=20fix(OpenAIConversationalAgent.py?= =?UTF-8?q?):=20ignore=20type=20errors=20for=20memory=5Fkey,=20system=5Fme?= =?UTF-8?q?ssage,=20prompt,=20agent,=20and=20tools=20variables=20=E2=9C=A8?= =?UTF-8?q?=20feat(OpenAIConversationalAgent.py):=20add=20support=20for=20?= =?UTF-8?q?return=5Fintermediate=5Fsteps=20parameter=20in=20AgentExecutor?= =?UTF-8?q?=20constructor=20to=20enable=20returning=20intermediate=20steps?= =?UTF-8?q?=20during=20conversation?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../components/agents/OpenAIConversationalAgent.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/src/backend/langflow/components/agents/OpenAIConversationalAgent.py b/src/backend/langflow/components/agents/OpenAIConversationalAgent.py index d3164e9a3..06c0d6779 100644 --- a/src/backend/langflow/components/agents/OpenAIConversationalAgent.py +++ b/src/backend/langflow/components/agents/OpenAIConversationalAgent.py @@ -54,17 +54,19 @@ class ConversationalAgent(CustomComponent): max_token_limit=max_token_limit, ) else: - memory_key = memory.memory_key + memory_key = memory.memory_key # type: ignore _system_message = system_message or _get_default_system_message() prompt = OpenAIFunctionsAgent.create_prompt( - system_message=_system_message, + system_message=_system_message, # type: ignore extra_prompt_messages=[MessagesPlaceholder(variable_name=memory_key)], ) - agent = OpenAIFunctionsAgent(llm=llm, tools=tools, prompt=prompt) + agent = OpenAIFunctionsAgent( + llm=llm, tools=tools, prompt=prompt # type: ignore + ) return AgentExecutor( agent=agent, - tools=tools, + tools=tools, # type: ignore memory=memory, verbose=True, return_intermediate_steps=True,