Skip to content

Commit ade9410

Browse files
committed
fix test
1 parent 27c851d commit ade9410

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/integrations/huggingface_hub/test_huggingface_hub.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -68,7 +68,7 @@ def test_nonstreaming_chat_completion(
6868
tx = events[0]
6969
assert tx["type"] == "transaction"
7070
span = tx["spans"][0]
71-
assert span["op"] == "ai.chat_completions.create.huggingface_hub"
71+
assert span["op"] == "gen_ai.generate_text"
7272

7373
if send_default_pii and include_prompts:
7474
assert "hello" in span["data"][SPANDATA.AI_INPUT_MESSAGES]
@@ -127,7 +127,7 @@ def test_streaming_chat_completion(
127127
tx = events[0]
128128
assert tx["type"] == "transaction"
129129
span = tx["spans"][0]
130-
assert span["op"] == "ai.chat_completions.create.huggingface_hub"
130+
assert span["op"] == "gen_ai.generate_text"
131131

132132
if send_default_pii and include_prompts:
133133
assert "hello" in span["data"][SPANDATA.AI_INPUT_MESSAGES]

0 commit comments

Comments
 (0)