Skip to content

Commit 6b089fa

Browse files
committed
fix: ruff lint auto-fixes across hanzo-mcp and tests
1 parent e2b3096 commit 6b089fa

File tree

9 files changed

+47
-53
lines changed

9 files changed

+47
-53
lines changed

pkg/hanzo-mcp/hanzo_mcp/__init__.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -39,9 +39,9 @@ def override(obj): # type: ignore
3939
# Re-export canonical types from hanzoai core.
4040
# hanzo-mcp works standalone but prefers hanzoai when available.
4141
try:
42-
from hanzoai.protocols import PermissionPolicy, PermissionMode, PermissionOutcome
4342
from hanzoai.config import ConfigLoader, RuntimeConfig
44-
from hanzoai.session import Session, compact_session, CompactionConfig
45-
from hanzoai.mcp import normalize_mcp_name, mcp_tool_name, MCPClient
43+
from hanzoai.mcp import MCPClient, mcp_tool_name, normalize_mcp_name
44+
from hanzoai.protocols import PermissionMode, PermissionOutcome, PermissionPolicy
45+
from hanzoai.session import CompactionConfig, Session, compact_session
4646
except ImportError:
4747
pass # hanzoai not installed, MCP server still works standalone

pkg/hanzo-mcp/hanzo_mcp/config/__init__.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,6 @@
1616
from pathlib import Path
1717
from typing import Any, Dict, List, Optional
1818

19-
from pydantic import BaseModel, Field
20-
2119
# --- Re-export canonical config types from hanzoai ---
2220
from hanzoai.config import (
2321
ConfigEntry,
@@ -35,6 +33,7 @@
3533
OAuthConfig,
3634
RuntimeConfig,
3735
)
36+
from pydantic import BaseModel, Field
3837

