added tool mcp upload

api.py

@@ -73,6 +73,7 @@ async def upload_docs(
         saved_files.append(file.filename)
     return {"status": "ok", "uploaded_files": saved_files, "docs_path": docs_path}
 
+
 @app.post("/optimize_rag")
 def optimize_rag(req: OptimizeRequest):
     logger.info("Received optimize_rag request: %s", req.json())

app.py

@@ -6,6 +6,8 @@ import threading
 from models import OptimizeRequest, AutotuneRequest, QARequest
 from api import start_api
 
+
+
 # Start FastAPI server in background
 threading.Thread(target=start_api, daemon=True).start()
 
@@ -29,39 +31,53 @@ def upload_docs_tool(files, docs_path="data/docs"):
     - URLs (str)
     - file-like objects (from Cursor attachment)
     """
-    import shutil
+    import shutil, tempfile
 
     os.makedirs(docs_path, exist_ok=True)
     files_payload = []
 
-    for f in files:
-        if isinstance(f, str) and (f.startswith("http://") or f.startswith("https://")):
-            # Download URL temporarily
-            fname = f.split("/")[-1]
-            tmp_path = os.path.join("/tmp", fname)
-            r = requests.get(f, stream=True)
-            with open(tmp_path, "wb") as out_file:
-                shutil.copyfileobj(r.raw, out_file)
-            files_payload.append(("files", open(tmp_path, "rb")))
-        elif isinstance(f, str):
-            # Local file path
-            files_payload.append(("files", open(f, "rb")))
-        else:
-            # Assume file-like object (Cursor attachment)
-            files_payload.append(("files", f))
+    temp_files = []
 
     try:
+        for f in files:
+            if isinstance(f, str) and f.startswith(("http://", "https://")):
+                # Download URL to a temp file
+                fname = os.path.basename(f)
+                tmp = tempfile.NamedTemporaryFile(delete=False)
+                r = requests.get(f, stream=True)
+                shutil.copyfileobj(r.raw, tmp)
+                tmp.close()
+                temp_files.append(tmp.name)
+                files_payload.append(("files", open(tmp.name, "rb")))
+
+            elif isinstance(f, str):
+                # Local file path
+                files_payload.append(("files", open(f, "rb")))
+
+            else:
+                # File-like object
+                files_payload.append(("files", f))
+
         resp = requests.post(
             f"{BASE_INTERNAL}/upload_docs",
             files=files_payload,
             data={"docs_path": docs_path}
         )
+        resp.raise_for_status()
         return resp.json()
+
     finally:
         # Close all file handles
         for _, file_obj in files_payload:
             if not file_obj.closed:
                 file_obj.close()
+        # Clean up temp files
+        for tmp_file in temp_files:
+            try:
+                os.unlink(tmp_file)
+            except Exception:
+                pass
+
 
 
 def optimize_rag_tool(payload: str) -> str:
@@ -86,33 +102,7 @@ generate_qa_tool.__doc__ = QARequest.__doc__
 
 
 def model_to_json(model_cls) -> str:
-    return json.dumps({k: v.default for k, v in model_cls.__fields__.items()}, indent=2)
-
-
-def upload_docs_tool_mcp(payload: str, files=None):
-    """
-    📦 Upload documents to the server's docs folder via FastAPI /upload_docs.
-    Accepts:
-    - attached file(s) from Cursor MCP agent
-    - 'docs_path' in JSON payload
-    """
-    data = json.loads(payload) if payload else {}
-    docs_path = data.get("docs_path", DEFAULT_UPLOAD_PATH)
-
-    if not files:
-        return {"error": "No files provided to upload_docs_tool_mcp"}
-
-    os.makedirs(docs_path, exist_ok=True)
-
-    files_payload = [("files", (f.name, f, "application/octet-stream")) for f in files]
-
-    resp = requests.post(
-        f"{BASE_INTERNAL}/upload_docs",
-        files=files_payload,
-        data={"docs_path": docs_path}
-    )
-
-    return resp.json()
+    return json.dumps({k: v.default for k, v in model_cls.model_fields.items()}, indent=2)
 
 
 # Default inputs
@@ -167,19 +157,6 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
         gr.Markdown("---")
 
 if __name__ == "__main__":
-    # Register MCP tools
-    demo.add_mcp_tool(
-        name="ragmint_upload_docs_mcp",
-        func=upload_docs_tool_mcp,
-        description="📂 Upload documents to data/docs for use in RAG optimization and autotuning",
-        parameters={
-            "type": "object",
-            "properties": {
-                "docs_path": {"type": "string", "description": "Target directory for uploaded docs"}
-            }
-        },
-        file_input=True  # <<< enables file attachments from Cursor
-    )
 
     demo.launch(
         server_name="0.0.0.0",