Merge pull request #130 from ai-agent-assembly/v0.0.1/AAASM-3017/lint/clear_mypy

[AAASM-3017] 🚨 (python-sdk): Clear pre-existing mypy errors

Author: Chisanan232

.pre-commit-config.yaml

@@ -45,17 +45,25 @@ repos:
             build|
             dist"""
 
-  - repo: https://github.com/pre-commit/mirrors-mypy
-    rev: v1.16.0
+  - repo: local
     hooks:
       - id: mypy
-        package: agent_assembly,test
-  #      exclude: ^test/unit_test.{1,64}.py
+        name: mypy
+        # Run mypy inside the uv-managed project env so it resolves the SDK's
+        # runtime + optional framework deps (pydantic, httpx, pydantic-ai, …).
+        # The upstream mirrors-mypy hook installs into an isolated venv that lacks
+        # these, which made every BaseModel/framework subclass resolve to `Any`.
+        # Type-check the whole project via mypy.ini's `packages =` discovery (no
+        # file paths) so the generated proto .py/.pyi pairs don't trip a
+        # "Duplicate module" error and the hook stays identical to the configured
+        # `mypy` invocation.
+        entry: uv run mypy
+        language: system
+        types: [python]
+        pass_filenames: false
         args:
-  #        - --strict
           - --ignore-missing-imports
           - --show-traceback
-        additional_dependencies: [types-PyYAML>=6.0.12.9]
 
   - repo: https://github.com/astral-sh/uv-pre-commit
     # uv version.

agent_assembly/__init__.py

@@ -95,22 +95,26 @@ def __dir__() -> list[str]:
 
 
 if TYPE_CHECKING:
-    from agent_assembly.adapters import FrameworkAdapter, GovernanceInterceptor
-    from agent_assembly.core import AssemblyContext, init_assembly
+    from agent_assembly.adapters import FrameworkAdapter as FrameworkAdapter
+    from agent_assembly.adapters import GovernanceInterceptor as GovernanceInterceptor
+    from agent_assembly.core import AssemblyContext as AssemblyContext
+    from agent_assembly.core import init_assembly as init_assembly
     from agent_assembly.exceptions import (
-        AdapterValidationError,
-        AgentError,
-        AssemblyError,
-        ConfigurationError,
-        GatewayError,
-        MCPToolBlockedError,
-        PolicyError,
-        ToolExecutionBlockedError,
+        AdapterValidationError as AdapterValidationError,
     )
-    from agent_assembly.types import AuditEvent, CallStackNode, CallStackNodeKind
+    from agent_assembly.exceptions import AgentError as AgentError
+    from agent_assembly.exceptions import AssemblyError as AssemblyError
+    from agent_assembly.exceptions import ConfigurationError as ConfigurationError
+    from agent_assembly.exceptions import GatewayError as GatewayError
+    from agent_assembly.exceptions import MCPToolBlockedError as MCPToolBlockedError
+    from agent_assembly.exceptions import PolicyError as PolicyError
+    from agent_assembly.exceptions import (
+        ToolExecutionBlockedError as ToolExecutionBlockedError,
+    )
+    from agent_assembly.types import AuditEvent as AuditEvent
+    from agent_assembly.types import CallStackNode as CallStackNode
+    from agent_assembly.types import CallStackNodeKind as CallStackNodeKind
 
     with contextlib.suppress(ImportError):
-        from agent_assembly._core import (
-            GovernanceEvent,
-            RuntimeClient,
-        )
+        from agent_assembly._core import GovernanceEvent as GovernanceEvent
+        from agent_assembly._core import RuntimeClient as RuntimeClient

agent_assembly/adapters/crewai/patch.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-import importlib
+import importlib as importlib
 from collections.abc import Mapping
 from dataclasses import dataclass
 from functools import wraps

agent_assembly/adapters/google_adk/patch.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-import importlib
+import importlib as importlib
 import inspect
 from collections.abc import Mapping
 from dataclasses import dataclass

agent_assembly/adapters/mcp/patch.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-import importlib
+import importlib as importlib
 import importlib.util
 import inspect
 from dataclasses import dataclass

agent_assembly/adapters/openai_agents/patch.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-import importlib
+import importlib as importlib
 import importlib.util
 import inspect
 from dataclasses import dataclass, field

agent_assembly/adapters/pydantic_ai/patch.py

@@ -2,7 +2,7 @@
 
 from __future__ import annotations
 
-import importlib
+import importlib as importlib
 import inspect
 from collections.abc import Mapping
 from dataclasses import dataclass

agent_assembly/client/emitter.py

@@ -4,6 +4,7 @@
 
 import logging
 import threading
+from typing import Any
 
 from agent_assembly.client.gateway import GatewayClient
 
@@ -21,7 +22,7 @@ def emit(
         source_agent_id: str,
         target_agent_id: str,
         edge_type: str,
-        metadata: dict | None = None,
+        metadata: dict[str, Any] | None = None,
     ) -> None:
         """Schedule a fire-and-forget edge report on a daemon thread."""
         t = threading.Thread(
@@ -36,7 +37,7 @@ def _send(
         source_agent_id: str,
         target_agent_id: str,
         edge_type: str,
-        metadata: dict | None,
+        metadata: dict[str, Any] | None,
     ) -> None:
         try:
             self._client.report_edge(source_agent_id, target_agent_id, edge_type, metadata)

agent_assembly/client/gateway.py

@@ -96,7 +96,7 @@ def __exit__(self, *args: object) -> None:
         """Context manager exit."""
         self.close()
 
-    async def register_agent(self) -> dict:
+    async def register_agent(self) -> dict[str, Any]:
         """
         Register the agent with the governance gateway.
 
