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