Skip to content

Commit b1963ef

Browse files
Fix AWS Bedrock ARN parsing issue in converse methods (#544)
* Fix AWS Bedrock ARN parsing issue in converse methods - Replace modelId.split('.') with parse_vendor_and_model_name_from_model_id - Fixes crash when using ARN or cross-region model IDs with multiple dots - Makes patch_converse and patch_converse_stream consistent with other methods Resolves ValueError: too many values to unpack (expected 2) when using: - ARN format: arn:aws:bedrock:us-east-1:<account_id>:inference-profile/us.anthropic.claude-3-haiku-20240307-v1:0 - Cross-region format: us.anthropic.claude-sonnet-4-20250514-v1:0 Co-Authored-By: [email protected] <[email protected]> * Bump version to 3.8.21 for ARN parsing bug fix Co-Authored-By: [email protected] <[email protected]> --------- Co-authored-by: devin-ai-integration[bot] <158243242+devin-ai-integration[bot]@users.noreply.github.com> Co-authored-by: [email protected] <[email protected]>
1 parent e66173c commit b1963ef

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

src/langtrace_python_sdk/instrumentation/aws_bedrock/patch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -107,7 +107,7 @@ def traced_method(wrapped, instance, args, kwargs):
107107
def patch_converse_stream(original_method, tracer, version):
108108
def traced_method(*args, **kwargs):
109109
modelId = kwargs.get("modelId")
110-
(vendor, _) = modelId.split(".")
110+
vendor, _ = parse_vendor_and_model_name_from_model_id(modelId)
111111
input_content = [
112112
{
113113
"role": message.get("role", "user"),
@@ -142,7 +142,7 @@ def traced_method(*args, **kwargs):
142142
def patch_converse(original_method, tracer, version):
143143
def traced_method(*args, **kwargs):
144144
modelId = kwargs.get("modelId")
145-
(vendor, _) = modelId.split(".")
145+
vendor, _ = parse_vendor_and_model_name_from_model_id(modelId)
146146
input_content = [
147147
{
148148
"role": message.get("role", "user"),
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
__version__ = "3.8.20"
1+
__version__ = "3.8.21"

0 commit comments

Comments
 (0)