From 1faf64b540ae0089a61d2ea9c4aa432ca5b1ed7c Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Fri, 3 Jan 2025 14:16:45 -0800 Subject: [PATCH] linter fixes --- .../model_context_protocol/model_context_protocol.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) 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