Update bedrock.py - support of other endpoint url (esp. for users of … (

langchain-ai#7592) Added an _endpoint_url_ attribute to Bedrock(LLM) class - I have access to Bedrock only via us-west-2 endpoint and needed to change the endpoint url, this could be useful to other users
AlmyAI · Jul 12, 2023 · 93a84f6 · 93a84f6
1 parent 22525ba
commit 93a84f6
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/langchain/llms/bedrock.py b/langchain/llms/bedrock.py
@@ -95,6 +95,9 @@ class Bedrock(LLM):
     model_kwargs: Optional[Dict] = None
     """Key word arguments to pass to the model."""
 
+    endpoint_url: Optional[str] = None
+    """Needed if you don't want to default to us-east-1 endpoint"""
+
     class Config:
         """Configuration for this pydantic object."""
 
@@ -120,6 +123,8 @@ def validate_environment(cls, values: Dict) -> Dict:
             client_params = {}
             if values["region_name"]:
                 client_params["region_name"] = values["region_name"]
+            if values["endpoint_url"]:
+                client_params["endpoint_url"] = values["endpoint_url"]
 
             values["client"] = session.client("bedrock", **client_params)