github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-06-20 22:03:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

langchain[patch]: init_chat_model() to import ChatOllama from langchain-ollama and fallback on langchain-community (#24821)

Browse Source 
Description: init_chat_model() should import ChatOllama from
`langchain-ollama`. If that fails, fallback to `langchain-community`
This commit is contained in:
Jerron Lim 2024-07-31 02:16:10 +08:00 committed by GitHub
parent 3a7f3d46c3
commit d8f3ea82db
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
libs/langchain/langchain/chat_models/base.py
View File

@ -118,7 +118,7 @@ def init_chat_model(
- mistralai (langchain-mistralai) - mistralai (langchain-mistralai)
- huggingface (langchain-huggingface) - huggingface (langchain-huggingface)
- groq (langchain-groq) - groq (langchain-groq)
- ollama (langchain-community) - ollama (langchain-ollama)
Will attempt to infer model_provider from model if not specified. The Will attempt to infer model_provider from model if not specified. The
following providers will be inferred based on these model prefixes: following providers will be inferred based on these model prefixes:
@ -336,8 +336,12 @@ def _init_chat_model_helper(
return ChatFireworks(model=model, **kwargs) return ChatFireworks(model=model, **kwargs)
elif model_provider == "ollama": elif model_provider == "ollama":
_check_pkg("langchain_community") try:
from langchain_community.chat_models import ChatOllama _check_pkg("langchain_ollama")
from langchain_ollama import ChatOllama
except ImportError:
_check_pkg("langchain_community")
from langchain_community import ChatOllama
return ChatOllama(model=model, **kwargs) return ChatOllama(model=model, **kwargs)
elif model_provider == "together": elif model_provider == "together":