diff --git a/core/cat/factory/custom_llm.py b/core/cat/factory/custom_llm.py index a29dbc41..a263040f 100644 --- 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 index ab70f1ab..bb088e94 100644 --- 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 index 436b84f1..d7a1f85c 100644 --- 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",