Revamp stuff

README.md

@@ -4,7 +4,6 @@ emoji: 🐢
 colorFrom: green
 colorTo: red
 sdk: gradio
-sdk_version: 5.27.1
 app_file: app.py
 pinned: false
 license: mit

app.py

@@ -11,6 +11,7 @@ from utils.huggingface_mcp_llamaindex import connect_and_get_tools, call_tool
 from prompts.devstral_coding_prompt import devstral_code_gen_sys_prompt, devstral_code_gen_user_prompt
 from dotenv import load_dotenv
 import os
+import asyncio
 load_dotenv()
 
 # Import Modal inference function

requirements.txt

@@ -1,6 +1,7 @@
-llama-index-tools-mcp==0.2.4
+mcp
+llama-index-tools-mcp
 google-genai==1.19.0
-gradio==5.27.1
+gradio
 pandas==2.3.0
 python-dotenv==1.0.1
 openpyxl==3.1.5

utils/huggingface_mcp_llamaindex.py

@@ -1,6 +1,7 @@
 from llama_index.tools.mcp import BasicMCPClient
 from dotenv import load_dotenv
 import os
+import asyncio
 load_dotenv()
 HF_TOKEN = os.getenv("HF_TOKEN")
 async def connect_and_get_tools():
@@ -8,10 +9,13 @@ async def connect_and_get_tools():
     http_client = BasicMCPClient("https://huggingface.co/mcp", headers={"Authorization": f"Bearer {HF_TOKEN}"})
 
     # List available tools
-    tools = await http_client.list_tools()
+    tools = await asyncio.wait_for(http_client.list_tools(), timeout=10.0)
     return tools
 
 async def call_tool(tool_name, tool_args):
     http_client = BasicMCPClient("https://huggingface.co/mcp", headers={"Authorization": f"Bearer {HF_TOKEN}"}) 
-    result = await http_client.call_tool(tool_name, tool_args)
+    result = await asyncio.wait_for(
+            http_client.call_tool(tool_name, tool_args), 
+            timeout=30.0
+        )
     return result