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
6 changes: 6 additions & 0 deletions hud/agents/claude/agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -403,6 +403,12 @@ async def get_response(
if thinking_parts:
result.reasoning = "\n".join(thinking_parts)
result.finish_reason = response.stop_reason
if response.stop_reason == "refusal" and response.stop_details is not None:
explanation = response.stop_details.explanation
if explanation:
result.refusal = explanation
elif response.stop_details.category is not None:
result.refusal = f"This request was refused ({response.stop_details.category})."
Comment thread
cursor[bot] marked this conversation as resolved.
return result

@staticmethod
Expand Down
45 changes: 44 additions & 1 deletion hud/agents/tests/test_claude_agent.py
Original file line number Diff line number Diff line change
Expand Up @@ -54,11 +54,12 @@ def __init__(self, *outcomes: Any) -> None:
self.beta = SimpleNamespace(messages=FakeMessages(*outcomes))


def _final(*content: Any, stop_reason: str) -> Any:
def _final(*content: Any, stop_reason: str, stop_details: Any = None) -> Any:
"""A fake ``BetaMessage``: content blocks plus the always-present envelope."""
return SimpleNamespace(
content=list(content),
stop_reason=stop_reason,
stop_details=stop_details,
model="claude-test-v9",
usage=SimpleNamespace(input_tokens=11, output_tokens=7, cache_read_input_tokens=3),
)
Expand Down Expand Up @@ -128,6 +129,48 @@ async def test_get_response_done_on_text_only() -> None:
assert result.done is True
assert result.content == "done"
assert result.tool_calls == []
assert result.refusal is None


async def test_get_response_surfaces_refusal_explanation() -> None:
explanation = (
"This request triggered restrictions on violative cyber content and was "
"blocked under Anthropic's Usage Policy."
)
final = _final(
stop_reason="refusal",
stop_details=SimpleNamespace(
type="refusal",
category="cyber",
explanation=explanation,
),
)
agent = _agent(final)
result = await agent.get_response(_state(agent))

assert result.finish_reason == "refusal"
assert result.refusal == explanation
assert result.done is True
assert result.tool_calls == []


async def test_get_response_surfaces_refusal_category_when_explanation_missing() -> None:
final = _final(
stop_reason="refusal",
stop_details=SimpleNamespace(
type="refusal",
category="cyber",
explanation=None,
),
)
agent = _agent(final)
result = await agent.get_response(_state(agent))

assert result.finish_reason == "refusal"
assert result.refusal is not None
assert "cyber" in result.refusal
assert result.done is True
assert result.tool_calls == []


async def test_get_response_collects_thinking() -> None:
Expand Down
4 changes: 2 additions & 2 deletions pyproject.toml
Original file line number Diff line number Diff line change
Expand Up @@ -21,7 +21,7 @@ dependencies = [
"fastmcp==3.2.0",
# For all inference agents
"openai>=2.45.0",
"anthropic>=0.78.0",
"anthropic>=0.98.0",
"google-genai",
# CLI dependencies
"typer>=0.9.0",
Expand Down Expand Up @@ -122,7 +122,7 @@ agents = []

# AWS Bedrock support for ClaudeAgent
bedrock = [
"anthropic[bedrock]>=0.78.0",
"anthropic[bedrock]>=0.98.0",
]

# Development dependencies - includes testing, linting, and automation tools
Expand Down
Loading