Skip to content
Open
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
23 commits
Select commit Hold shift + click to select a range
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
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
11 changes: 9 additions & 2 deletions sdks/python/src/agent_control/agents.py
Original file line number Diff line number Diff line change
Expand Up @@ -16,7 +16,9 @@ async def register_agent(
steps: list[dict[str, Any]] | None = None,
conflict_mode: Literal["strict", "overwrite"] = "overwrite",
) -> dict[str, Any]:
"""Register an agent with the server via /initAgent endpoint."""
"""Register an agent with the server via /initAgent endpoint.

"""
ensure_evaluators_discovered()

agent_dict = agent.to_dict()
Expand All @@ -27,7 +29,12 @@ async def register_agent(
"conflict_mode": conflict_mode,
}

response = await client.http_client.post("/api/v1/agents/initAgent", json=payload)
headers = None
response = await client.http_client.post(
"/api/v1/agents/initAgent",
json=payload,
headers=headers,
)
response.raise_for_status()
return cast(dict[str, Any], response.json())

Expand Down
Loading
Loading