Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
18 changes: 18 additions & 0 deletions sdks/python/src/agent_control/__init__.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,9 +54,12 @@ async def process(input: str) -> str:
Agent,
ControlAction,
ControlDefinition,
ControlMatch,
ControlScope,
ControlSelector,
EvaluationRequest,
EvaluationResult,
EvaluatorResult,
EvaluatorSpec,
Step,
StepSchema,
Expand Down Expand Up @@ -102,9 +105,12 @@ async def process(input: str) -> str:
Agent,
ControlAction,
ControlDefinition,
ControlMatch,
ControlScope,
ControlSelector,
EvaluationRequest,
EvaluationResult,
EvaluatorResult,
EvaluatorSpec,
Step,
StepSchema,
Expand All @@ -119,6 +125,15 @@ class ControlDefinition:
class ControlSelector:
pass

class ControlScope:
pass

class ControlMatch:
pass

class EvaluatorResult:
pass

class ControlAction:
pass

Expand Down Expand Up @@ -1092,6 +1107,9 @@ async def main():
"EvaluationResult",
"ControlDefinition",
"ControlSelector",
"ControlScope",
"ControlAction",
"ControlMatch",
"EvaluatorSpec",
"EvaluatorResult",
]
18 changes: 18 additions & 0 deletions sdks/python/tests/test_init_validation.py
Original file line number Diff line number Diff line change
@@ -1,10 +1,28 @@
"""Validation tests for agent_control.init()."""

import pytest
from agent_control_models import ControlMatch as ModelControlMatch
from agent_control_models import ControlScope as ModelControlScope
from agent_control_models import EvaluatorResult as ModelEvaluatorResult

import agent_control


def test_init_rejects_invalid_uuid() -> None:
with pytest.raises(ValueError, match="agent_id must be a valid UUID"):
agent_control.init(agent_name="Invalid UUID Agent", agent_id="not-a-uuid")


def test_init_exports_control_scope() -> None:
assert agent_control.ControlScope is ModelControlScope
assert "ControlScope" in agent_control.__all__


def test_init_exports_control_match() -> None:
assert agent_control.ControlMatch is ModelControlMatch
assert "ControlMatch" in agent_control.__all__


def test_init_exports_evaluator_result() -> None:
assert agent_control.EvaluatorResult is ModelEvaluatorResult
assert "EvaluatorResult" in agent_control.__all__
Loading