diff --git a/docs/servers/tasks.mdx b/docs/servers/tasks.mdx index 7e73a4860c..049efd84cf 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 9d5a85698f..14c0abc5b4 100644 --- a/src/fastmcp/__init__.py +++ b/src/fastmcp/__init__.py @@ -14,7 +14,6 @@ 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 @@ "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 6870301b51..c008799968 100644 --- a/tests/server/tasks/test_task_config_modes.py +++ b/tests/server/tasks/test_task_config_modes.py @@ -8,9 +8,10 @@ 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 ffff3cfc04..bdd234894a 100644 --- a/tests/server/tasks/test_task_mount.py +++ b/tests/server/tasks/test_task_mount.py @@ -10,9 +10,10 @@ 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)