ci: fix python sdk checks

Author: Patel230

.github/workflows/ci.yml

@@ -85,7 +85,7 @@ jobs:
           python -m pip install --upgrade pip pip-audit
           pip install -e ".[dev]"
       - name: pip-audit
-        run: pip-audit
+        run: pip-audit . --skip-editable
 
   build:
     name: build (sdist + wheel)

pyproject.toml

@@ -39,7 +39,7 @@ dev = [
     "pytest-asyncio>=0.21",
     "respx>=0.21",
     "ruff>=0.4.0",
-    "mypy>=1.0",
+    "mypy>=1.0,<2",
 ]
 
 # Read the package version from the repo-root VERSION file at build time.
@@ -85,3 +85,7 @@ python_version = "3.9"
 strict = true
 warn_return_any = true
 warn_unused_configs = true
+
+[[tool.mypy.overrides]]
+module = ["opentelemetry.*"]
+ignore_missing_imports = true

release-please-config.json

@@ -21,7 +21,7 @@
         { "type": "chore",    "section": "Chores",        "hidden": true },
         { "type": "style",    "section": "Style",         "hidden": true }
       ],
-      "extra-files": [{"type":"version-txt","path":"VERSION"}]
+      "version-file": "VERSION"
     }
   }
 }

src/hawk/__init__.py

@@ -3,6 +3,13 @@
 from ._version import __version__
 from .agent import Agent, AgentConfig, AsyncAgent
 from .client import AsyncHawkClient, HawkClient
+from .discovery import (
+    AgentCard,
+    AgentResolver,
+    CompositeResolver,
+    FileResolver,
+    WellKnownResolver,
+)
 from .errors import (
     AuthenticationError,
     BadRequestError,
@@ -13,11 +20,19 @@
     RateLimitError,
     ServiceUnavailableError,
 )
+from .evaluate import (
+    BenchmarkResults,
+    EvalResult,
+    EvalTask,
+    run_benchmark,
+    run_benchmark_async,
+)
+from .memory_tools import MemoryTools
 from .plan import Plan, PlanNotebook, SubTask, SubTaskState
 from .retry import DEFAULT_RETRY_CONFIG, RetryConfig
 from .streaming import AsyncStreamReader, StreamReader
-from .tools import Tool, chat_with_tools, chat_with_tools_async, tool
 from .toolkit import BackgroundTask, ToolGroup, Toolkit
+from .tools import Tool, chat_with_tools, chat_with_tools_async, tool
 from .tracing import (
     configure_tracing,
     detect_provider,
@@ -41,98 +56,83 @@
     ToolCall,
     Usage,
 )
