Skip to content

Commit 7ffca66

Browse files
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: karthik@scale3labs.com <karthik@scale3labs.com>
1 parent e66173c commit 7ffca66

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

  • src/langtrace_python_sdk/instrumentation/aws_bedrock

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"),

0 commit comments

Comments
 (0)