fix: resolve all flake8 linting issues - line length and unused imports

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

Author: Copilot

python/solana_ai_registries/constants.py

@@ -102,8 +102,12 @@
 # A2AMPL Token
 A2AMPL_DECIMALS: Final[int] = 9
 A2AMPL_BASE_UNIT: Final[int] = 1_000_000_000  # 1 A2AMPL = 10^9 base units
-A2AMPL_TOKEN_MINT_MAINNET: Final[str] = "Cpzvdx6pppc9TNArsGsqgShCsKC9NCCjA2gtzHvUpump"
-A2AMPL_TOKEN_MINT_DEVNET: Final[str] = "A2AMPLyncKHwfSnwRNsJ2qsjsetgo9fGkP8YZPsDZ9mE"
+A2AMPL_TOKEN_MINT_MAINNET: Final[str] = (
+    "Cpzvdx6pppc9TNArsGsqgShCsKC9NCCjA2gtzHvUpump"
+)
+A2AMPL_TOKEN_MINT_DEVNET: Final[str] = (
+    "A2AMPLyncKHwfSnwRNsJ2qsjsetgo9fGkP8YZPsDZ9mE"
+)
 
 # Agent Registry Fees (in base units)
 AGENT_REGISTRATION_FEE: Final[int] = 100_000_000_000  # 100 A2AMPL
@@ -158,7 +162,9 @@
 # ============================================================================
 
 # Registry Programs
-AGENT_REGISTRY_PROGRAM_ID: Final[str] = "AgentReg11111111111111111111111111111111111"
+AGENT_REGISTRY_PROGRAM_ID: Final[str] = (
+    "AgentReg11111111111111111111111111111111111"
+)
 MCP_SERVER_REGISTRY_PROGRAM_ID: Final[str] = "TBD"  # To be updated
 
 # Authorized External Programs
@@ -232,7 +238,9 @@ def get_token_mint_for_cluster(cluster: str) -> str:
         raise ValueError(f"Unsupported cluster: {cluster}")
 
 
-def validate_string_length(value: str, max_length: int, field_name: str) -> None:
+def validate_string_length(
+    value: str, max_length: int, field_name: str
+) -> None:
     """Validate string length against maximum constraint.
 
     Args:
@@ -262,5 +270,6 @@ def validate_url(url: str, field_name: str) -> None:
     allowed_schemes = ("http://", "https://", "ipfs://", "ar://")
     if not any(url.startswith(scheme) for scheme in allowed_schemes):
         raise ValueError(
-            f"{field_name} must start with one of: {', '.join(allowed_schemes)}"
+            f"{field_name} must start with one of: "
+            f"{', '.join(allowed_schemes)}"
         )

python/solana_ai_registries/exceptions.py

@@ -113,7 +113,9 @@ def __init__(self, required: int, available: int, token_mint: str):
             available: Available amount in base units
             token_mint: Token mint address
         """
