fix: resolve all mypy type annotation errors in Python SDK

Co-authored-by: 0xrinegade <[email protected]>

Author: Copilot

python/mypy.ini

@@ -0,0 +1,16 @@
+[mypy]
+python_version = 3.12
+warn_return_any = True
+warn_unused_configs = True
+disallow_untyped_defs = True
+disallow_incomplete_defs = True
+check_untyped_defs = True
+disallow_untyped_decorators = True
+exclude = tests/
+
+# Ignore missing imports for test dependencies and external packages
+[mypy-pytest.*]
+ignore_missing_imports = True
+
+[mypy-solders.*]
+ignore_missing_imports = True

python/solana_ai_registries/exceptions.py

@@ -5,7 +5,7 @@
 providing clear error messages and proper inheritance hierarchy.
 """
 
-from typing import Any, Dict, Optional
+from typing import Any, Dict, List, Optional
 
 
 class SolanaAIRegistriesError(Exception):
@@ -46,7 +46,7 @@ class TransactionError(SolanaAIRegistriesError):
     """Raised when a Solana transaction fails."""
 
     def __init__(
-        self, message: str, signature: Optional[str] = None, logs: Optional[list] = None
+        self, message: str, signature: Optional[str] = None, logs: Optional[List[str]] = None
     ):
         """Initialize transaction error.
 
@@ -55,7 +55,7 @@ def __init__(
             signature: Transaction signature if available
             logs: Transaction logs if available
         """
-        details = {}
+        details: Dict[str, Any] = {}
         if signature:
             details["signature"] = signature
         if logs:

python/solana_ai_registries/types.py

@@ -73,7 +73,7 @@ class ServiceEndpoint:
     url: str
     description: Optional[str] = None
 
-    def __post_init__(self):
+    def __post_init__(self) -> None:
         """Validate endpoint data after initialization."""
         validate_string_length(self.protocol, 64, "protocol")
         validate_string_length(self.url, 256, "endpoint URL")
@@ -91,7 +91,7 @@ class AgentSkill:
     tags: List[str] = field(default_factory=list)
     metadata: Optional[Dict[str, Any]] = None
 
-    def __post_init__(self):
+    def __post_init__(self) -> None:
         """Validate skill data after initialization."""
         validate_string_length(self.skill_id, 64, "skill ID")
         validate_string_length(self.name, 128, "skill name")
@@ -129,7 +129,7 @@ class AgentRegistryEntry:
     updated_at: int = 0
     state_version: int = 1
 
-    def __post_init__(self):
+    def __post_init__(self) -> None:
         """Validate agent data after initialization."""
         # Validate required fields
         validate_string_length(self.agent_id, MAX_AGENT_ID_LEN, "agent_id")
@@ -213,7 +213,7 @@ class McpTool:
     input_schema: Optional[str] = None
     output_schema: Optional[str] = None
 
-    def __post_init__(self):
+    def __post_init__(self) -> None:
         """Validate tool data after initialization."""
         validate_string_length(self.name, 64, "tool name")
         if len(self.tags) > 3:
@@ -231,7 +231,7 @@ class McpResource:
     description: Optional[str] = None
     tags: List[str] = field(default_factory=list)
 
-    def __post_init__(self):
+    def __post_init__(self) -> None:
         """Validate resource data after initialization."""
         validate_string_length(self.uri_pattern, 128, "URI pattern")
         if len(self.tags) > 3:
@@ -248,7 +248,7 @@ class McpPrompt:
     description: Optional[str] = None
     tags: List[str] = field(default_factory=list)
 
-    def __post_init__(self):
+    def __post_init__(self) -> None:
         """Validate prompt data after initialization."""
         validate_string_length(self.name, 64, "prompt name")
         if len(self.tags) > 3:
@@ -290,7 +290,7 @@ class McpServerRegistryEntry:
     updated_at: int = 0
     state_version: int = 1
 
-    def __post_init__(self):
+    def __post_init__(self) -> None:
         """Validate MCP server data after initialization."""
         validate_string_length(self.server_id, MAX_SERVER_ID_LEN, "server_id")
         validate_string_length(self.name, MAX_SERVER_NAME_LEN, "name")

python/tests/conftest.py

@@ -18,8 +18,8 @@
 from solana_ai_registries.types import AgentRegistryEntry, McpServerRegistryEntry
 
 
-@pytest.fixture(scope="session")
-def event_loop():
+@pytest.fixture(scope="session")  # type: ignore[misc]
+def event_loop() -> Any:
     """Create an instance of the default event loop for the test session."""
     loop = asyncio.get_event_loop_policy().new_event_loop()
     yield loop
@@ -53,10 +53,10 @@ def mock_client() -> Mock:
 @pytest.fixture
 async def devnet_client() -> AsyncGenerator[SolanaAIRegistriesClient, None]:
     """Create a real client connected to devnet for integration tests."""
-    client = SolanaAIRegistriesClient(DEFAULT_DEVNET_RPC)
+    client = SolanaAIRegistriesClient()
     yield client
-    # Cleanup if needed
-    await client.close()
+    # Cleanup if needed - client currently has no close method
+    # await client.close()
 
 
 @pytest.fixture
@@ -194,7 +194,7 @@ def mock_token_balance() -> Dict[str, Any]:
 pytest_plugins = ["pytest_asyncio"]
 
 
-def pytest_configure(config):
+def pytest_configure(config: Any) -> None:
     """Configure pytest with custom markers."""
     config.addinivalue_line(
         "markers", "unit: fast unit tests that don't require network"
@@ -209,22 +209,22 @@ def pytest_configure(config):
 
 
 @pytest.fixture(autouse=True)
-def reset_environment_variables(monkeypatch):
+def reset_environment_variables(monkeypatch: Any) -> None:
     """Reset environment variables for each test."""
     # Clear any environment variables that might affect tests
     monkeypatch.delenv("SOLANA_RPC_URL", raising=False)
     monkeypatch.delenv("A2AMPL_TOKEN_MINT", raising=False)
 
 
 # Helper functions for tests
-def assert_valid_signature(signature: str):
+def assert_valid_signature(signature: str) -> None:
     """Assert that a signature is valid format."""
     assert isinstance(signature, str)
     assert len(signature) >= 87  # Minimum length for base58 signature
     assert len(signature) <= 88  # Maximum length for base58 signature
 
 
-def assert_valid_pubkey(pubkey: str):
+def assert_valid_pubkey(pubkey: str) -> None:
     """Assert that a public key is valid format."""
     assert isinstance(pubkey, str)
     assert len(pubkey) == 44  # Standard length for base58 public key

python/tests/test_imports.py

@@ -3,7 +3,7 @@
 import pytest
 
 
-def test_package_imports():
+def test_package_imports() -> None:
     """Test that all main package components can be imported."""
     # Core imports
     from solana_ai_registries import (
@@ -22,7 +22,7 @@ def test_package_imports():
     assert True
 
 
-def test_placeholder_classes_exist():
+def test_placeholder_classes_exist() -> None:
     """Test that placeholder classes are instantiable."""
     from solana_ai_registries import (
         AgentRegistry,