-from .discovery import (
-    AgentCard,
-    AgentResolver,
-    CompositeResolver,
-    FileResolver,
-    WellKnownResolver,
-)
-from .evaluate import (
-    BenchmarkResults,
-    EvalResult,
-    EvalTask,
-    run_benchmark,
-    run_benchmark_async,
-)
-from .memory_tools import MemoryTools
 from .workflow import AsyncWorkflow, Workflow
 
 __all__ = [
-    # Version
-    "__version__",
-    # Client
-    "HawkClient",
-    "AsyncHawkClient",
+    "DEFAULT_RETRY_CONFIG",
     # Agent
     "Agent",
-    "AsyncAgent",
+    # Discovery
+    "AgentCard",
     "AgentConfig",
-    # Streaming
-    "StreamReader",
+    "AgentResolver",
+    "AsyncAgent",
+    "AsyncHawkClient",
     "AsyncStreamReader",
-    # Tools
-    "Tool",
-    "tool",
-    "chat_with_tools",
-    "chat_with_tools_async",
-    # Plan
-    "PlanNotebook",
-    "Plan",
-    "SubTask",
-    "SubTaskState",
-    # Toolkit
-    "Toolkit",
-    "ToolGroup",
-    "BackgroundTask",
-    # Workflow
-    "Workflow",
     "AsyncWorkflow",
-    # Retry
-    "RetryConfig",
-    "DEFAULT_RETRY_CONFIG",
+    "AuthenticationError",
+    "BackgroundTask",
+    "BadRequestError",
+    "BenchmarkResults",
     # Types
     "ChatRequest",
     "ChatResponse",
+    "CompositeResolver",
+    "EvalResult",
+    # Evaluate
+    "EvalTask",
+    "FileResolver",
+    "ForbiddenError",
+    # Errors
+    "HawkAPIError",
+    # Client
+    "HawkClient",
     "HealthResponse",
+    "InternalServerError",
+    # Memory
+    "MemoryTools",
     "Message",
     "ModelStat",
+    "NotFoundError",
     "PaginatedResponse",
+    "Plan",
+    # Plan
+    "PlanNotebook",
+    "RateLimitError",
+    # Retry
+    "RetryConfig",
+    "ServiceUnavailableError",
     "SessionDetail",
     "SessionSummary",
     "StatsResponse",
     "StreamEvent",
     "StreamEventType",
+    # Streaming
+    "StreamReader",
+    "SubTask",
+    "SubTaskState",
+    # Tools
+    "Tool",
     "ToolCall",
+    "ToolGroup",
+    # Toolkit
+    "Toolkit",
     "Usage",
+    "WellKnownResolver",
+    # Workflow
+    "Workflow",
+    # Version
+    "__version__",
+    "chat_with_tools",
+    "chat_with_tools_async",
     # Tracing
     "configure_tracing",
     "detect_provider",
     "is_tracing_enabled",
+    "run_benchmark",
+    "run_benchmark_async",
+    "tool",
     "trace",
     "trace_chat",
     "trace_tool",
-    # Errors
-    "HawkAPIError",
-    "BadRequestError",
-    "AuthenticationError",
-    "ForbiddenError",
-    "NotFoundError",
-    "RateLimitError",
-    "InternalServerError",
-    "ServiceUnavailableError",
-    # Evaluate
-    "EvalTask",
-    "EvalResult",
-    "BenchmarkResults",
-    "run_benchmark",
-    "run_benchmark_async",
-    # Discovery
-    "AgentCard",
-    "AgentResolver",
-    "FileResolver",
-    "WellKnownResolver",
-    "CompositeResolver",
-    # Memory
-    "MemoryTools",
 ]

src/hawk/agent.py

@@ -3,12 +3,14 @@
 from __future__ import annotations
 
 from dataclasses import dataclass, field
-from typing import Any, Optional
+from typing import TYPE_CHECKING, Any
 
-from .client import AsyncHawkClient, HawkClient
-from .streaming import AsyncStreamReader, StreamReader
 from .tools import Tool, chat_with_tools, chat_with_tools_async
-from .types import ChatResponse
+
+if TYPE_CHECKING:
+    from .client import AsyncHawkClient, HawkClient
+    from .streaming import AsyncStreamReader, StreamReader
+    from .types import ChatResponse
 
 
 @dataclass
@@ -26,12 +28,12 @@ class AgentConfig:
     """
 
     name: str = "hawk-agent"
-    model: Optional[str] = None
-    system_prompt: Optional[str] = None
+    model: str | None = None
+    system_prompt: str | None = None
     tools: list[Tool] = field(default_factory=list)
     max_rounds: int = 10
-    temperature: Optional[float] = None
-    top_p: Optional[float] = None
+    temperature: float | None = None
+    top_p: float | None = None
 
 
 class Agent:
@@ -51,15 +53,15 @@ class Agent:
     def __init__(
         self,
         client: HawkClient,
-        config: Optional[AgentConfig] = None,
+        config: AgentConfig | None = None,
     ) -> None:
         self._client = client
         self._config = config or AgentConfig()
-        self._session_id: Optional[str] = None
+        self._session_id: str | None = None
         self._history: list[dict[str, str]] = []
 
     @property
-    def session_id(self) -> Optional[str]:
+    def session_id(self) -> str | None:
         """Current session ID, if one has been established."""
         return self._session_id
 
@@ -132,15 +134,15 @@ class AsyncAgent:
     def __init__(
         self,
         client: AsyncHawkClient,
-        config: Optional[AgentConfig] = None,
+        config: AgentConfig | None = None,
     ) -> None:
         self._client = client
         self._config = config or AgentConfig()
-        self._session_id: Optional[str] = None
+        self._session_id: str | None = None
         self._history: list[dict[str, str]] = []
 
     @property
-    def session_id(self) -> Optional[str]:
+    def session_id(self) -> str | None:
         """Current session ID."""
         return self._session_id