--amend

rohit-kadhe · rohit-kadhe · commit e06c94389187 · 2024-09-05T09:20:59.000-06:00
diff --git a/src/langtrace_python_sdk/instrumentation/anthropic/patch.py b/src/langtrace_python_sdk/instrumentation/anthropic/patch.py
@@ -112,7 +112,7 @@ def set_response_attributes(
                     if typ == "text":
                         content = result.content[0].text
                         set_event_completion(
-                            span, [{type: typ, role: role, content: content}]
+                            span, [{"type": typ, role: role, content: content}]
                         )
 
             if (
diff --git a/src/tests/anthropic/test_anthropic.py b/src/tests/anthropic/test_anthropic.py
@@ -27,6 +27,7 @@ def test_anthropic(anthropic_client, exporter):
         "stream": False,
         "max_tokens": 1024,
     }
+
     anthropic_client.messages.create(**kwargs)
     spans = exporter.get_finished_spans()
     completion_span = spans[-1]

Original file line number	Diff line number	Diff line change
`@@ -112,7 +112,7 @@ def set_response_attributes(`
`112`	`112`	`if typ == "text":`
`113`	`113`	`content = result.content[0].text`
`114`	`114`	`set_event_completion(`
`115`		`- span, [{type: typ, role: role, content: content}]`
	`115`	`+ span, [{"type": typ, role: role, content: content}]`
`116`	`116`	`)`
`117`	`117`
`118`	`118`	`if (`
Original file line number	Diff line number	Diff line change
`@@ -27,6 +27,7 @@ def test_anthropic(anthropic_client, exporter):`
`27`	`27`	`"stream": False,`
`28`	`28`	`"max_tokens": 1024,`
`29`	`29`	`}`
	`30`	`+`
`30`	`31`	`anthropic_client.messages.create(**kwargs)`
`31`	`32`	`spans = exporter.get_finished_spans()`
`32`	`33`	`completion_span = spans[-1]`