Fixes model id for custom model arns (#127)

langchain-ai · Jul 31, 2024 · 46ea587 · 46ea587
1 parent 767fa5a
commit 46ea587
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/libs/aws/langchain_aws/llms/bedrock.py b/libs/aws/langchain_aws/llms/bedrock.py
@@ -600,7 +600,7 @@ def _get_provider(self) -> str:
         return self.model_id.split(".")[0]
 
     def _get_model(self) -> str:
-        return self.model_id.split(".")[1]
+        return self.model_id.split(".", maxsplit=1)[-1]
 
     @property
     def _model_is_anthropic(self) -> bool: