diff --git a/pyproject.toml b/pyproject.toml index bbf12a8..67b3582 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -23,8 +23,8 @@ classifiers = [ "Programming Language :: Python :: 3.13" ] dependencies = [ - "fastmcp==2.12.3", # Pin to exact version to fix PublicIdentifier schema bug - "httpx>=0.25.0", + "fastmcp==4.0.5", + "httpx2>=2.13", "pydantic>=2.0.0", ] diff --git a/src/clients/cortex.py b/src/clients/cortex.py index d0746d3..712225f 100644 --- a/src/clients/cortex.py +++ b/src/clients/cortex.py @@ -1,6 +1,6 @@ """Cortex API client configuration.""" -import httpx +import httpx2 as httpx from ..config import Config from ..utils.logging import get_logger diff --git a/src/components/customizers.py b/src/components/customizers.py index dd2ccca..33ee728 100644 --- a/src/components/customizers.py +++ b/src/components/customizers.py @@ -1,11 +1,11 @@ """Component customization for MCP server.""" -from fastmcp.server.openapi import ( - HTTPRoute, +from fastmcp.server.providers.openapi import ( OpenAPIResource, OpenAPIResourceTemplate, OpenAPITool, ) +from fastmcp.utilities.openapi.models import HTTPRoute from ..utils.logging import get_logger diff --git a/src/routes/mappers.py b/src/routes/mappers.py index c893014..e52523a 100644 --- a/src/routes/mappers.py +++ b/src/routes/mappers.py @@ -1,6 +1,7 @@ """Route mapping logic for MCP server.""" -from fastmcp.server.openapi import HTTPRoute, MCPType +from fastmcp.server.providers.openapi import MCPType +from fastmcp.utilities.openapi.models import HTTPRoute from ..utils.logging import get_logger diff --git a/src/utils/logging.py b/src/utils/logging.py index a2fe5c2..abd3db7 100644 --- a/src/utils/logging.py +++ b/src/utils/logging.py @@ -26,7 +26,7 @@ def setup_logging(log_level: str | None = None) -> logging.Logger: ) if not Config.DEBUG: - logging.getLogger("httpx").setLevel(logging.WARNING) + logging.getLogger("httpx2").setLevel(logging.WARNING) logging.getLogger("httpcore").setLevel(logging.WARNING) logging.getLogger("langchain").setLevel(logging.INFO) logging.getLogger("langchain_community").setLevel(logging.INFO) diff --git a/tests/test_components.py b/tests/test_components.py index d34e572..db5beb7 100644 --- a/tests/test_components.py +++ b/tests/test_components.py @@ -1,7 +1,11 @@ """Tests for component customization.""" from unittest.mock import Mock -from fastmcp.server.openapi import OpenAPIResource, OpenAPIResourceTemplate, OpenAPITool +from fastmcp.server.providers.openapi import ( + OpenAPIResource, + OpenAPIResourceTemplate, + OpenAPITool, +) from src.components.customizers import customize_components diff --git a/tests/test_routes.py b/tests/test_routes.py index 65938b2..97127cb 100644 --- a/tests/test_routes.py +++ b/tests/test_routes.py @@ -1,7 +1,7 @@ """Tests for route mapping logic.""" from unittest.mock import Mock -from fastmcp.server.openapi import MCPType +from fastmcp.server.providers.openapi import MCPType from src.routes.mappers import custom_route_mapper