From 6b53326854a19f57fc57037670ad6c3f369a75c3 Mon Sep 17 00:00:00 2001 From: Robert Bost Date: Tue, 3 Jun 2025 12:43:21 -0600 Subject: [PATCH 1/2] mcp: Remove CLI functionality with missing deps --- src/llama_stack_client/lib/tools/mcp_oauth.py | 31 ------------------- 1 file changed, 31 deletions(-) diff --git a/src/llama_stack_client/lib/tools/mcp_oauth.py b/src/llama_stack_client/lib/tools/mcp_oauth.py index a3c03416..83525857 100644 --- a/src/llama_stack_client/lib/tools/mcp_oauth.py +++ b/src/llama_stack_client/lib/tools/mcp_oauth.py @@ -10,7 +10,6 @@ import uuid from http.server import BaseHTTPRequestHandler, HTTPServer -import fire import requests logging.basicConfig(level=logging.INFO) @@ -265,33 +264,3 @@ def __init__(self, server_address, auth_code_callback): def get_oauth_token_for_mcp_server(url: str) -> str | None: helper = McpOAuthHelper(url) return helper.initiate_auth_flow() - - -async def run_main(url: str): - from mcp import ClientSession - from mcp.client.sse import sse_client - - token = get_oauth_token_for_mcp_server(url) - if not token: - return - - headers = { - "Authorization": f"Bearer {token}", - } - - async with sse_client(url, headers=headers) as streams: - async with ClientSession(*streams) as session: - await session.initialize() - result = await session.list_tools() - - logger.info(f"Tools: {len(result.tools)}, showing first 5:") - for t in result.tools[:5]: - logger.info(f"{t.name}: {t.description}") - - -def main(url: str): - asyncio.run(run_main(url)) - - -if __name__ == "__main__": - fire.Fire(main) From 52815c7a65e75a89d7f3c326a0843ee4111776de Mon Sep 17 00:00:00 2001 From: Robert Bost Date: Wed, 4 Jun 2025 08:57:14 -0600 Subject: [PATCH 2/2] mcp: Remove unused import --- src/llama_stack_client/lib/tools/mcp_oauth.py | 1 - 1 file changed, 1 deletion(-) diff --git a/src/llama_stack_client/lib/tools/mcp_oauth.py b/src/llama_stack_client/lib/tools/mcp_oauth.py index 83525857..611623ee 100644 --- a/src/llama_stack_client/lib/tools/mcp_oauth.py +++ b/src/llama_stack_client/lib/tools/mcp_oauth.py @@ -1,4 +1,3 @@ -import asyncio import base64 import hashlib import logging