admin-valentin/openai-agents-python
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

More fetch_normalized_spans

Browse source
This commit is contained in:
Alex Hall 2025-03-21 18:15:52 +02:00
parent 153f703211
commit a00b61f355
1 changed files with 14 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

26
tests/test_tracing.py
View file

@ -288,7 +288,7 @@ def test_disabled_tracing():
def enabled_trace_disabled_span(): def enabled_trace_disabled_span():
with trace(workflow_name="test", trace_id="123"): with trace(workflow_name="test", trace_id="trace_123"):
with agent_span(name="agent_1"): with agent_span(name="agent_1"):
with function_span(name="function_1", disabled=True): with function_span(name="function_1", disabled=True):
with generation_span(): with generation_span():
@ -298,17 +298,19 @@ def enabled_trace_disabled_span():
def test_enabled_trace_disabled_span(): def test_enabled_trace_disabled_span():
enabled_trace_disabled_span() enabled_trace_disabled_span()
spans, traces = fetch_ordered_spans(), fetch_traces() assert fetch_normalized_spans() == snapshot(
assert len(spans) == 1 # Only the agent span is recorded [
assert len(traces) == 1 # The trace is recorded {
"workflow_name": "test",
trace = traces[0] "children": [
standard_trace_checks(trace, name_check="test") {
trace_id = trace.trace_id "type": "agent",
"data": {"name": "agent_1"},
first_span = spans[0] }
standard_span_checks(first_span, trace_id=trace_id, parent_id=None, span_type="agent") ],
assert first_span.span_data.name == "agent_1" }
]
)
def test_start_and_end_called_manual(): def test_start_and_end_called_manual():