Fix warnings for langchain imports

Farama-Foundation · Oct 25, 2023 · 7245638 · 7245638
1 parent 55fcf07
commit 7245638
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git 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
@@ -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
@@ -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