From 24bfa062bf57b205f54806be3bd79e988a2b081d Mon Sep 17 00:00:00 2001 From: Tommaso De Lorenzo Date: Thu, 19 Dec 2024 18:06:21 +0100 Subject: [PATCH] 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** --- libs/langchain/langchain/chat_models/base.py | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/libs/langchain/langchain/chat_models/base.py b/libs/langchain/langchain/chat_models/base.py index 730b7f8908f95..c8a96fa3e5511 100644 --- a/libs/langchain/langchain/chat_models/base.py +++ b/libs/langchain/langchain/chat_models/base.py @@ -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 @@ def _init_chat_model_helper( "groq", "bedrock", "bedrock_converse", + "google_anthropic_vertex", }