From 400f8fcaf05f046bec81d916832ffdedb417531b Mon Sep 17 00:00:00 2001
From: "Joel Z. Leibo" <jzl@google.com>
Date: Tue, 17 Sep 2024 01:41:57 -0700
Subject: [PATCH] Replace deprecated reference to langchain.llms with
 langchain_community.llms

PiperOrigin-RevId: 675468248
Change-Id: I7fc7ad2a559733c715659768f098789ff78828ce
---
 concordia/language_model/langchain_ollama_model.py | 2 +-
 setup.py                                           | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/concordia/language_model/langchain_ollama_model.py b/concordia/language_model/langchain_ollama_model.py
index e1b60599..e08aa830 100644
--- 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
index 1c287afd..0c08874e 100644
--- 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',