Skip to content
2 changes: 0 additions & 2 deletions models/src/agent_control_models/evaluation.py
Original file line number Diff line number Diff line change
Expand Up @@ -127,8 +127,6 @@ class EvaluationResponse(BaseModel):
default=None,
description="List of controls that were evaluated but did not match (if any)",
)


class EvaluationResult(EvaluationResponse):
"""
Client-side result model for evaluation analysis.
Expand Down
10 changes: 10 additions & 0 deletions sdks/python/src/agent_control/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -95,8 +95,12 @@ async def handle_input(user_message: str) -> str:
sync_shutdown_observability,
)
from .telemetry import (
clear_control_event_sink,
clear_trace_context_provider,
emit_control_events,
get_trace_context_from_provider,
has_control_event_sink,
set_control_event_sink,
set_trace_context_provider,
)
from .tracing import (
Expand Down Expand Up @@ -473,6 +477,7 @@ async def handle(message: str):

# Re-init behavior: always stop existing loop before mutating shared agent/session globals.
_stop_policy_refresh_loop()
clear_control_event_sink()

# Configure logging if provided (do this early before any logging happens)
if log_config:
Expand Down Expand Up @@ -643,6 +648,7 @@ def _reset_state() -> None:
state.server_controls = None
state.server_url = None
state.api_key = None
clear_control_event_sink()


async def ashutdown() -> None:
Expand Down Expand Up @@ -1307,6 +1313,10 @@ async def main():
"set_trace_context_provider",
"get_trace_context_from_provider",
"clear_trace_context_provider",
"set_control_event_sink",
"has_control_event_sink",
"emit_control_events",
"clear_control_event_sink",
# Observability
"init_observability",
"add_event",
Expand Down
Loading
Loading