diff --git a/llama_stack/providers/remote/tool_runtime/model_context_protocol/model_context_protocol.py b/llama_stack/providers/remote/tool_runtime/model_context_protocol/model_context_protocol.py index 537ae3ab5..19ada8457 100644 --- a/llama_stack/providers/remote/tool_runtime/model_context_protocol/model_context_protocol.py +++ b/llama_stack/providers/remote/tool_runtime/model_context_protocol/model_context_protocol.py @@ -7,6 +7,9 @@ from typing import Any, Dict, List from urllib.parse import urlparse +from mcp import ClientSession +from mcp.client.sse import sse_client + from llama_stack.apis.tools import ( MCPToolGroupDef, ToolDef, @@ -18,9 +21,6 @@ from llama_stack.apis.tools import ( ) from llama_stack.providers.datatypes import ToolsProtocolPrivate -from mcp import ClientSession -from mcp.client.sse import sse_client - from .config import ModelContextProtocolConfig