Replace deprecated reference to langchain.llms with langchain_communi…

…ty.llms PiperOrigin-RevId: 675468248 Change-Id: I7fc7ad2a559733c715659768f098789ff78828ce
google-deepmind · Sep 17, 2024 · 400f8fc · 400f8fc
1 parent 57ba3af
commit 400f8fc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/concordia/language_model/langchain_ollama_model.py b/concordia/language_model/langchain_ollama_model.py
@@ -19,7 +19,7 @@
 from concordia.language_model import language_model
 from concordia.utils import measurements as measurements_lib
 from concordia.utils import sampling
-from langchain.llms import ollama
+from langchain_community.llms import ollama
 
 from typing_extensions import override
 

diff --git a/setup.py b/setup.py
@@ -72,7 +72,7 @@ def _remove_excluded(description: str) -> str:
         'google-cloud-aiplatform',
         'google-generativeai',
         'ipython',
-        'langchain',
+        'langchain-community',
         'matplotlib',
         'mistralai',
         'numpy',