ollama universal chat model

2026-02-10 03:00:59 +00:00 · 2024-07-29 17:38:04 -07:00
1 changed files with 3 additions and 3 deletions
--- a/libs/langchain/langchain/chat_models/base.py
+++ b/libs/langchain/langchain/chat_models/base.py
@@ -118,7 +118,7 @@ def init_chat_model(
                - mistralai (langchain-mistralai)
                - huggingface (langchain-huggingface)
                - groq (langchain-groq)
-                - ollama (langchain-community)
+                - ollama (langchain-ollama)

            Will attempt to infer model_provider from model if not specified. The
            following providers will be inferred based on these model prefixes:
@@ -336,8 +336,8 @@ def _init_chat_model_helper(

        return ChatFireworks(model=model, **kwargs)
    elif model_provider == "ollama":
-        _check_pkg("langchain_community")
-        from langchain_community.chat_models import ChatOllama
+        _check_pkg("langchain_ollama")
+        from langchain_ollama import ChatOllama

        return ChatOllama(model=model, **kwargs)
    elif model_provider == "together":