Skip to content

Commit 8382fd5

Browse files
committed
Accept Agno OpenAI Chat span names
1 parent 915c2ad commit 8382fd5

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

py/src/braintrust/wrappers/agno/test_agno.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,10 @@ def test_agno_simple_agent_execution(memory_logger):
9494
assert root_span["metrics"]["duration"] > 0
9595

9696
llm_span = spans[1]
97-
assert llm_span["span_attributes"]["name"] == "OpenAI.response"
97+
assert llm_span["span_attributes"]["name"] in {
98+
"OpenAI.response",
99+
"OpenAI Chat.response",
100+
}
98101
assert llm_span["span_attributes"]["type"].value == "llm"
99102
assert llm_span["span_parents"] == [root_span["span_id"]]
100103
assert llm_span["metadata"]["model"] == "gpt-4o-mini"

0 commit comments

Comments
 (0)