From c540f00d81e754205b2aad7260707760a6b6256c Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Wed, 22 May 2024 13:54:37 -0700 Subject: [PATCH] Refactor import statements for OpenAI conversational agent (#1949) * Refactor import statements for OpenAI conversational agent --- .../langflow/components/agents/OpenAIConversationalAgent.py | 2 +- .../base/langflow/components/model_specs/ChatOpenAISpecs.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/base/langflow/components/agents/OpenAIConversationalAgent.py b/src/backend/base/langflow/components/agents/OpenAIConversationalAgent.py index 0edf8ea55..032704d9c 100644 --- a/src/backend/base/langflow/components/agents/OpenAIConversationalAgent.py +++ b/src/backend/base/langflow/components/agents/OpenAIConversationalAgent.py @@ -8,7 +8,7 @@ from langchain.prompts import SystemMessagePromptTemplate from langchain.prompts.chat import MessagesPlaceholder from langchain.schema.memory import BaseMemory from langchain.tools import Tool -from langchain_community.chat_models import ChatOpenAI +from langchain_openai import ChatOpenAI from langflow.field_typing.range_spec import RangeSpec from langflow.interface.custom.custom_component import CustomComponent diff --git a/src/backend/base/langflow/components/model_specs/ChatOpenAISpecs.py b/src/backend/base/langflow/components/model_specs/ChatOpenAISpecs.py index 5a7d47f04..d6560580b 100644 --- a/src/backend/base/langflow/components/model_specs/ChatOpenAISpecs.py +++ b/src/backend/base/langflow/components/model_specs/ChatOpenAISpecs.py @@ -1,6 +1,6 @@ from typing import Optional -from langchain_community.chat_models.openai import ChatOpenAI +from langchain_openai import ChatOpenAI from langflow.base.models.openai_constants import MODEL_NAMES from langflow.field_typing import BaseLanguageModel, NestedDict