github/langchain
1
0
Fork 0
mirror of https://github.com/hwchase17/langchain.git synced 2025-07-07 05:30:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

langchain: add support for Google Anthropic Vertex AI model garden provider in init_chat_model (#28177)

Browse Source 
Simple modification to add support for anthropic models deployed in
Google Vertex AI model garden in `init_chat_model` importing
`ChatAnthropicVertex`

- [v] **Lint and test**
This commit is contained in:
Tommaso De Lorenzo 2024-12-19 18:06:21 +01:00 committed by GitHub
parent ff7b01af88
commit 24bfa062bf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -116,6 +116,7 @@ def init_chat_model(
- 'huggingface' -> langchain-huggingface - 'huggingface' -> langchain-huggingface
- 'groq' -> langchain-groq - 'groq' -> langchain-groq
- 'ollama' -> langchain-ollama - 'ollama' -> langchain-ollama
- 'google_anthropic_vertex' -> langchain-google-vertexai
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:
@ -410,6 +411,11 @@ def _init_chat_model_helper(
from langchain_aws import ChatBedrockConverse from langchain_aws import ChatBedrockConverse
return ChatBedrockConverse(model=model, **kwargs) return ChatBedrockConverse(model=model, **kwargs)
elif model_provider == "google_anthropic_vertex":
_check_pkg("langchain_google_vertexai")
from langchain_google_vertexai.model_garden import ChatAnthropicVertex
return ChatAnthropicVertex(model=model, **kwargs)
else: else:
supported = ", ".join(_SUPPORTED_PROVIDERS) supported = ", ".join(_SUPPORTED_PROVIDERS)
raise ValueError( raise ValueError(
@ -433,6 +439,7 @@ _SUPPORTED_PROVIDERS = {
"groq", "groq",
"bedrock", "bedrock",
"bedrock_converse", "bedrock_converse",
"google_anthropic_vertex",
} }