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

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

View File

@ -116,6 +116,7 @@ def init_chat_model(
- 'huggingface' -> langchain-huggingface
- 'groq' -> langchain-groq
- 'ollama' -> langchain-ollama
- 'google_anthropic_vertex' -> langchain-google-vertexai
Will attempt to infer model_provider from model if not specified. The
following providers will be inferred based on these model prefixes:
@ -410,6 +411,11 @@ def _init_chat_model_helper(
from langchain_aws import ChatBedrockConverse
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:
supported = ", ".join(_SUPPORTED_PROVIDERS)
raise ValueError(
@ -433,6 +439,7 @@ _SUPPORTED_PROVIDERS = {
"groq",
"bedrock",
"bedrock_converse",
"google_anthropic_vertex",
}