From 724563817818112ccddd80d60cefa5b04df5c1e5 Mon Sep 17 00:00:00 2001 From: Elliot Tower Date: Wed, 25 Oct 2023 17:21:37 -0400 Subject: [PATCH] Fix warnings for langchain imports --- chatarena/environments/umshini/symmetric_content_moderation.py | 2 +- chatarena/environments/umshini/symmetric_deception.py | 2 +- docs/tutorials/umshini/debate_langchain.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/chatarena/environments/umshini/symmetric_content_moderation.py b/chatarena/environments/umshini/symmetric_content_moderation.py index c9047d22..a1b9a7a4 100644 --- a/chatarena/environments/umshini/symmetric_content_moderation.py +++ b/chatarena/environments/umshini/symmetric_content_moderation.py @@ -14,7 +14,7 @@ from chatarena.environments import TimeStep from chatarena.message import Message, MessagePool -from langchain import PromptTemplate +from langchain.prompts import PromptTemplate from chatarena.environments.umshini.base import UmshiniBaseEnv diff --git a/chatarena/environments/umshini/symmetric_deception.py b/chatarena/environments/umshini/symmetric_deception.py index 172f2bcc..05a54a26 100644 --- a/chatarena/environments/umshini/symmetric_deception.py +++ b/chatarena/environments/umshini/symmetric_deception.py @@ -12,7 +12,7 @@ from chatarena.environments import TimeStep from chatarena.message import Message, MessagePool -from langchain import PromptTemplate +from langchain.prompts import PromptTemplate from chatarena.environments.umshini.base import UmshiniBaseEnv diff --git a/docs/tutorials/umshini/debate_langchain.py b/docs/tutorials/umshini/debate_langchain.py index 86eba032..909cad9b 100644 --- a/docs/tutorials/umshini/debate_langchain.py +++ b/docs/tutorials/umshini/debate_langchain.py @@ -1,5 +1,5 @@ """Simple example of running the Umshini debate environment locally using LangChain agents. This can be used to test agents before participating in a tournament.""" -from langchain import OpenAI +from langchain.llms import OpenAI from langchain.agents import AgentType, initialize_agent from langchain.memory import ConversationBufferMemory