File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed
tests/integrations/huggingface_hub Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff 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 ]
You can’t perform that action at this time.
0 commit comments