Skip to content

Commit 35ae584

Browse files
committed
test(anthropic) fix tests to use correct span op field
1 parent 0439961 commit 35ae584

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

tests/integrations/anthropic/test_anthropic.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -112,7 +112,7 @@ def test_nonstreaming_create_message(
112112
assert len(event["spans"]) == 1
113113
(span,) = event["spans"]
114114

115-
assert span["op"] == OP.ANTHROPIC_MESSAGES_CREATE
115+
assert span["op"] == OP.GEN_AI_CHAT
116116
assert span["description"] == "Anthropic messages create"
117117
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
118118

@@ -184,7 +184,7 @@ async def test_nonstreaming_create_message_async(
184184
assert len(event["spans"]) == 1
185185
(span,) = event["spans"]
186186

187-
assert span["op"] == OP.ANTHROPIC_MESSAGES_CREATE
187+
assert span["op"] == OP.GEN_AI_CHAT
188188
assert span["description"] == "Anthropic messages create"
189189
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
190190

@@ -287,7 +287,7 @@ def test_streaming_create_message(
287287
assert len(event["spans"]) == 1
288288
(span,) = event["spans"]
289289

290-
assert span["op"] == OP.ANTHROPIC_MESSAGES_CREATE
290+
assert span["op"] == OP.GEN_AI_CHAT
291291
assert span["description"] == "Anthropic messages create"
292292
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
293293

@@ -394,7 +394,7 @@ async def test_streaming_create_message_async(
394394
assert len(event["spans"]) == 1
395395
(span,) = event["spans"]
396396

397-
assert span["op"] == OP.ANTHROPIC_MESSAGES_CREATE
397+
assert span["op"] == OP.GEN_AI_CHAT
398398
assert span["description"] == "Anthropic messages create"
399399
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
400400

@@ -528,7 +528,7 @@ def test_streaming_create_message_with_input_json_delta(
528528
assert len(event["spans"]) == 1
529529
(span,) = event["spans"]
530530

531-
assert span["op"] == OP.ANTHROPIC_MESSAGES_CREATE
531+
assert span["op"] == OP.GEN_AI_CHAT
532532
assert span["description"] == "Anthropic messages create"
533533
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
534534

@@ -668,7 +668,7 @@ async def test_streaming_create_message_with_input_json_delta_async(
668668
assert len(event["spans"]) == 1
669669
(span,) = event["spans"]
670670

671-
assert span["op"] == OP.ANTHROPIC_MESSAGES_CREATE
671+
assert span["op"] == OP.GEN_AI_CHAT
672672
assert span["description"] == "Anthropic messages create"
673673
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
674674

0 commit comments

Comments
 (0)