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:13:04 +02:00
parent cb0eb8e254
commit 7581696b38
1 changed files with 49 additions and 44 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

93
tests/test_tracing.py
View file

@ -54,7 +54,7 @@ def simple_tracing():
x = trace("test") x = trace("test")
x.start() x.start()
span_1 = agent_span(name="agent_1", parent=x) span_1 = agent_span(name="agent_1", span_id="span_1", parent=x)
span_1.start() span_1.start()
span_1.finish() span_1.finish()
@ -73,33 +73,36 @@ def simple_tracing():
def test_simple_tracing() -> None: def test_simple_tracing() -> None:
simple_tracing() simple_tracing()
spans, traces = fetch_ordered_spans(), fetch_traces() assert fetch_normalized_spans(keep_span_id=True) == snapshot(
assert len(spans) == 3 [
assert len(traces) == 1 {
"workflow_name": "test",
trace = traces[0] "children": [
standard_trace_checks(trace, name_check="test") {
trace_id = trace.trace_id "type": "agent",
"id": "span_1",
first_span = spans[0] "data": {"name": "agent_1"},
standard_span_checks(first_span, trace_id=trace_id, parent_id=None, span_type="agent") },
assert first_span.span_data.name == "agent_1" {
"type": "custom",
second_span = spans[1] "id": "span_2",
standard_span_checks(second_span, trace_id=trace_id, parent_id=None, span_type="custom") "data": {"name": "custom_1", "data": {}},
assert second_span.span_id == "span_2" "children": [
assert second_span.span_data.name == "custom_1" {
"type": "custom",
third_span = spans[2] "id": "span_3",
standard_span_checks( "data": {"name": "custom_2", "data": {}},
third_span, trace_id=trace_id, parent_id=second_span.span_id, span_type="custom" }
],
},
],
}
]
) )
assert third_span.span_id == "span_3"
assert third_span.span_data.name == "custom_2"
def ctxmanager_spans(): def ctxmanager_spans():
with trace(workflow_name="test", trace_id="123", group_id="456"): with trace(workflow_name="test", trace_id="trace_123", group_id="456"):
with custom_span(name="custom_1", span_id="span_1"): with custom_span(name="custom_1", span_id="span_1"):
with custom_span(name="custom_2", span_id="span_1_inner"): with custom_span(name="custom_2", span_id="span_1_inner"):
pass pass
@ -111,27 +114,29 @@ def ctxmanager_spans():
def test_ctxmanager_spans() -> None: def test_ctxmanager_spans() -> None:
ctxmanager_spans() ctxmanager_spans()
spans, traces = fetch_ordered_spans(), fetch_traces() assert fetch_normalized_spans(keep_span_id=True) == snapshot(
assert len(spans) == 3 [
assert len(traces) == 1 {
"workflow_name": "test",
trace = traces[0] "group_id": "456",
standard_trace_checks(trace, name_check="test") "children": [
trace_id = trace.trace_id {
"type": "custom",
first_span = spans[0] "id": "span_1",
standard_span_checks(first_span, trace_id=trace_id, parent_id=None, span_type="custom") "data": {"name": "custom_1", "data": {}},
assert first_span.span_id == "span_1" "children": [
{
first_inner_span = spans[1] "type": "custom",
standard_span_checks( "id": "span_1_inner",
first_inner_span, trace_id=trace_id, parent_id=first_span.span_id, span_type="custom" "data": {"name": "custom_2", "data": {}},
}
],
},
{"type": "custom", "id": "span_2", "data": {"name": "custom_2", "data": {}}},
],
}
]
) )
assert first_inner_span.span_id == "span_1_inner"
second_span = spans[2]
standard_span_checks(second_span, trace_id=trace_id, parent_id=None, span_type="custom")
assert second_span.span_id == "span_2"
async def run_subtask(span_id: str | None = None) -> None: async def run_subtask(span_id: str | None = None) -> None: