@skilder-ai/runtime 0.7.7 → 0.7.8

package/dist/index.js

@@ -152421,6 +152421,8 @@ var ToolServerService = class ToolServerService2 extends Service {
     this.isPythonServer = false;
     this.onShutdownCallback = async () => {
     };
+    this.debounceTimer = null;
+    this.pendingTools = null;
     this.logger.info(`Initializing MCP server service for ${this.config.name}`);
     this.client = new Client({
       name: this.config.name,
@@ -152511,12 +152513,28 @@ var ToolServerService = class ToolServerService2 extends Service {
       throw error48;
     }
     this.logger.debug("Connected to MCP server");
-    const originalTools = await this.client.listTools();
-    this.tools.next(originalTools.tools);
+    const emitTools = (tools) => {
+      this.pendingTools = tools;
+      if (this.debounceTimer)
+        clearTimeout(this.debounceTimer);
+      this.debounceTimer = setTimeout(() => {
+        if (this.pendingTools) {
+          this.tools.next(this.pendingTools);
+          this.pendingTools = null;
+        }
+        this.debounceTimer = null;
+      }, 200);
+    };
     this.client.setNotificationHandler(ToolListChangedNotificationSchema, async () => {
-      const updatedTools = await this.client.listTools();
-      this.tools.next(updatedTools.tools);
+      try {
+        const updatedTools = await this.client.listTools();
+        emitTools(updatedTools.tools);
+      } catch (error48) {
+        this.logger.error({ err: error48 }, "Failed to refresh tool list after ToolListChanged notification");
+      }
     });
+    const originalTools = await this.client.listTools();
+    emitTools(originalTools.tools);
   }
   /**
    * Extract the most relevant error line from stderr output.
@@ -152536,6 +152554,14 @@ var ToolServerService = class ToolServerService2 extends Service {
   }
   async shutdown() {
     this.logger.info("Stopping");
+    if (this.debounceTimer) {
+      clearTimeout(this.debounceTimer);
+      this.debounceTimer = null;
+      if (this.pendingTools) {
+        this.tools.next(this.pendingTools);
+        this.pendingTools = null;
+      }
+    }
     this.tools.complete();
     await this.onShutdownCallback();
     if (this.transport instanceof StdioClientTransport) {