Skip to content

Commit 1810084

Browse files
committed
bring back assert
1 parent 6f2706d commit 1810084

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

tests/integrations/openai_agents/test_openai_agents.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -450,6 +450,7 @@ def simple_test_tool(message: str) -> str:
450450
assert transaction["transaction"] == "test_agent workflow"
451451
assert transaction["contexts"]["trace"]["origin"] == "auto.ai.openai_agents"
452452

453+
assert agent_span["description"] == "invoke_agent test_agent"
453454
assert agent_span["origin"] == "auto.ai.openai_agents"
454455
assert agent_span["data"]["gen_ai.agent.name"] == "test_agent"
455456
assert agent_span["data"]["gen_ai.operation.name"] == "invoke_agent"
@@ -507,6 +508,7 @@ def simple_test_tool(message: str) -> str:
507508
]
508509
)
509510

511+
assert tool_span["description"] == "execute_tool simple_test_tool"
510512
assert tool_span["data"]["gen_ai.agent.name"] == "test_agent"
511513
assert tool_span["data"]["gen_ai.operation.name"] == "execute_tool"
512514
assert (
@@ -528,6 +530,7 @@ def simple_test_tool(message: str) -> str:
528530
assert tool_span["data"]["gen_ai.tool.output"] == "Tool executed with: hello"
529531
assert tool_span["data"]["gen_ai.tool.type"] == "function"
530532

533+
assert ai_client_span2["description"] == "chat gpt-4"
531534
assert ai_client_span2["data"]["gen_ai.agent.name"] == "test_agent"
532535
assert ai_client_span2["data"]["gen_ai.operation.name"] == "chat"
533536
assert (

0 commit comments

Comments
 (0)