refactor: replace mcp.types as types to from mcp import types (#1986)
This commit is contained in:
parent
4fc49c62bd
commit
1a8c14a5b8
6 changed files with 66 additions and 73 deletions
|
|
@ -5,8 +5,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.server.lowlevel.server import Server
|
||||
from mcp.shared.exceptions import MCPError
|
||||
from mcp.types import (
|
||||
|
|
|
|||
|
|
@ -2,7 +2,7 @@ from unittest.mock import AsyncMock, Mock
|
|||
|
||||
import pytest
|
||||
|
||||
import mcp.types as types
|
||||
from mcp import types
|
||||
from mcp.server.lowlevel.server import Server
|
||||
from mcp.server.session import ServerSession
|
||||
from mcp.shared.session import RequestResponder
|
||||
|
|
|
|||
|
|
@ -4,7 +4,7 @@ from tempfile import NamedTemporaryFile
|
|||
|
||||
import pytest
|
||||
|
||||
import mcp.types as types
|
||||
from mcp import types
|
||||
from mcp.server.lowlevel.server import ReadResourceContents, Server
|
||||
|
||||
|
||||
|
|
|
|||
|
|
@ -3,7 +3,7 @@ from typing import Any
|
|||
import anyio
|
||||
import pytest
|
||||
|
||||
import mcp.types as types
|
||||
from mcp import types
|
||||
from mcp.client.session import ClientSession
|
||||
from mcp.server import Server
|
||||
from mcp.server.lowlevel import NotificationOptions
|
||||
|
|
|
|||
|
|
@ -9,7 +9,7 @@ This is critical for HTTP transport where requests can arrive in any order.
|
|||
import anyio
|
||||
import pytest
|
||||
|
||||
import mcp.types as types
|
||||
from mcp import types
|
||||
from mcp.server.models import InitializationOptions
|
||||
from mcp.server.session import ServerSession
|
||||
from mcp.shared.message import SessionMessage
|
||||
|
|
|
|||
|
|
@ -13,7 +13,7 @@ from typing import Any
|
|||
import anyio
|
||||
import pytest
|
||||
|
||||
import mcp.types as types
|
||||
from mcp import types
|
||||
from mcp.server.models import InitializationOptions
|
||||
from mcp.server.session import ServerSession
|
||||
from mcp.shared.exceptions import StatelessModeNotSupported
|
||||
|
|
|
|||
Loading…
Reference in a new issue