Skip to content

Commit ec1c83f

Browse files
add system check to more tests
1 parent 7710b40 commit ec1c83f

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
@@ -962,6 +962,7 @@ def test_stream_messages_with_input_json_delta(
962962

963963
assert span["op"] == OP.GEN_AI_CHAT
964964
assert span["description"] == "chat model"
965+
assert span["data"][SPANDATA.GEN_AI_SYSTEM] == "anthropic"
965966
assert span["data"][SPANDATA.GEN_AI_OPERATION_NAME] == "chat"
966967
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
967968

@@ -1279,6 +1280,7 @@ async def test_stream_message_with_input_json_delta_async(
12791280

12801281
assert span["op"] == OP.GEN_AI_CHAT
12811282
assert span["description"] == "chat model"
1283+
assert span["data"][SPANDATA.GEN_AI_SYSTEM] == "anthropic"
12821284
assert span["data"][SPANDATA.GEN_AI_OPERATION_NAME] == "chat"
12831285
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
12841286

@@ -2078,6 +2080,7 @@ def test_stream_messages_with_system_prompt(
20782080

20792081
assert span["op"] == OP.GEN_AI_CHAT
20802082
assert span["description"] == "chat model"
2083+
assert span["data"][SPANDATA.GEN_AI_SYSTEM] == "anthropic"
20812084
assert span["data"][SPANDATA.GEN_AI_OPERATION_NAME] == "chat"
20822085
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
20832086

@@ -2209,6 +2212,7 @@ async def test_stream_message_with_system_prompt_async(
22092212

22102213
assert span["op"] == OP.GEN_AI_CHAT
22112214
assert span["description"] == "chat model"
2215+
assert span["data"][SPANDATA.GEN_AI_SYSTEM] == "anthropic"
22122216
assert span["data"][SPANDATA.GEN_AI_OPERATION_NAME] == "chat"
22132217
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
22142218

0 commit comments

Comments
 (0)