feat: update base_url of anthropic

langchain-ai · Mar 6, 2024 · 261ab38 · 261ab38
1 parent 81985b3
commit 261ab38
Showing 1 changed file with 6 additions and 6 deletions.
diff --git a/libs/partners/anthropic/langchain_anthropic/chat_models.py b/libs/partners/anthropic/langchain_anthropic/chat_models.py
@@ -194,12 +194,12 @@ def validate_environment(cls, values: Dict) -> Dict:
             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()
-        )
-        values["_async_client"] = anthropic.AsyncClient(
-            api_key=anthropic_api_key.get_secret_value()
-        )
+        api_key = anthropic_api_key.get_secret_value()
+        api_url = values.get("anthropic_api_url")
+        if not api_url:
+            api_url = "https://api.anthropic.com"
+        values["_client"] = anthropic.Client(api_key=api_key, base_url=api_url)
+        values["_async_client"] = anthropic.AsyncClient(api_key=api_key, base_url=api_url)
         return values
 
     def _format_params(