3938
# --- Re-export tool config ---
4039
from .tool_config import (

pkg/hanzo-mcp/hanzo_mcp/prompts/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,12 +1,12 @@
11
import os
22

3+
from hanzoai.session import CompactionConfig, CompactionResult, Session, compact_session
34
from mcp.server import FastMCP
45

56
from hanzo_mcp.prompts.compact_conversation import (
67
COMPACT_CONVERSATION_PROMPT,
78
compact_or_prompt,
89
)
9-
from hanzoai.session import CompactionConfig, CompactionResult, Session, compact_session
1010
from hanzo_mcp.prompts.create_release import CREATE_RELEASE_PROMPT
1111
from hanzo_mcp.prompts.enhanced_prompts import (
1212
CONFIGURATION_GUIDE_PROMPT,

tests/test_auth_pkce.py

Lines changed: 12 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -3,32 +3,32 @@
33
Run: .venv/bin/python -m pytest tests/test_auth_pkce.py -v
44
"""
55

6-
import base64
7-
import hashlib
8-
import json
96
import os
7+
import json
108
import stat
9+
import base64
10+
import hashlib
1111
import tempfile
1212
import unittest
1313
from pathlib import Path
14-
from unittest.mock import AsyncMock, MagicMock, patch
1514
from urllib.parse import parse_qs, urlparse
15+
from unittest.mock import AsyncMock, MagicMock, patch
1616

1717
from hanzoai.auth import (
18-
ANTHROPIC_AUTHORIZE_URL,
18+
OPENAI_ISSUER,
19+
HANZO_CLIENT_ID,
20+
OPENAI_CLIENT_ID,
1921
ANTHROPIC_CLIENT_ID,
2022
ANTHROPIC_TOKEN_URL,
21-
HANZO_CLIENT_ID,
23+
ANTHROPIC_AUTHORIZE_URL,
2224
HanzoAuth,
23-
OAuthAuthorizationRequest,
25+
PkceCodePair,
26+
OAuthTokenSet,
2427
OAuthCredentialStore,
28+
OAuthAuthorizationRequest,
2529
OAuthTokenExchangeRequest,
26-
OAuthTokenSet,
27-
OPENAI_CLIENT_ID,
28-
OPENAI_ISSUER,
29-
PkceCodePair,
30-
generate_pkce_pair,
3130
generate_state,
31+
generate_pkce_pair,
3232
)
3333

3434

tests/test_config.py

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -2,30 +2,29 @@
22

33
from __future__ import annotations
44

5-
import json
65
import os
6+
import json
77
from pathlib import Path
88

99
import pytest
1010

1111
from hanzoai.config import (
1212
ConfigEntry,
1313
ConfigError,
14+
McpWsConfig,
15+
OAuthConfig,
1416
ConfigLoader,
1517
ConfigSource,
16-
McpClaudeAiProxyConfig,
17-
McpOAuthConfig,
18-
McpRemoteConfig,
1918
McpSdkConfig,
20-
McpStdioConfig,
2119
McpTransport,
22-
McpWsConfig,
23-
OAuthConfig,
2420
RuntimeConfig,
21+
McpOAuthConfig,
22+
McpStdioConfig,
23+
McpRemoteConfig,
24+
McpClaudeAiProxyConfig,
2525
_deep_merge,
2626
)
2727

28-
2928
# ---------------------------------------------------------------------------
3029
# Deep merge
3130
# ---------------------------------------------------------------------------

tests/test_mcp.py

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,24 +1,23 @@
11
"""Tests for MCP client implementation."""
22

3-
import asyncio
4-
import json
53
import sys
4+
import json
5+
import asyncio
66
import textwrap
77

88
import pytest
99

1010
from hanzoai.mcp import (
11+
MCPClient,
1112
JsonRpcError,
13+
MCPHttpClient,
1214
JsonRpcRequest,
13-
JsonRpcResponse,
14-
MCPClient,
1515
MCPClientError,
16-
MCPHttpClient,
16+
JsonRpcResponse,
1717
mcp_tool_name,
1818
normalize_mcp_name,
1919
)
2020

21-
2221
# --- Name normalization tests ---
2322

2423

tests/test_parity.py

Lines changed: 9 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -6,14 +6,12 @@
66
"""
77
from __future__ import annotations
88

9-
import hashlib
109
import json
10+
import hashlib
1111
import tempfile
1212
from pathlib import Path
1313

14-
1514
# ── 1. SSE Parser ────────────────────────────────────────────────────────────
16-
1715
from hanzoai._streaming import SSEDecoder, ServerSentEvent
1816

1917

@@ -117,7 +115,7 @@ def test_sse_comment_only():
117115

118116
# ── 2. MCP Name Normalization ────────────────────────────────────────────────
119117

120-
from hanzoai.mcp import normalize_mcp_name, mcp_tool_name
118+
from hanzoai.mcp import mcp_tool_name, normalize_mcp_name
121119

122120

123121
# Parity: claw-code/rust/crates/runtime/src/mcp.rs:normalizes_server_names_for_mcp_tooling
@@ -147,8 +145,8 @@ def test_mcp_hyphens_kept():
147145

148146
from hanzoai.protocols import (
149147
PermissionMode,
150-
PermissionOutcome,
151148
PermissionPolicy,
149+
PermissionOutcome,
152150
PermissionRequest,
153151
)
154152

@@ -189,12 +187,12 @@ def test_permission_tool_override():
189187
# ── 4. Session Compaction ────────────────────────────────────────────────────
190188

191189
from hanzoai.session import (
192-
CompactionConfig,
193-
ConversationMessage,
194-
MessageRole,
195190
Session,
196191
TextBlock,
192+
MessageRole,
197193
ToolResultBlock,
194+
CompactionConfig,
195+
ConversationMessage,
198196
compact_session,
199197
format_compact_summary,
200198
)
@@ -302,14 +300,14 @@ def test_pkce_s256_rfc_vector():
302300

303301
from hanzoai.config import (
304302
ConfigEntry,
303+
McpWsConfig,
305304
ConfigLoader,
306305
ConfigSource,
306+
McpSdkConfig,
307+
McpTransport,
307308
McpStdioConfig,
308309
McpRemoteConfig,
309-
McpWsConfig,
310-
McpSdkConfig,
311310
McpClaudeAiProxyConfig,
312-
McpTransport,
313311
_parse_mcp_server,
314312
_sanitize_project_config,
315313
)

tests/test_session.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -7,19 +7,19 @@
77
import pytest
88

99
from hanzoai.session import (
10-
CompactionConfig,
11-
CompactionResult,
12-
ConversationMessage,
13-
MessageRole,
1410
Session,
1511
TextBlock,
1612
TokenUsage,
17-
ToolResultBlock,
13+
MessageRole,
1814
ToolUseBlock,
15+
ToolResultBlock,
16+
CompactionConfig,
17+
CompactionResult,
18+
ConversationMessage,
19+
should_compact,
1920
compact_session,
20-
estimate_session_tokens,
2121
format_compact_summary,
22-
should_compact,
22+
estimate_session_tokens,
2323
)
2424

2525

tests/test_sse_chunking.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1,14 +1,13 @@
11
"""Tests for SSE chunking and decoding in hanzoai._streaming."""
22
from __future__ import annotations
33

4+
import sys
45
import asyncio
56
from typing import Iterator, AsyncIterator
7+
from pathlib import Path
68

79
import pytest
810

9-
import sys
10-
from pathlib import Path
11-
1211
sys.path.insert(0, str(Path(__file__).parent.parent / "pkg" / "hanzoai"))
1312

1413
from hanzoai._streaming import SSEDecoder, ServerSentEvent

0 commit comments

Comments
 (0)