From b20c2640dac79551685b8aba095ebc6125df928c Mon Sep 17 00:00:00 2001 From: enfeng Date: Thu, 21 Mar 2024 12:04:55 +0800 Subject: [PATCH] anthropic[patch]: update base_url of anthropic (#18634) A small change ~ - [ ] **update base_url**: "package: langchain_anthropic" --------- Co-authored-by: yangenfeng Co-authored-by: Harrison Chase --- .../anthropic/langchain_anthropic/chat_models.py | 11 ++++++++--- 1 file changed, 8 insertions(+), 3 deletions(-) diff --git a/libs/partners/anthropic/langchain_anthropic/chat_models.py b/libs/partners/anthropic/langchain_anthropic/chat_models.py index 330bcb4d6e3..084c6c6ec8d 100644 --- a/libs/partners/anthropic/langchain_anthropic/chat_models.py +++ b/libs/partners/anthropic/langchain_anthropic/chat_models.py @@ -201,11 +201,16 @@ class ChatAnthropic(BaseChatModel): values.get("anthropic_api_key") or os.environ.get("ANTHROPIC_API_KEY") or "" ) values["anthropic_api_key"] = anthropic_api_key - values["_client"] = anthropic.Client( - api_key=anthropic_api_key.get_secret_value() + api_key = anthropic_api_key.get_secret_value() + api_url = ( + values.get("anthropic_api_url") + or os.environ.get("ANTHROPIC_API_URL") + or "https://api.anthropic.com" ) + values["anthropic_api_url"] = api_url + values["_client"] = anthropic.Client(api_key=api_key, base_url=api_url) values["_async_client"] = anthropic.AsyncClient( - api_key=anthropic_api_key.get_secret_value() + api_key=api_key, base_url=api_url ) return values