Skip to content

Commit

Permalink
langchain: add support for Google Anthropic Vertex AI model garden pr…
Browse files Browse the repository at this point in the history
…ovider 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**
  • Loading branch information
tommasodelorenzo authored Dec 19, 2024
1 parent ff7b01a commit 24bfa06
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions libs/langchain/langchain/chat_models/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand Down Expand Up @@ -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(
Expand All @@ -433,6 +439,7 @@ def _init_chat_model_helper(
"groq",
"bedrock",
"bedrock_converse",
"google_anthropic_vertex",
}


Expand Down

0 comments on commit 24bfa06

Please sign in to comment.