refactor: replace mcp.types as types to from mcp import types (#1986)
This commit is contained in:
parent
4fc49c62bd
commit
1a8c14a5b8
4 changed files with 66 additions and 73 deletions
|
|
@ -4,8 +4,7 @@ from unittest.mock import patch
|
|||
import anyio
|
||||
import pytest
|
||||
|
||||
import mcp.types as types
|
||||
from mcp import Client
|
||||
from mcp import Client, types
|
||||
from mcp.client.session import ClientSession
|
||||
from mcp.server import Server
|
||||
from mcp.server.lowlevel import NotificationOptions
|
||||
|
|
|
|||
|
|
@ -3,8 +3,7 @@ from typing import Any
|
|||
import anyio
|
||||
import pytest
|
||||
|
||||
import mcp.types as types
|
||||
from mcp import Client
|
||||
from mcp import Client, types
|
||||
from mcp.client.session import ClientSession
|
||||
from mcp.server.lowlevel.server import Server
|
||||
from mcp.shared.exceptions import MCPError
|
||||
|
|
|
|||
|
|
@ -19,7 +19,7 @@ from starlette.responses import Response
|
|||
from starlette.routing import Mount, Route
|
||||
|
||||
import mcp.client.sse
|
||||
import mcp.types as types
|
||||
from mcp import types
|
||||
from mcp.client.session import ClientSession
|
||||
from mcp.client.sse import _extract_session_id_from_endpoint, sse_client
|
||||
from mcp.server import Server
|
||||
|
|
|
|||
|
|
@ -25,8 +25,7 @@ from starlette.applications import Starlette
|
|||
from starlette.requests import Request
|
||||
from starlette.routing import Mount
|
||||
|
||||
import mcp.types as types
|
||||
from mcp import MCPError
|
||||
from mcp import MCPError, types
|
||||
from mcp.client.session import ClientSession
|
||||
from mcp.client.streamable_http import StreamableHTTPTransport, streamable_http_client
|
||||
from mcp.server import Server
|
||||
|
|
|
|||
Loading…
Reference in a new issue