Revert "fix: comment out skyfire input schema modification (#296)"

MQ37 · MQ37 · commit e80f900ad9cc · 2025-09-30T14:12:33.000+02:00
This reverts commit 86077f0.
diff --git a/src/mcp/server.ts b/src/mcp/server.ts
@@ -26,8 +26,10 @@ import log from '@apify/log';
 
 import { ApifyClient } from '../apify-client.js';
 import {
+    HelperTools,
     SERVER_NAME,
     SERVER_VERSION,
+    SKYFIRE_PAY_ID_PROPERTY_DESCRIPTION,
     SKYFIRE_README_CONTENT,
     SKYFIRE_TOOL_INSTRUCTIONS,
 } from '../const.js';
@@ -261,28 +263,28 @@ export class ActorsMcpServer {
             this.tools.set(wrap.tool.name, wrap);
         }
         // Handle Skyfire mode modifications once per tool upsert
-        // if (this.options.skyfireMode) {
-        //     for (const wrap of tools) {
-        //         if (wrap.type === 'actor'
-        //             || (wrap.type === 'internal' && wrap.tool.name === HelperTools.ACTOR_CALL)
-        //             || (wrap.type === 'internal' && wrap.tool.name === HelperTools.ACTOR_OUTPUT_GET)) {
-        //             // Add Skyfire instructions to description if not already present
-        //             if (!wrap.tool.description.includes(SKYFIRE_TOOL_INSTRUCTIONS)) {
-        //                 wrap.tool.description += `\n\n${SKYFIRE_TOOL_INSTRUCTIONS}`;
-        //             }
-        //             // Add skyfire-pay-id property if not present
-        //             if (wrap.tool.inputSchema && 'properties' in wrap.tool.inputSchema) {
-        //                 const props = wrap.tool.inputSchema.properties as Record<string, unknown>;
-        //                 if (!props['skyfire-pay-id']) {
-        //                     props['skyfire-pay-id'] = {
-        //                         type: 'string',
-        //                         description: SKYFIRE_PAY_ID_PROPERTY_DESCRIPTION,
-        //                     };
-        //                 }
-        //             }
-        //         }
-        //     }
-        // }
+        if (this.options.skyfireMode) {
+            for (const wrap of tools) {
+                if (wrap.type === 'actor'
+                    || (wrap.type === 'internal' && wrap.tool.name === HelperTools.ACTOR_CALL)
+                    || (wrap.type === 'internal' && wrap.tool.name === HelperTools.ACTOR_OUTPUT_GET)) {
+                    // Add Skyfire instructions to description if not already present
+                    if (!wrap.tool.description.includes(SKYFIRE_TOOL_INSTRUCTIONS)) {
+                        wrap.tool.description += `\n\n${SKYFIRE_TOOL_INSTRUCTIONS}`;
+                    }
+                    // Add skyfire-pay-id property if not present
+                    if (wrap.tool.inputSchema && 'properties' in wrap.tool.inputSchema) {
+                        const props = wrap.tool.inputSchema.properties as Record<string, unknown>;
+                        if (!props['skyfire-pay-id']) {
+                            props['skyfire-pay-id'] = {
+                                type: 'string',
+                                description: SKYFIRE_PAY_ID_PROPERTY_DESCRIPTION,
+                            };
+                        }
+                    }
+                }
+            }
+        }
         if (shouldNotifyToolsChangedHandler) this.notifyToolsChangedHandler();
         return tools;
     }
