Fix weird type errors

plugins-external/mcp-server/src/server.ts

@@ -1,6 +1,6 @@
+import { StreamableHTTPTransport } from '@hono/mcp';
 import { serve } from '@hono/node-server';
 import { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
-import { WebStandardStreamableHTTPServerTransport } from '@modelcontextprotocol/sdk/server/webStandardStreamableHttp.js';
 import { Hono } from 'hono';
 import { registerFolderTools } from './tools/folder.js';
 import { registerHttpRequestTools } from './tools/httpRequest.js';
@@ -10,49 +10,53 @@ import { registerWorkspaceTools } from './tools/workspace.js';
 import type { McpServerContext } from './types.js';
 
 export function createMcpServer(ctx: McpServerContext, port: number) {
-  const server = new McpServer({
+  const mcpServer = new McpServer({
     name: 'yaak-mcp-server',
     version: '0.1.0',
   });
 
   // Register all tools
-  registerToastTools(server, ctx);
-  registerHttpRequestTools(server, ctx);
-  registerFolderTools(server, ctx);
-  registerWindowTools(server, ctx);
-  registerWorkspaceTools(server, ctx);
+  registerToastTools(mcpServer, ctx);
+  registerHttpRequestTools(mcpServer, ctx);
+  registerFolderTools(mcpServer, ctx);
+  registerWindowTools(mcpServer, ctx);
+  registerWorkspaceTools(mcpServer, ctx);
 
-  // Create a stateless transport
-  const transport = new WebStandardStreamableHTTPServerTransport();
-
-  // Create Hono app
   const app = new Hono();
+  const transport = new StreamableHTTPTransport();
 
-  // MCP endpoint - reuse the same transport for all requests
-  app.all('/mcp', (c) => transport.handleRequest(c.req.raw));
-
-  // Connect server to transport
-  server.connect(transport).then(() => {
-    console.log(`MCP Server running at http://127.0.0.1:${port}/mcp`);
+  app.all('/mcp', async (c) => {
+    if (!mcpServer.isConnected()) {
+      // Connect the mcp with the transport
+      await mcpServer.connect(transport);
       ctx.yaak.toast.show({
-      message: `MCP Server running on port ${port}`,
-      icon: 'check_circle',
-      color: 'success',
+        message: `MCP Server connected`,
+        icon: 'info',
+        color: 'info',
+        timeout: 5000,
       });
+    }
+    return transport.handleRequest(c);
   });
 
-  // Start the HTTP server
   const honoServer = serve({
-    fetch: app.fetch,
     port,
     hostname: '127.0.0.1',
+    fetch: app.fetch,
+  });
+
+  ctx.yaak.toast.show({
+    message: `MCP Server running on port ${honoServer.address}`,
+    icon: 'info',
+    color: 'info',
+    timeout: 30000,
   });
 
   return {
-    server,
+    server: mcpServer,
     close: async () => {
       honoServer.close();
-      await server.close();
+      await mcpServer.close();
     },
   };
 }

plugins-external/mcp-server/src/tools/toast.ts

@@ -31,12 +31,12 @@ export function registerToastTools(server: McpServer, ctx: McpServerContext) {
     {
       title: 'Show Toast',
       description: 'Show a toast notification in Yaak',
-      inputSchema: z.object({
+      inputSchema: {
         message: z.string().describe('The message to display'),
         icon: z.enum(ICON_VALUES).optional().describe('Icon name'),
         color: z.enum(COLOR_VALUES).optional().describe('Toast color'),
         timeout: z.number().optional().describe('Timeout in milliseconds'),
-      }),
+      },
     },
     async ({ message, icon, color, timeout }) => {
       await ctx.yaak.toast.show({

plugins-external/mcp-server/src/tools/window.ts

@@ -1,5 +1,4 @@
 import type { McpServer } from '@modelcontextprotocol/sdk/server/mcp.js';
-import * as z from 'zod';
 import type { McpServerContext } from '../types.js';
 import { getWorkspaceContext } from './helpers.js';
 
@@ -9,7 +8,7 @@ export function registerWindowTools(server: McpServer, ctx: McpServerContext) {
     {
       title: 'Get Workspace ID',
       description: 'Get the current workspace ID',
-      inputSchema: z.object({}),
+      inputSchema: {},
     },
     async () => {
       const workspaceCtx = await getWorkspaceContext(ctx);
@@ -31,7 +30,7 @@ export function registerWindowTools(server: McpServer, ctx: McpServerContext) {
     {
       title: 'Get Environment ID',
       description: 'Get the current environment ID',
-      inputSchema: z.object({}),
+      inputSchema: {},
     },
     async () => {
       const workspaceCtx = await getWorkspaceContext(ctx);