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