@@ -125,11 +125,12 @@ async def register_agent(self) -> dict:
                 json=body if body else None,
             )
             response.raise_for_status()
-            return response.json()
+            data: dict[str, Any] = response.json()
+            return data
         except httpx.HTTPError as e:
             raise GatewayError(f"Failed to register agent: {e}") from e
 
-    async def check_policy_compliance(self, action: str) -> dict:
+    async def check_policy_compliance(self, action: str) -> dict[str, Any]:
         """
         Check if an action complies with governance policies.
 
@@ -148,7 +149,8 @@ async def check_policy_compliance(self, action: str) -> dict:
                 json={"action": action},
             )
             response.raise_for_status()
-            return response.json()
+            data: dict[str, Any] = response.json()
+            return data
         except httpx.HTTPError as e:
             raise GatewayError(f"Failed to check policy compliance: {e}") from e
 
@@ -157,8 +159,8 @@ def report_edge(
         source_agent_id: str,
         target_agent_id: str,
         edge_type: str,
-        metadata: dict | None = None,
-    ) -> dict:
+        metadata: dict[str, Any] | None = None,
+    ) -> dict[str, Any]:
         """
         Report a directed edge between two agents to the topology store.
 
@@ -176,7 +178,7 @@ def report_edge(
         """
         import json as _json
 
-        body: dict = {
+        body: dict[str, str] = {
             "source_agent_id": source_agent_id,
             "target_agent_id": target_agent_id,
             "edge_type": edge_type,
@@ -186,7 +188,8 @@ def report_edge(
         try:
             response = self.client.post("/topology/edges", json=body)
             response.raise_for_status()
-            return response.json()
+            data: dict[str, Any] = response.json()
+            return data
         except httpx.HTTPError as e:
             raise GatewayError(f"Failed to report edge: {e}") from e
 

agent_assembly/core/assembly.py

@@ -78,7 +78,7 @@ class AssemblyContext:
     def __enter__(self) -> AssemblyContext:
         return self
 
-    def __exit__(self, exc_type: object, exc: object, tb: object) -> bool:
+    def __exit__(self, exc_type: object, exc: object, tb: object) -> Literal[False]:
         del exc_type, exc, tb
         self.shutdown()
         return False