Fix tool test in UI and other changes

Signed-off-by: Madhav Kandukuri <[email protected]>

Author: madhav165

mcpgateway/cache/session_registry.py

@@ -407,6 +407,7 @@ async def respond(
         server_id: Optional[str],
         user: json,
         session_id: str,
+        base_url: str,
     ) -> None:
         """Respond to broadcast message is transport relevant to session_id is found locally
 
@@ -425,7 +426,7 @@ async def respond(
             transport = self.get_session_sync(session_id)
             if transport:
                 message = json.loads(self._session_message.get("message"))
-                await self.generate_response(message=message, transport=transport, server_id=server_id, user=user)
+                await self.generate_response(message=message, transport=transport, server_id=server_id, user=user, base_url=base_url)
 
         elif self._backend == "redis":
             await self._pubsub.subscribe(session_id)
@@ -440,7 +441,7 @@ async def respond(
                         message = json.loads(message)
                     transport = self.get_session_sync(session_id)
                     if transport:
-                        await self.generate_response(message=message, transport=transport, server_id=server_id, user=user)
+                        await self.generate_response(message=message, transport=transport, server_id=server_id, user=user, base_url=base_url)
             except asyncio.CancelledError:
                 logger.info(f"PubSub listener for session {session_id} cancelled")
             finally:
@@ -494,7 +495,7 @@ async def message_check_loop(session_id):
                         transport = self.get_session_sync(session_id)
                         if transport:
                             logger.info("Ready to respond")
-                            await self.generate_response(message=message, transport=transport, server_id=server_id, user=user)
+                            await self.generate_response(message=message, transport=transport, server_id=server_id, user=user, base_url=base_url)
 
                             await asyncio.to_thread(_db_remove, session_id, record.message)
 
@@ -671,7 +672,7 @@ async def handle_initialize_logic(self, body: dict) -> InitializeResult:
             instructions=("MCP Gateway providing federated tools, resources and prompts. Use /admin interface for configuration."),
         )
 
-    async def generate_response(self, message: json, transport: SSETransport, server_id: Optional[str], user: dict):
+    async def generate_response(self, message: json, transport: SSETransport, server_id: Optional[str], user: dict, base_url: str):
         """
         Generates response according to SSE specifications
 
@@ -745,9 +746,10 @@ async def generate_response(self, message: json, transport: SSETransport, server
                     "id": 1,
                 }
                 headers = {"Authorization": f"Bearer {user['token']}", "Content-Type": "application/json"}
+                rpc_url = base_url + "/rpc"
                 async with httpx.AsyncClient(timeout=settings.federation_timeout, verify=not settings.skip_ssl_verify) as client:
                     rpc_response = await client.post(
-                        f"http://localhost:{settings.port}/rpc",
+                        url=rpc_url,
                         json=rpc_input,
                         headers=headers,
                     )

mcpgateway/db.py

@@ -1086,3 +1086,6 @@ def init_db():
         Base.metadata.create_all(bind=engine)
     except SQLAlchemyError as e:
         raise Exception(f"Failed to initialize database: {str(e)}")
+
+if __name__ == "__main__":
+    init_db()

mcpgateway/main.py

@@ -620,7 +620,7 @@ async def sse_endpoint(request: Request, server_id: int, user: str = Depends(req
         await session_registry.add_session(transport.session_id, transport)
         response = await transport.create_sse_response(request)
 
-        asyncio.create_task(session_registry.respond(server_id, user, session_id=transport.session_id))
+        asyncio.create_task(session_registry.respond(server_id, user, session_id=transport.session_id, base_url=base_url))
 
         tasks = BackgroundTasks()
         tasks.add_task(session_registry.remove_session, transport.session_id)
@@ -1744,7 +1744,7 @@ async def utility_sse_endpoint(request: Request, user: str = Depends(require_aut
         await transport.connect()
         await session_registry.add_session(transport.session_id, transport)
 
-        asyncio.create_task(session_registry.respond(None, user, session_id=transport.session_id))
+        asyncio.create_task(session_registry.respond(None, user, session_id=transport.session_id, base_url=base_url))
 
         response = await transport.create_sse_response(request)
         tasks = BackgroundTasks()

mcpgateway/static/admin.js

@@ -218,7 +218,7 @@ document.addEventListener("DOMContentLoaded", function () {
       if (mode === "ui") {
         window.schemaEditor.setValue(generateSchema());
       }
-      // Save CodeMirror editors' contents into the underlying textareas
+      // Save CodeMirror editors’ contents into the underlying textareas
       if (window.headersEditor) {
         window.headersEditor.save();
       }
@@ -1579,7 +1579,15 @@ async function runToolTest() {
   const formData = new FormData(form);
   const params = {};
   for (const [key, value] of formData.entries()) {
-    params[key] = value;
+    if(isNaN(value)) {
+      if(value.toLowerCase() ===  "true" || value.toLowerCase() === "false") {
+        params[key] = Boolean(value.toLowerCase() ===  "true");
+      } else {
+        params[key] = value;
+      }
+    } else {
+      params[key] = Number(value);
+    }
   }
 
   // Build the JSON-RPC payload using the tool's name as the method
@@ -1591,7 +1599,7 @@ async function runToolTest() {
   };
 
   // Send the request to your /rpc endpoint
-  fetch("/rpc", {
+  fetch(`${window.ROOT_PATH}/rpc`, {
     method: "POST",
     headers: {
       "Content-Type": "application/json", // ← make sure we include this

pyproject.toml

@@ -75,6 +75,16 @@ postgres = [
     "psycopg2-binary>=2.9.10",
 ]
 
+# Async SQLite Driver (optional)
+aiosqlite = [
+    "aiosqlite>=0.21.0",
+]
+
+# Async PostgreSQL driver (optional)
+asyncpg = [
+    "asyncpg>=0.30.0",
+]
+
 # Optional dependency groups (development)
 dev = [
     "argparse-manpage>=4.6",

run-gunicorn.sh

@@ -36,6 +36,9 @@ if [[ "${SSL}" == "true" ]]; then
     echo "✓  TLS enabled – using ${CERT_FILE} / ${KEY_FILE}"
 fi
 
+# Initialize databases
+python -m mcpgateway.db
+
 exec gunicorn -c gunicorn.config.py \
     --worker-class uvicorn.workers.UvicornWorker \
     --workers "${GUNICORN_WORKERS}" \