Skip to content

Commit 52046fd

Browse files
committed
fixed test
1 parent 73514da commit 52046fd

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

‎sdks/python/src/agent_control/control_decorators.py‎

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -351,6 +351,7 @@ async def _evaluate(
351351
"metadata": m.result.metadata,
352352
},
353353
"control_execution_id": m.control_execution_id,
354+
"execution_duration_ms": m.execution_duration_ms,
354355
"steering_context": (
355356
{"message": m.steering_context.message}
356357
if m.steering_context else None
@@ -372,6 +373,7 @@ async def _evaluate(
372373
"metadata": e.result.metadata,
373374
},
374375
"control_execution_id": e.control_execution_id,
376+
"execution_duration_ms": e.execution_duration_ms,
375377
"steering_context": (
376378
{"message": e.steering_context.message}
377379
if e.steering_context else None
@@ -393,6 +395,7 @@ async def _evaluate(
393395
"metadata": nm.result.metadata,
394396
},
395397
"control_execution_id": nm.control_execution_id,
398+
"execution_duration_ms": nm.execution_duration_ms,
396399
"steering_context": (
397400
{"message": nm.steering_context.message}
398401
if nm.steering_context else None
@@ -738,7 +741,7 @@ def _log_single_control(
738741
matched=matched,
739742
action=normalize_action(str(control_data.get("action", "observe"))),
740743
confidence=control_data.get("result", {}).get("confidence", 0.0),
741-
duration_ms=control_data.get("result", {}).get("execution_duration_ms"),
744+
duration_ms=control_data.get("execution_duration_ms"),
742745
control_execution_id=control_data.get("control_execution_id"),
743746
check_stage=check_stage,
744747
)

0 commit comments

Comments
 (0)