Skip to content

Commit cb0d7c5

Browse files
committed
updated tests
1 parent 86dabfa commit cb0d7c5

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
@@ -113,7 +113,7 @@ def test_nonstreaming_create_message(
113113
(span,) = event["spans"]
114114

115115
assert span["op"] == OP.GEN_AI_CHAT
116-
assert span["description"] == "Anthropic messages create"
116+
assert span["description"] == "chat model"
117117
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
118118

119119
if send_default_pii and include_prompts:
@@ -185,7 +185,7 @@ async def test_nonstreaming_create_message_async(
185185
(span,) = event["spans"]
186186

187187
assert span["op"] == OP.GEN_AI_CHAT
188-
assert span["description"] == "Anthropic messages create"
188+
assert span["description"] == "chat model"
189189
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
190190

191191
if send_default_pii and include_prompts:
@@ -288,7 +288,7 @@ def test_streaming_create_message(
288288
(span,) = event["spans"]
289289

290290
assert span["op"] == OP.GEN_AI_CHAT
291-
assert span["description"] == "Anthropic messages create"
291+
assert span["description"] == "chat model"
292292
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
293293

294294
if send_default_pii and include_prompts:
@@ -395,7 +395,7 @@ async def test_streaming_create_message_async(
395395
(span,) = event["spans"]
396396

397397
assert span["op"] == OP.GEN_AI_CHAT
398-
assert span["description"] == "Anthropic messages create"
398+
assert span["description"] == "chat model"
399399
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
400400

401401
if send_default_pii and include_prompts:
@@ -529,7 +529,7 @@ def test_streaming_create_message_with_input_json_delta(
529529
(span,) = event["spans"]
530530

531531
assert span["op"] == OP.GEN_AI_CHAT
532-
assert span["description"] == "Anthropic messages create"
532+
assert span["description"] == "chat model"
533533
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
534534

535535
if send_default_pii and include_prompts:
@@ -669,7 +669,7 @@ async def test_streaming_create_message_with_input_json_delta_async(
669669
(span,) = event["spans"]
670670

671671
assert span["op"] == OP.GEN_AI_CHAT
672-
assert span["description"] == "Anthropic messages create"
672+
assert span["description"] == "chat model"
673673
assert span["data"][SPANDATA.GEN_AI_REQUEST_MODEL] == "model"
674674

675675
if send_default_pii and include_prompts:

0 commit comments

Comments
 (0)