@devness/useai 0.6.27 → 0.6.28

package/dist/index.js

@@ -684,7 +684,7 @@ var VERSION;
 var init_version = __esm({
   "../shared/dist/constants/version.js"() {
     "use strict";
-    VERSION = "0.6.27";
+    VERSION = "0.6.28";
   }
 });
 
@@ -34110,6 +34110,7 @@ var init_mcp_map = __esm({
 // src/register-tools.ts
 var register_tools_exports = {};
 __export(register_tools_exports, {
+  installGracefulToolHandler: () => installGracefulToolHandler,
   registerTools: () => registerTools
 });
 import { createHash as createHash3, randomUUID as randomUUID3 } from "crypto";
@@ -34757,6 +34758,23 @@ function registerTools(server2, session2, opts) {
     }
   );
 }
+function installGracefulToolHandler(mcpServer) {
+  const requestHandlers = mcpServer.server?._requestHandlers;
+  if (!requestHandlers?.get) return;
+  const originalHandler = requestHandlers.get("tools/call");
+  if (!originalHandler) return;
+  requestHandlers.set("tools/call", async (request, extra) => {
+    try {
+      return await originalHandler(request, extra);
+    } catch (error2) {
+      const message = error2 instanceof Error ? error2.message : String(error2);
+      return {
+        content: [{ type: "text", text: message }],
+        isError: true
+      };
+    }
+  });
+}
 var init_register_tools = __esm({
   "src/register-tools.ts"() {
     "use strict";
@@ -36362,6 +36380,7 @@ async function startDaemon(port) {
               }
             }
           });
+          installGracefulToolHandler(mcpServer);
           const transport = new StreamableHTTPServerTransport({
             sessionIdGenerator: () => randomUUID4(),
             onsessioninitialized: (newSid) => {
@@ -36729,13 +36748,14 @@ var { McpServer: McpServer2 } = await Promise.resolve().then(() => (init_mcp(),
 var { StdioServerTransport: StdioServerTransport2 } = await Promise.resolve().then(() => (init_stdio2(), stdio_exports));
 var { VERSION: VERSION2, ensureDir: ensureDir2 } = await Promise.resolve().then(() => (init_dist(), dist_exports));
 var { SessionState: SessionState2 } = await Promise.resolve().then(() => (init_session_state(), session_state_exports));
-var { registerTools: registerTools2 } = await Promise.resolve().then(() => (init_register_tools(), register_tools_exports));
+var { registerTools: registerTools2, installGracefulToolHandler: installGracefulToolHandler2 } = await Promise.resolve().then(() => (init_register_tools(), register_tools_exports));
 var session = new SessionState2();
 var server = new McpServer2({
   name: "UseAI",
   version: VERSION2
 });
 registerTools2(server, session);
+installGracefulToolHandler2(server);
 async function main() {
   ensureDir2();
   try {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@devness/useai",
-  "version": "0.6.27",
+  "version": "0.6.28",
   "description": "Track your AI-assisted development workflow. MCP server that records usage metrics across all your AI tools.",
   "keywords": [
     "mcp",