Skip to content

Commit 2de1593

Browse files
committed
Replace deprecated openai imports from langchain and langchain_community
1 parent 3e76cd5 commit 2de1593

File tree

3 files changed

+3
-3
lines changed

3 files changed

+3
-3
lines changed

examples/llm/agents/common.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@
1818
from langchain.agents import initialize_agent
1919
from langchain.agents import load_tools
2020
from langchain.agents.agent import AgentExecutor
21-
from langchain.llms.openai import OpenAI
21+
from langchain_openai import OpenAI
2222

2323
from morpheus.config import Config
2424
from morpheus.pipeline.linear_pipeline import LinearPipeline

tests/morpheus_llm/llm/nodes/test_langchain_agent_node.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,8 +32,8 @@
3232
from langchain.agents import initialize_agent
3333
from langchain.callbacks.manager import AsyncCallbackManagerForToolRun
3434
from langchain.callbacks.manager import CallbackManagerForToolRun
35-
from langchain_community.chat_models.openai import ChatOpenAI
3635
from langchain_core.tools import BaseTool
36+
from langchain_openai import ChatOpenAI
3737
except ImportError:
3838
pass
3939

tests/morpheus_llm/llm/test_agents_simple_pipe.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,8 +42,8 @@
4242
from langchain.agents.tools import Tool
4343
from langchain.schema import Generation
4444
from langchain.schema import LLMResult
45-
from langchain_community.llms import OpenAI # pylint: disable=no-name-in-module
4645
from langchain_community.utilities import serpapi
46+
from langchain_openai import OpenAI # pylint: disable=no-name-in-module
4747
except ImportError:
4848
pass
4949

0 commit comments

Comments
 (0)