Skip to content

Commit 9db9d66

Browse files
Debug Agentclaude
andcommitted
style: fix ruff-format and ruff-lint pre-commit issues
Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1 parent 1e8a062 commit 9db9d66

2 files changed

Lines changed: 12 additions & 9 deletions

File tree

openhands-agent-server/openhands/agent_server/sockets.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -132,9 +132,11 @@ async def _accept_authenticated_websocket(
132132
timeout=_FIRST_MESSAGE_AUTH_TIMEOUT_SECONDS,
133133
)
134134
data = json.loads(raw)
135-
except (asyncio.TimeoutError, json.JSONDecodeError, WebSocketDisconnect):
135+
except (TimeoutError, json.JSONDecodeError, WebSocketDisconnect):
136136
logger.warning("WebSocket first-message auth failed: bad or missing payload")
137-
await _safe_close_websocket(websocket, code=4001, reason="Authentication failed")
137+
await _safe_close_websocket(
138+
websocket, code=4001, reason="Authentication failed"
139+
)
138140
return False
139141

140142
if (

tests/agent_server/test_websocket_first_message_auth.py

Lines changed: 8 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,9 @@ async def test_legacy_query_param_valid_key():
4646
ws = _make_mock_websocket()
4747
with patch("openhands.agent_server.sockets.get_default_config") as mock_config:
4848
mock_config.return_value.session_api_keys = ["sk-oh-valid"]
49-
result = await _accept_authenticated_websocket(ws, session_api_key="sk-oh-valid")
49+
result = await _accept_authenticated_websocket(
50+
ws, session_api_key="sk-oh-valid"
51+
)
5052

5153
assert result is True
5254
ws.accept.assert_called_once()
@@ -58,7 +60,9 @@ async def test_legacy_query_param_invalid_key():
5860
ws = _make_mock_websocket()
5961
with patch("openhands.agent_server.sockets.get_default_config") as mock_config:
6062
mock_config.return_value.session_api_keys = ["sk-oh-valid"]
61-
result = await _accept_authenticated_websocket(ws, session_api_key="sk-oh-wrong")
63+
result = await _accept_authenticated_websocket(
64+
ws, session_api_key="sk-oh-wrong"
65+
)
6266

6367
assert result is False
6468
ws.close.assert_called_once_with(code=4001, reason="Authentication failed")
@@ -202,8 +206,7 @@ async def test_events_socket_first_message_auth_e2e():
202206
from openhands.agent_server.sockets import events_socket
203207

204208
ws = _make_mock_websocket()
205-
# First call: auth message (read by _accept_authenticated_websocket via receive_text)
206-
# Then receive_json calls: disconnect
209+
# Auth via receive_text, then receive_json raises disconnect.
207210
ws.receive_text.return_value = json.dumps(
208211
{"type": "auth", "session_api_key": "sk-oh-valid"}
209212
)
@@ -220,9 +223,7 @@ async def test_events_socket_first_message_auth_e2e():
220223
patch("openhands.agent_server.sockets.get_default_config") as mock_config,
221224
):
222225
mock_config.return_value.session_api_keys = ["sk-oh-valid"]
223-
mock_conv_service.get_event_service = AsyncMock(
224-
return_value=mock_event_service
225-
)
226+
mock_conv_service.get_event_service = AsyncMock(return_value=mock_event_service)
226227

227228
await events_socket(uuid4(), ws, session_api_key=None)
228229

0 commit comments

Comments
 (0)