from llama_index.tools.mcp import BasicMCPClient | |
from dotenv import load_dotenv | |
import os | |
import asyncio | |
load_dotenv() | |
HF_TOKEN = os.getenv("HF_TOKEN") | |
print(HF_TOKEN) | |
http_client = BasicMCPClient("https://huggingface.co/mcp",timeout=60, headers={"Authorization": f"Bearer {HF_TOKEN}"}) | |
print(http_client) | |
async def connect_and_get_tools(): | |
# List available tools | |
tools = await asyncio.wait_for(http_client.list_tools(), timeout=10.0) | |
return tools | |
async def call_tool(tool_name, tool_args): | |
result = await asyncio.wait_for( | |
http_client.call_tool(tool_name, tool_args), | |
timeout=30.0 | |
) | |
return result | |