aixtools 0.2.19__tar.gz → 0.2.20__tar.gz

{aixtools-0.2.19 → aixtools-0.2.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aixtools
-Version: 0.2.19
+Version: 0.2.20
 Summary: Tools for AI exploration and debugging
 Requires-Python: >=3.11.2
 Description-Content-Type: text/markdown

{aixtools-0.2.19 → aixtools-0.2.20}/aixtools/_version.py

@@ -28,7 +28,7 @@ version_tuple: VERSION_TUPLE
 commit_id: COMMIT_ID
 __commit_id__: COMMIT_ID
 
-__version__ = version = '0.2.19'
-__version_tuple__ = version_tuple = (0, 2, 19)
+__version__ = version = '0.2.20'
+__version_tuple__ = version_tuple = (0, 2, 20)
 
-__commit_id__ = commit_id = 'g09e184e90'
+__commit_id__ = commit_id = 'ge7f4f14bb'

{aixtools-0.2.19 → aixtools-0.2.20}/aixtools/mcp/__init__.py

@@ -4,8 +4,12 @@ Model Context Protocol (MCP) implementation for AI agent communication.
 
 from aixtools.mcp.exceptions import AixToolError
 from aixtools.mcp.fast_mcp_log import FastMcpLog
+from aixtools.mcp.middleware import AixErrorHandlingMiddleware
+from aixtools.mcp.server import create_mcp_server
 
 __all__ = [
-    "FastMcpLog",
+    "AixErrorHandlingMiddleware",
     "AixToolError",
+    "FastMcpLog",
+    "create_mcp_server",
 ]

aixtools-0.2.19/aixtools/logging/mcp_middleware.py → aixtools-0.2.20/aixtools/mcp/middleware.py

@@ -7,7 +7,7 @@ import traceback
 from fastmcp.server.middleware.error_handling import ErrorHandlingMiddleware
 from fastmcp.server.middleware.middleware import MiddlewareContext
 
-from aixtools.mcp import AixToolError
+from aixtools.mcp.exceptions import AixToolError
 
 
 class AixErrorHandlingMiddleware(ErrorHandlingMiddleware):

aixtools-0.2.20/aixtools/mcp/server.py

@@ -0,0 +1,72 @@
+"""Utilities for FastMCP servers."""
+
+from typing import Any
+
+from fastmcp import FastMCP
+from fastmcp.server.middleware.logging import LoggingMiddleware
+from fastmcp.server.middleware.timing import TimingMiddleware
+from fastmcp.utilities.types import NotSet
+
+from aixtools.auth.auth import AccessTokenAuthProvider
+from aixtools.logging.logging_config import get_logger
+from aixtools.mcp.middleware import AixErrorHandlingMiddleware
+
+
+def create_mcp_server(
+    *,
+    name: str,
+    instructions: str | None = None,
+    **kwargs: Any,
+) -> FastMCP:
+    """
+    MCP server instance with preconfigured auth and middleware.
+
+    All FastMCP constructor parameters are supported via **kwargs.
+
+    Args:
+        name: Server name
+        instructions: Optional server instructions
+        **kwargs: All other FastMCP constructor parameters:
+            - version: str | None
+            - auth: AuthProvider | None (AccessTokenAuthProvider if not set, pass None to disable)
+            - middleware: list[Middleware] | None (custom middleware if not set, pass None to disable)
+            - lifespan: Callable | None
+            - tool_serializer: Callable[[Any], str] | None
+            - cache_expiration_seconds: float | None
+            - on_duplicate_tools: DuplicateBehavior | None
+            - on_duplicate_resources: DuplicateBehavior | None
+            - on_duplicate_prompts: DuplicateBehavior | None
+            - resource_prefix_format: Literal["protocol", "path"] | None
+            - mask_error_details: bool | None
+            - tools: list[Tool | Callable] | None
+            - tool_transformations: dict[str, ToolTransformConfig] | None
+            - dependencies: list[str] | None
+            - include_tags: set[str] | None
+            - exclude_tags: set[str] | None
+            - include_fastmcp_meta: bool | None
+
+    Returns:
+        Configured FastMCP server instance
+    """
+    middleware = kwargs.pop("middleware", NotSet)
+    auth = kwargs.pop("auth", NotSet)
+
+    if middleware is NotSet:
+        middleware = [
+            LoggingMiddleware(include_payloads=True, logger=get_logger("middleware.log")),
+            AixErrorHandlingMiddleware(include_traceback=True, logger=get_logger("middleware.err")),
+            TimingMiddleware(logger=get_logger("middleware.timing")),
+        ]
+
+    if auth is NotSet:
+        auth = AccessTokenAuthProvider()
+
+    mcp_args = {
+        "name": name,
+        "instructions": instructions,
+        "auth": auth,
+        "middleware": middleware,
+        **kwargs,
+    }
+
+    return FastMCP(**mcp_args)

{aixtools-0.2.19 → aixtools-0.2.20}/aixtools.egg-info/SOURCES.txt

@@ -61,7 +61,6 @@ aixtools/logging/log_objects.py
 aixtools/logging/logging_config.py
 aixtools/logging/mcp_log_models.py
 aixtools/logging/mcp_logger.py
-aixtools/logging/mcp_middleware.py
 aixtools/logging/model_patch_logging.py
 aixtools/logging/open_telemetry.py
 aixtools/mcp/__init__.py
@@ -71,6 +70,8 @@ aixtools/mcp/example_server.py
 aixtools/mcp/exceptions.py
 aixtools/mcp/fast_mcp_log.py
 aixtools/mcp/faulty_mcp.py
+aixtools/mcp/middleware.py
+aixtools/mcp/server.py
 aixtools/model_patch/model_patch.py
 aixtools/server/__init__.py
 aixtools/server/app_mounter.py