-        message = f"Insufficient funds: required {required}, available {available}"
+        message = (
+            f"Insufficient funds: required {required}, available {available}"
+        )
         details = {
             "required": required,
             "available": available,
@@ -171,7 +173,8 @@ def __init__(self, setting: str, value: Any, expected: str):
             expected: Description of expected value
         """
         message = (
-            f"Invalid configuration for '{setting}': expected {expected}, got {value}"
+            f"Invalid configuration for '{setting}': "
+            f"expected {expected}, got {value}"
         )
         details = {"setting": setting, "value": value, "expected": expected}
         super().__init__(message, details)

python/solana_ai_registries/types.py

@@ -79,7 +79,9 @@ def __post_init__(self) -> None:
         validate_string_length(self.url, 256, "endpoint URL")
         validate_url(self.url, "endpoint URL")
         if self.description:
-            validate_string_length(self.description, 512, "endpoint description")
+            validate_string_length(
+                self.description, 512, "endpoint description"
+            )
 
 
 @dataclass
@@ -146,7 +148,9 @@ def __post_init__(self) -> None:
             validate_string_length(self.provider_url, 256, "provider_url")
             validate_url(self.provider_url, "provider_url")
         if self.documentation_url:
-            validate_string_length(self.documentation_url, 256, "documentation_url")
+            validate_string_length(
+                self.documentation_url, 256, "documentation_url"
+            )
             validate_url(self.documentation_url, "documentation_url")
 
         # Validate collections
@@ -180,7 +184,8 @@ def from_account_data(cls, data: Dict[str, Any]) -> "AgentRegistryEntry":
             provider_url=data.get("provider_url"),
             documentation_url=data.get("documentation_url"),
             service_endpoints=[
-                ServiceEndpoint(**ep) for ep in data.get("service_endpoints", [])
+                ServiceEndpoint(**ep)
+                for ep in data.get("service_endpoints", [])
             ],
             capabilities_flags=data.get("capabilities_flags", 0),
             supported_input_modes=data.get("supported_input_modes", []),
@@ -189,7 +194,9 @@ def from_account_data(cls, data: Dict[str, Any]) -> "AgentRegistryEntry":
             security_info_uri=data.get("security_info_uri"),
             aea_address=data.get("aea_address"),
             economic_intent_summary=data.get("economic_intent_summary"),
-            supported_aea_protocols_hash=data.get("supported_aea_protocols_hash"),
+            supported_aea_protocols_hash=data.get(
+                "supported_aea_protocols_hash"
+            ),
             extended_metadata_uri=data.get("extended_metadata_uri"),
             tags=data.get("tags", []),
             created_at=data.get("created_at", 0),
@@ -314,7 +321,9 @@ def __post_init__(self) -> None:
             validate_string_length(tag, 32, "server tag")
 
     @classmethod
-    def from_account_data(cls, data: Dict[str, Any]) -> "McpServerRegistryEntry":
+    def from_account_data(
+        cls, data: Dict[str, Any]
+    ) -> "McpServerRegistryEntry":
         """Create instance from on-chain account data.
 
         Args:
@@ -334,8 +343,13 @@ def from_account_data(cls, data: Dict[str, Any]) -> "McpServerRegistryEntry":
                 resource_count=cap_data.get("resource_count", 0),
                 prompt_count=cap_data.get("prompt_count", 0),
                 tools=[McpTool(**tool) for tool in cap_data.get("tools", [])],
-                resources=[McpResource(**res) for res in cap_data.get("resources", [])],
-                prompts=[McpPrompt(**prompt) for prompt in cap_data.get("prompts", [])],
+                resources=[
+                    McpResource(**res) for res in cap_data.get("resources", [])
+                ],
+                prompts=[
+                    McpPrompt(**prompt)
+                    for prompt in cap_data.get("prompts", [])
+                ],
             )
 
         return cls(

python/tests/conftest.py

@@ -14,8 +14,10 @@
 from solders.pubkey import Pubkey
 
 from solana_ai_registries import SolanaAIRegistriesClient
-from solana_ai_registries.constants import DEFAULT_DEVNET_RPC
-from solana_ai_registries.types import AgentRegistryEntry, McpServerRegistryEntry
+from solana_ai_registries.types import (
+    AgentRegistryEntry,
+    McpServerRegistryEntry,
+)
 
 
 @pytest.fixture(scope="session")  # type: ignore[misc]
@@ -126,7 +128,9 @@ def mock_mcp_server_data() -> Dict[str, Any]:
                     "description": "Get current weather for a location",
                     "tags": ["weather", "current"],
                     "input_schema": '{"location": "string"}',
-                    "output_schema": '{"temperature": "number", "conditions": "string"}',
+                    "output_schema": (
+                        '{"temperature": "number", "conditions": "string"}'
+                    ),
                 }
             ],
             "resources": [
@@ -164,7 +168,10 @@ def mock_mcp_server_entry(
 @pytest.fixture
 def mock_transaction_signature() -> str:
     """Mock transaction signature for testing."""
-    return "5j7XiLkBJbkzCqFbp8XjJ6Ci9k2YfRvB4Y6fJr8bNfQ9YmZrVs8P2V5zQqKpWqGrE8F4TtJzN7YzXvKgRrNfQ2"
+    return (
+        "5j7XiLkBJbkzCqFbp8XjJ6Ci9k2YfRvB4Y6fJr8bNfQ9YmZrVs8P2V5zQqKpWqGrE8F4T"
+        "tJzN7YzXvKgRrNfQ2"
+    )
 
 
 @pytest.fixture
@@ -205,7 +212,9 @@ def pytest_configure(config: Any) -> None:
     config.addinivalue_line(
         "markers", "devnet: tests specifically for devnet environment"
     )
-    config.addinivalue_line("markers", "slow: tests that take longer than 1 second")
+    config.addinivalue_line(
+        "markers", "slow: tests that take longer than 1 second"
+    )
 
 
 @pytest.fixture(autouse=True)

python/tests/test_imports.py

@@ -1,7 +1,5 @@
 """Basic import test to verify the package is correctly installed."""
 
-import pytest
-
 
 def test_package_imports() -> None:
     """Test that all main package components can be imported."""
@@ -18,8 +16,15 @@ def test_package_imports() -> None:
     # Type imports
     from solana_ai_registries.types import AgentStatus, McpServerStatus
 
-    # All imports successful
-    assert True
+    # Verify imports by checking they are not None
+    assert SolanaAIRegistriesClient is not None
+    assert AgentRegistry is not None
+    assert McpServerRegistry is not None
+    assert PaymentManager is not None
+    assert AGENT_REGISTRY_PROGRAM_ID is not None
+    assert SolanaAIRegistriesError is not None
+    assert AgentStatus is not None
+    assert McpServerStatus is not None
 
 
 def test_placeholder_classes_exist() -> None: