Skip to content

Commit 7710b40

Browse files
test(anthropic): Check system and response ID attributes on spans created by stream()
1 parent 31ee64a commit 7710b40

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/integrations/anthropic/test_anthropic.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -419,6 +419,7 @@ def test_stream_messages(
419419

420420
assert span["op"] == OP.GEN_AI_CHAT
421421
assert span["description"] == "chat model"
422+
assert span["data"][SPANDATA.GEN_AI_SYSTEM] == "anthropic"
422423
assert span["data"][SPANDATA.GEN_AI_OPERATION_NAME] == "chat"
423424
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
424425

@@ -437,6 +438,7 @@ def test_stream_messages(
437438
assert span["data"][SPANDATA.GEN_AI_USAGE_OUTPUT_TOKENS] == 10
438439
assert span["data"][SPANDATA.GEN_AI_USAGE_TOTAL_TOKENS] == 20
439440
assert span["data"][SPANDATA.GEN_AI_RESPONSE_STREAMING] is True
441+
assert span["data"][SPANDATA.GEN_AI_RESPONSE_ID] == "msg_01XFDUDYJgAACzvnptvVoYEL"
440442

441443

442444
@pytest.mark.asyncio
@@ -658,6 +660,7 @@ async def test_stream_message_async(
658660

659661
assert span["op"] == OP.GEN_AI_CHAT
660662
assert span["description"] == "chat model"
663+
assert span["data"][SPANDATA.GEN_AI_SYSTEM] == "anthropic"
661664
assert span["data"][SPANDATA.GEN_AI_OPERATION_NAME] == "chat"
662665
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
663666

@@ -676,6 +679,7 @@ async def test_stream_message_async(
676679
assert span["data"][SPANDATA.GEN_AI_USAGE_OUTPUT_TOKENS] == 10
677680
assert span["data"][SPANDATA.GEN_AI_USAGE_TOTAL_TOKENS] == 20
678681
assert span["data"][SPANDATA.GEN_AI_RESPONSE_STREAMING] is True
682+
assert span["data"][SPANDATA.GEN_AI_RESPONSE_ID] == "msg_01XFDUDYJgAACzvnptvVoYEL"
679683

680684

681685
@pytest.mark.skipif(

0 commit comments

Comments
 (0)