Update langchain ollama

cheshire-cat-ai · Nov 8, 2024 · 27aba7e · 27aba7e
1 parent 0dff2b7
commit 27aba7e
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 2 deletions.
diff --git a/core/cat/factory/custom_llm.py b/core/cat/factory/custom_llm.py
@@ -3,7 +3,7 @@
 
 from langchain_core.language_models.llms import LLM
 from langchain_openai.chat_models import ChatOpenAI
-from langchain_community.chat_models.ollama import ChatOllama
+from langchain_ollama import ChatOllama
 
 
 

diff --git a/core/cat/looking_glass/cheshire_cat.py b/core/cat/looking_glass/cheshire_cat.py
@@ -2,7 +2,6 @@
 import base64
 from typing import List, Dict
 import requests
-from requests.exceptions import HTTPError
 from typing_extensions import Protocol
 
 from pydantic import BaseModel

diff --git a/core/pyproject.toml b/core/pyproject.toml
@@ -24,6 +24,7 @@ dependencies = [
     "qdrant_client==1.11.0",
     "langchain==0.2.1",
     "langchain-community==0.2.1",
+    "langchain-ollama==0.1.3",
     "langchain-openai==0.1.7",
     "langchain-anthropic==0.1.23",
     "langchain-google-genai==1.0.8",