Skip to content

Commit ef49cde

Browse files
committed
format
1 parent b164e41 commit ef49cde

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

tests/test_manual_graph_instrumentation.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -32,9 +32,7 @@ def end_agent():
3232
return {"status": "completed", "data": "final_data"}
3333

3434
# Run the workflow within a trace context
35-
with langfuse.start_as_current_span(
36-
name="agent_workflow", as_type="agent"
37-
):
35+
with langfuse.start_as_current_span(name="agent_workflow", as_type="agent"):
3836
langfuse.update_current_trace(name=trace_name)
3937

4038
start_result = start_agent()
@@ -119,9 +117,7 @@ def end_agent():
119117
return {"status": "completed", "summary": "all_tools_processed"}
120118

121119
# Execute the parallel workflow
122-
with langfuse.start_as_current_span(
123-
name="parallel_workflow", as_type="span"
124-
):
120+
with langfuse.start_as_current_span(name="parallel_workflow", as_type="span"):
125121
langfuse.update_current_trace(name=trace_name)
126122
start_result = start_agent()
127123

@@ -189,5 +185,3 @@ def end_agent():
189185
assert (
190186
len(tool_observations) == 3
191187
), f"Expected 3 tool observations, got {len(tool_observations)}"
192-
193-

0 commit comments

Comments
 (0)