Spaces:
Running
Running
Refactor MCPConfig import in transports.py
Browse files- Consolidate MCPConfig and infer_transport_type_from_url imports into a single line.
- Remove unnecessary TYPE_CHECKING block for MCPConfig import.
- Clean up imports for better readability.
src/fastmcp/client/transports.py
CHANGED
|
@@ -8,7 +8,7 @@ import sys
|
|
| 8 |
import warnings
|
| 9 |
from collections.abc import AsyncIterator, Callable
|
| 10 |
from pathlib import Path
|
| 11 |
-
from typing import
|
| 12 |
|
| 13 |
import anyio
|
| 14 |
import httpx
|
|
@@ -25,10 +25,7 @@ from fastmcp.client.auth.oauth import OAuth
|
|
| 25 |
from fastmcp.server.dependencies import get_http_headers
|
| 26 |
from fastmcp.server.server import FastMCP
|
| 27 |
from fastmcp.utilities.logging import get_logger
|
| 28 |
-
from fastmcp.utilities.mcp_config import infer_transport_type_from_url
|
| 29 |
-
|
| 30 |
-
if TYPE_CHECKING:
|
| 31 |
-
from fastmcp.utilities.mcp_config import MCPConfig
|
| 32 |
|
| 33 |
logger = get_logger(__name__)
|
| 34 |
|
|
@@ -746,7 +743,6 @@ class MCPConfigTransport(ClientTransport):
|
|
| 746 |
|
| 747 |
def __init__(self, config: MCPConfig | dict):
|
| 748 |
from fastmcp.client.client import Client
|
| 749 |
-
from fastmcp.utilities.mcp_config import MCPConfig
|
| 750 |
|
| 751 |
if isinstance(config, dict):
|
| 752 |
config = MCPConfig.from_dict(config)
|
|
|
|
| 8 |
import warnings
|
| 9 |
from collections.abc import AsyncIterator, Callable
|
| 10 |
from pathlib import Path
|
| 11 |
+
from typing import Any, Literal, TypedDict, TypeVar, cast, overload
|
| 12 |
|
| 13 |
import anyio
|
| 14 |
import httpx
|
|
|
|
| 25 |
from fastmcp.server.dependencies import get_http_headers
|
| 26 |
from fastmcp.server.server import FastMCP
|
| 27 |
from fastmcp.utilities.logging import get_logger
|
| 28 |
+
from fastmcp.utilities.mcp_config import MCPConfig, infer_transport_type_from_url
|
|
|
|
|
|
|
|
|
|
| 29 |
|
| 30 |
logger = get_logger(__name__)
|
| 31 |
|
|
|
|
| 743 |
|
| 744 |
def __init__(self, config: MCPConfig | dict):
|
| 745 |
from fastmcp.client.client import Client
|
|
|
|
| 746 |
|
| 747 |
if isinstance(config, dict):
|
| 748 |
config = MCPConfig.from_dict(config)
|