From b20de51512af1aca9ca06f37236b7eaaeb3268fd Mon Sep 17 00:00:00 2001 From: Jeremiah Lowin <153965+jlowin@users.noreply.github.com> Date: Tue, 9 Dec 2025 12:28:01 -0500 Subject: [PATCH] Remove TaskConfig and client from root exports (#2580) --- docs/servers/tasks.mdx | 3 ++- src/fastmcp/__init__.py | 3 --- tests/server/tasks/test_task_config_modes.py | 3 ++- tests/server/tasks/test_task_mount.py | 3 ++- 4 files changed, 6 insertions(+), 6 deletions(-) diff --git a/docs/servers/tasks.mdx b/docs/servers/tasks.mdx index 7e73a4860..049efd84c 100644 --- a/docs/servers/tasks.mdx +++ b/docs/servers/tasks.mdx @@ -69,7 +69,8 @@ For fine-grained control over task execution behavior, use `TaskConfig` instead | `"required"` | Error: task required | Executes as background task | ```python -from fastmcp import FastMCP, TaskConfig +from fastmcp import FastMCP +from fastmcp.server.tasks import TaskConfig mcp = FastMCP("MyServer") diff --git a/src/fastmcp/__init__.py b/src/fastmcp/__init__.py index 9d5a85698..14c0abc5b 100644 --- a/src/fastmcp/__init__.py +++ b/src/fastmcp/__init__.py @@ -14,7 +14,6 @@ if settings.log_enabled: from fastmcp.server.server import FastMCP from fastmcp.server.context import Context -from fastmcp.server.tasks.config import TaskConfig import fastmcp.server from fastmcp.client import Client @@ -32,7 +31,5 @@ __all__ = [ "Client", "Context", "FastMCP", - "TaskConfig", - "client", "settings", ] diff --git a/tests/server/tasks/test_task_config_modes.py b/tests/server/tasks/test_task_config_modes.py index 6870301b5..c00879996 100644 --- a/tests/server/tasks/test_task_config_modes.py +++ b/tests/server/tasks/test_task_config_modes.py @@ -8,9 +8,10 @@ Tests that the server correctly enforces task execution modes: import pytest -from fastmcp import FastMCP, TaskConfig +from fastmcp import FastMCP from fastmcp.client import Client from fastmcp.exceptions import ToolError +from fastmcp.server.tasks import TaskConfig class TestTaskConfigNormalization: diff --git a/tests/server/tasks/test_task_mount.py b/tests/server/tasks/test_task_mount.py index ffff3cfc0..bdd234894 100644 --- a/tests/server/tasks/test_task_mount.py +++ b/tests/server/tasks/test_task_mount.py @@ -10,9 +10,10 @@ import asyncio import pytest from docket import Docket -from fastmcp import FastMCP, TaskConfig +from fastmcp import FastMCP from fastmcp.client import Client from fastmcp.server.dependencies import CurrentDocket, CurrentFastMCP +from fastmcp.server.tasks import TaskConfig @pytest.fixture(autouse=True)