@tyvm/knowhow 0.0.108 → 0.0.109-dev.2b94ba2

package/ts_build/src/workers/auth/WsMiddleware.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WsMiddleware.js","sourceRoot":"","sources":["../../../../src/workers/auth/WsMiddleware.ts"],"names":[],"mappings":";;;;;;AACA,oDAAkC;AAgBlC,MAAa,iBAAiB;IACpB,GAAG,GAAqB,EAAE,CAAC;IAEnC,GAAG,CAAC,EAAkB;QACpB,IAAI,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QAClB,OAAO,IAAI,CAAC;IACd,CAAC;IAOD,MAAM,CAAC,EAAa,EAAE,SAA0C;QAC9D,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,IAAqB,EAAE,EAAE;YAC/C,IAAI,CAAC,GAAG,CAAC,CAAC;YACV,MAAM,IAAI,GAAG,KAAK,EAAE,GAAW,EAAiB,EAAE;gBAChD,IAAI,GAAG,EAAE,CAAC;oBACR,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;oBAC9D,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;oBAC5B,OAAO;gBACT,CAAC;gBACD,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC;gBACzB,IAAI,EAAE,EAAE,CAAC;oBACP,MAAM,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC3B,CAAC;qBAAM,CAAC;oBACN,SAAS,CAAC,IAAI,CAAC,CAAC;gBAClB,CAAC;YACH,CAAC,CAAC;YACF,MAAM,IAAI,EAAE,CAAC;QACf,CAAC,CAAC,CAAC;IACL,CAAC;IAeD,UAAU,CAAC,EAAa;QACtB,MAAM,YAAY,GAAG,IAAI,gBAAY,EAAE,CAAC;QAGxC,EAAE,CAAC,EAAE,CAAC,SAAS,EAAE,KAAK,EAAE,IAAqB,EAAE,EAAE;YAC/C,IAAI,CAAC,GAAG,CAAC,CAAC;YACV,MAAM,IAAI,GAAG,KAAK,EAAE,GAAW,EAAiB,EAAE;gBAChD,IAAI,GAAG,EAAE,CAAC;oBACR,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;oBAC9D,EAAE,CAAC,KAAK,CAAC,IAAI,EAAE,GAAG,CAAC,OAAO,CAAC,CAAC;oBAC5B,OAAO;gBACT,CAAC;gBACD,MAAM,EAAE,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,CAAC;gBACzB,IAAI,EAAE,EAAE,CAAC;oBACP,MAAM,EAAE,CAAC,EAAE,EAAE,IAAI,EAAE,IAAI,CAAC,CAAC;gBAC3B,CAAC;qBAAM,CAAC;oBAEN,YAAY,CAAC,IAAI,CAAC,SAAS,EAAE,IAAI,CAAC,CAAC;gBACrC,CAAC;YACH,CAAC,CAAC;YACF,MAAM,IAAI,EAAE,CAAC;QACf,CAAC,CAAC,CAAC;QAKH,MAAM,UAAU,GAAG,EAAE,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC;QACjC,EAAU,CAAC,EAAE,GAAG,CAAC,KAAa,EAAE,QAAkC,EAAE,EAAE;YACrE,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;gBACxB,YAAY,CAAC,EAAE,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;gBACrC,OAAO,EAAE,CAAC;YACZ,CAAC;YACD,OAAO,UAAU,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QACrC,CAAC,CAAC;IACJ,CAAC;CACF;AAjFD,8CAiFC"}

package/ts_build/src/workers/auth/authMiddleware.d.ts

@@ -0,0 +1,3 @@
+import { WorkerPasskeyAuthService } from "./WorkerPasskeyAuth";
+import { WsMiddlewareFn } from "./WsMiddleware";
+export declare function makeAuthMiddleware(authService: WorkerPasskeyAuthService): WsMiddlewareFn;

package/ts_build/src/workers/auth/authMiddleware.js

@@ -0,0 +1,60 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.makeAuthMiddleware = makeAuthMiddleware;
+function makeAuthMiddleware(authService) {
+    return async (ws, data, next) => {
+        if (!authService.isLocked()) {
+            return next();
+        }
+        let parsed;
+        try {
+            const raw = typeof data === "string" ? data : data.toString("utf-8");
+            parsed = JSON.parse(raw);
+        }
+        catch {
+            ws.send(JSON.stringify({
+                type: "auth:locked",
+                message: "Worker is locked. Send { type: 'auth:getChallenge' } first.",
+            }));
+            return;
+        }
+        if (parsed.type === "auth:getChallenge") {
+            const challenge = authService.generateChallenge();
+            ws.send(JSON.stringify({
+                type: "auth:challenge",
+                challenge,
+                credentialId: authService.getCredentialId(),
+                timestamp: Math.floor(Date.now() / 1000),
+            }));
+            return;
+        }
+        if (parsed.type === "auth:response") {
+            const result = await authService.unlock({
+                signature: parsed.signature,
+                credentialId: parsed.credentialId,
+                authenticatorData: parsed.authenticatorData,
+                clientDataJSON: parsed.clientDataJSON,
+                challenge: parsed.challenge,
+            });
+            if (result.success) {
+                const info = authService.getSessionInfo();
+                ws.send(JSON.stringify({
+                    type: "auth:success",
+                    expiresAt: info.expiresAt,
+                }));
+            }
+            else {
+                ws.send(JSON.stringify({
+                    type: "auth:failure",
+                    reason: result.reason ?? "unknown",
+                }));
+            }
+            return;
+        }
+        ws.send(JSON.stringify({
+            type: "auth:locked",
+            message: "Worker is locked. Send { type: 'auth:getChallenge' } to authenticate.",
+        }));
+    };
+}
+//# sourceMappingURL=authMiddleware.js.map

package/ts_build/src/workers/auth/authMiddleware.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"authMiddleware.js","sourceRoot":"","sources":["../../../../src/workers/auth/authMiddleware.ts"],"names":[],"mappings":";;AAyBA,gDA8EC;AA9ED,SAAgB,kBAAkB,CAChC,WAAqC;IAErC,OAAO,KAAK,EAAE,EAAa,EAAE,IAAqB,EAAE,IAAI,EAAE,EAAE;QAE1D,IAAI,CAAC,WAAW,CAAC,QAAQ,EAAE,EAAE,CAAC;YAC5B,OAAO,IAAI,EAAE,CAAC;QAChB,CAAC;QAGD,IAAI,MAAW,CAAC;QAChB,IAAI,CAAC;YACH,MAAM,GAAG,GAAG,OAAO,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,CAAC;YACrE,MAAM,GAAG,IAAI,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;QAC3B,CAAC;QAAC,MAAM,CAAC;YACP,EAAE,CAAC,IAAI,CACL,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,aAAa;gBACnB,OAAO,EAAE,6DAA6D;aACvE,CAAC,CACH,CAAC;YACF,OAAO;QACT,CAAC;QAGD,IAAI,MAAM,CAAC,IAAI,KAAK,mBAAmB,EAAE,CAAC;YACxC,MAAM,SAAS,GAAG,WAAW,CAAC,iBAAiB,EAAE,CAAC;YAClD,EAAE,CAAC,IAAI,CACL,IAAI,CAAC,SAAS,CAAC;gBACb,IAAI,EAAE,gBAAgB;gBACtB,SAAS;gBACT,YAAY,EAAE,WAAW,CAAC,eAAe,EAAE;gBAC3C,SAAS,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,IAAI,CAAC;aACzC,CAAC,CACH,CAAC;YACF,OAAO;QACT,CAAC;QAGD,IAAI,MAAM,CAAC,IAAI,KAAK,eAAe,EAAE,CAAC;YACpC,MAAM,MAAM,GAAG,MAAM,WAAW,CAAC,MAAM,CAAC;gBACtC,SAAS,EAAE,MAAM,CAAC,SAAS;gBAC3B,YAAY,EAAE,MAAM,CAAC,YAAY;gBACjC,iBAAiB,EAAE,MAAM,CAAC,iBAAiB;gBAC3C,cAAc,EAAE,MAAM,CAAC,cAAc;gBACrC,SAAS,EAAE,MAAM,CAAC,SAAS;aAC5B,CAAC,CAAC;YAEH,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,MAAM,IAAI,GAAG,WAAW,CAAC,cAAc,EAAE,CAAC;gBAC1C,EAAE,CAAC,IAAI,CACL,IAAI,CAAC,SAAS,CAAC;oBACb,IAAI,EAAE,cAAc;oBACpB,SAAS,EAAE,IAAI,CAAC,SAAS;iBAC1B,CAAC,CACH,CAAC;YACJ,CAAC;iBAAM,CAAC;gBACN,EAAE,CAAC,IAAI,CACL,IAAI,CAAC,SAAS,CAAC;oBACb,IAAI,EAAE,cAAc;oBACpB,MAAM,EAAE,MAAM,CAAC,MAAM,IAAI,SAAS;iBACnC,CAAC,CACH,CAAC;YACJ,CAAC;YAED,OAAO;QACT,CAAC;QAGD,EAAE,CAAC,IAAI,CACL,IAAI,CAAC,SAAS,CAAC;YACb,IAAI,EAAE,aAAa;YACnB,OAAO,EACL,uEAAuE;SAC1E,CAAC,CACH,CAAC;IAEJ,CAAC,CAAC;AACJ,CAAC"}

package/ts_build/src/workers/auth/types.d.ts

@@ -1,3 +1,10 @@
+export interface AuthGetChallengeMessage {
+    type: "auth:getChallenge";
+}
+export interface AuthLockedMessage {
+    type: "auth:locked";
+    message: string;
+}
 export interface AuthChallengeMessage {
     type: "auth:challenge";
     challenge: string;
@@ -20,7 +27,7 @@ export interface AuthFailureMessage {
     type: "auth:failure";
     reason: "invalid_signature" | "expired" | "unknown_credential";
 }
-export type AuthMessage = AuthChallengeMessage | AuthResponseMessage | AuthSuccessMessage | AuthFailureMessage;
+export type AuthMessage = AuthGetChallengeMessage | AuthChallengeMessage | AuthResponseMessage | AuthSuccessMessage | AuthFailureMessage | AuthLockedMessage;
 export interface PasskeyCredential {
     publicKey: string;
     credentialId: string;

package/ts_build/src/workers/tools/index.d.ts

@@ -2,10 +2,12 @@ export * from "./listAllowedPorts";
 export * from "./getChallenge";
 export * from "./unlock";
 export * from "./lock";
+export * from "./reloadConfig";
 import { listAllowedPorts } from "./listAllowedPorts";
 export { makeGetChallengeTool } from "./getChallenge";
 export { makeUnlockTool } from "./unlock";
 export { makeLockTool } from "./lock";
+export { makeReloadConfigTool } from "./reloadConfig";
 declare const _default: {
     tools: {
         listAllowedPorts: typeof listAllowedPorts;

package/ts_build/src/workers/tools/index.js

@@ -14,11 +14,12 @@ var __exportStar = (this && this.__exportStar) || function(m, exports) {
     for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.makeLockTool = exports.makeUnlockTool = exports.makeGetChallengeTool = void 0;
+exports.makeReloadConfigTool = exports.makeLockTool = exports.makeUnlockTool = exports.makeGetChallengeTool = void 0;
 __exportStar(require("./listAllowedPorts"), exports);
 __exportStar(require("./getChallenge"), exports);
 __exportStar(require("./unlock"), exports);
 __exportStar(require("./lock"), exports);
+__exportStar(require("./reloadConfig"), exports);
 const listAllowedPorts_1 = require("./listAllowedPorts");
 var getChallenge_1 = require("./getChallenge");
 Object.defineProperty(exports, "makeGetChallengeTool", { enumerable: true, get: function () { return getChallenge_1.makeGetChallengeTool; } });
@@ -26,6 +27,8 @@ var unlock_1 = require("./unlock");
 Object.defineProperty(exports, "makeUnlockTool", { enumerable: true, get: function () { return unlock_1.makeUnlockTool; } });
 var lock_1 = require("./lock");
 Object.defineProperty(exports, "makeLockTool", { enumerable: true, get: function () { return lock_1.makeLockTool; } });
+var reloadConfig_1 = require("./reloadConfig");
+Object.defineProperty(exports, "makeReloadConfigTool", { enumerable: true, get: function () { return reloadConfig_1.makeReloadConfigTool; } });
 exports.default = {
     tools: { listAllowedPorts: listAllowedPorts_1.listAllowedPorts },
     definitions: [listAllowedPorts_1.listAllowedPortsDefinition],

package/ts_build/src/workers/tools/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/workers/tools/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,qDAAmC;AACnC,iDAA+B;AAC/B,2CAAyB;AACzB,yCAAuB;AAEvB,yDAG4B;AAE5B,+CAAsD;AAA7C,oHAAA,oBAAoB,OAAA;AAC7B,mCAA0C;AAAjC,wGAAA,cAAc,OAAA;AACvB,+BAAsC;AAA7B,oGAAA,YAAY,OAAA;AAErB,kBAAe;IACb,KAAK,EAAE,EAAE,gBAAgB,EAAhB,mCAAgB,EAAE;IAC3B,WAAW,EAAE,CAAC,6CAA0B,CAAC;CAC1C,CAAC"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../../../src/workers/tools/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;;AAAA,qDAAmC;AACnC,iDAA+B;AAC/B,2CAAyB;AACzB,yCAAuB;AACvB,iDAA+B;AAE/B,yDAG4B;AAE5B,+CAAsD;AAA7C,oHAAA,oBAAoB,OAAA;AAC7B,mCAA0C;AAAjC,wGAAA,cAAc,OAAA;AACvB,+BAAsC;AAA7B,oGAAA,YAAY,OAAA;AACrB,+CAAsD;AAA7C,oHAAA,oBAAoB,OAAA;AAE7B,kBAAe;IACb,KAAK,EAAE,EAAE,gBAAgB,EAAhB,mCAAgB,EAAE;IAC3B,WAAW,EAAE,CAAC,6CAA0B,CAAC;CAC1C,CAAC"}

package/ts_build/src/workers/tools/reloadConfig.d.ts

@@ -0,0 +1,14 @@
+import { Tool } from "../../clients/types";
+import { McpServerService } from "../../services/McpServer";
+import { McpService } from "../../services/Mcp";
+import { ToolsService } from "../../services/Tools";
+export interface ReloadConfigResult {
+    success: boolean;
+    toolCount: number;
+    mcpCount: number;
+    message: string;
+}
+export declare function makeReloadConfigTool(Mcp: McpService, Tools: ToolsService, mcpServer: McpServerService, setToolsToUse: (tools: ReturnType<typeof Tools.getToolsByNames>) => void): {
+    reloadConfig: () => Promise<ReloadConfigResult>;
+    reloadConfigDefinition: Tool;
+};

package/ts_build/src/workers/tools/reloadConfig.js

@@ -0,0 +1,48 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.makeReloadConfigTool = makeReloadConfigTool;
+const config_1 = require("../../config");
+function makeReloadConfigTool(Mcp, Tools, mcpServer, setToolsToUse) {
+    const reloadConfig = async () => {
+        try {
+            const freshConfig = await (0, config_1.getConfig)();
+            await Mcp.closeAll();
+            await Mcp.connectToConfigured(Tools);
+            const allowedToolNames = freshConfig.worker?.allowedTools ?? Tools.getToolNames();
+            const newToolsToUse = Tools.getToolsByNames(allowedToolNames);
+            setToolsToUse(newToolsToUse);
+            mcpServer.withTools(newToolsToUse);
+            const mcpCount = freshConfig.mcps?.length ?? 0;
+            return {
+                success: true,
+                toolCount: newToolsToUse.length,
+                mcpCount,
+                message: `Config reloaded: ${newToolsToUse.length} tools active, ${mcpCount} MCP(s) configured`,
+            };
+        }
+        catch (err) {
+            const message = err instanceof Error ? err.message : String(err);
+            return {
+                success: false,
+                toolCount: 0,
+                mcpCount: 0,
+                message: `Failed to reload config: ${message}`,
+            };
+        }
+    };
+    const reloadConfigDefinition = {
+        type: "function",
+        function: {
+            name: "reloadConfig",
+            description: "Reload the worker config from disk, reconnect all MCPs, and rebuild the active tool list. " +
+                "Call this after running `knowhow cloudworker --pull <id>` to apply updated MCPs without restarting the worker.",
+            parameters: {
+                type: "object",
+                properties: {},
+                required: [],
+            },
+        },
+    };
+    return { reloadConfig, reloadConfigDefinition };
+}
+//# sourceMappingURL=reloadConfig.js.map

package/ts_build/src/workers/tools/reloadConfig.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"reloadConfig.js","sourceRoot":"","sources":["../../../../src/workers/tools/reloadConfig.ts"],"names":[],"mappings":";;AAsBA,oDA6DC;AAlFD,yCAAyC;AAqBzC,SAAgB,oBAAoB,CAClC,GAAe,EACf,KAAmB,EACnB,SAA2B,EAC3B,aAAwE;IAExE,MAAM,YAAY,GAAG,KAAK,IAAiC,EAAE;QAC3D,IAAI,CAAC;YAEH,MAAM,WAAW,GAAG,MAAM,IAAA,kBAAS,GAAE,CAAC;YAGtC,MAAM,GAAG,CAAC,QAAQ,EAAE,CAAC;YAGrB,MAAM,GAAG,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;YAGrC,MAAM,gBAAgB,GACpB,WAAW,CAAC,MAAM,EAAE,YAAY,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;YAC3D,MAAM,aAAa,GAAG,KAAK,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC;YAC9D,aAAa,CAAC,aAAa,CAAC,CAAC;YAG7B,SAAS,CAAC,SAAS,CAAC,aAAa,CAAC,CAAC;YAEnC,MAAM,QAAQ,GAAG,WAAW,CAAC,IAAI,EAAE,MAAM,IAAI,CAAC,CAAC;YAE/C,OAAO;gBACL,OAAO,EAAE,IAAI;gBACb,SAAS,EAAE,aAAa,CAAC,MAAM;gBAC/B,QAAQ;gBACR,OAAO,EAAE,oBAAoB,aAAa,CAAC,MAAM,kBAAkB,QAAQ,oBAAoB;aAChG,CAAC;QACJ,CAAC;QAAC,OAAO,GAAG,EAAE,CAAC;YACb,MAAM,OAAO,GAAG,GAAG,YAAY,KAAK,CAAC,CAAC,CAAC,GAAG,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC;YACjE,OAAO;gBACL,OAAO,EAAE,KAAK;gBACd,SAAS,EAAE,CAAC;gBACZ,QAAQ,EAAE,CAAC;gBACX,OAAO,EAAE,4BAA4B,OAAO,EAAE;aAC/C,CAAC;QACJ,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,sBAAsB,GAAS;QACnC,IAAI,EAAE,UAAmB;QACzB,QAAQ,EAAE;YACR,IAAI,EAAE,cAAc;YACpB,WAAW,EACT,4FAA4F;gBAC5F,gHAAgH;YAClH,UAAU,EAAE;gBACV,IAAI,EAAE,QAAQ;gBACd,UAAU,EAAE,EAAE;gBACd,QAAQ,EAAE,EAAE;aACb;SACF;KACF,CAAC;IAEF,OAAO,EAAE,YAAY,EAAE,sBAAsB,EAAE,CAAC;AAClD,CAAC"}

package/ts_build/tests/services/WorkerReloadConfig.test.d.ts

@@ -0,0 +1 @@
+export {};

package/ts_build/tests/services/WorkerReloadConfig.test.js

@@ -0,0 +1,86 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+const Mcp_1 = require("../../src/services/Mcp");
+const Tools_1 = require("../../src/services/Tools");
+async function simulateReloadConfig(Mcp, Tools, mcpServer, getConfig, toolsToUseRef) {
+    const freshConfig = await getConfig();
+    await Mcp.closeAll();
+    await Mcp.connectToConfigured(Tools);
+    const allowedToolNames = freshConfig.worker?.allowedTools ?? Tools.getToolNames();
+    toolsToUseRef.value = Tools.getToolsByNames(allowedToolNames);
+    mcpServer.withTools(toolsToUseRef.value);
+}
+describe("Worker reloadConfig handler", () => {
+    let Mcp;
+    let Tools;
+    let mcpServer;
+    let toolsToUseRef;
+    beforeEach(() => {
+        Mcp = new Mcp_1.McpService();
+        Tools = new Tools_1.ToolsService();
+        mcpServer = { withTools: jest.fn() };
+        toolsToUseRef = { value: [] };
+        jest.spyOn(Mcp, "closeAll").mockResolvedValue(undefined);
+        jest.spyOn(Mcp, "connectToConfigured").mockResolvedValue(undefined);
+    });
+    afterEach(() => {
+        jest.restoreAllMocks();
+    });
+    it("should call closeAll() to tear down existing MCP connections", async () => {
+        const getConfig = jest
+            .fn()
+            .mockResolvedValue({ worker: { allowedTools: [] } });
+        await simulateReloadConfig(Mcp, Tools, mcpServer, getConfig, toolsToUseRef);
+        expect(Mcp.closeAll).toHaveBeenCalledTimes(1);
+    });
+    it("should call connectToConfigured() to reconnect MCPs from fresh config", async () => {
+        const getConfig = jest
+            .fn()
+            .mockResolvedValue({ worker: { allowedTools: [] } });
+        await simulateReloadConfig(Mcp, Tools, mcpServer, getConfig, toolsToUseRef);
+        expect(Mcp.connectToConfigured).toHaveBeenCalledWith(Tools);
+    });
+    it("should rebuild the tool list from allowedTools in fresh config", async () => {
+        const toolsByNamesSpy = jest
+            .spyOn(Tools, "getToolsByNames")
+            .mockReturnValue([{ function: { name: "execCommand" } }]);
+        const getConfig = jest
+            .fn()
+            .mockResolvedValue({ worker: { allowedTools: ["execCommand"] } });
+        await simulateReloadConfig(Mcp, Tools, mcpServer, getConfig, toolsToUseRef);
+        expect(toolsByNamesSpy).toHaveBeenCalledWith(["execCommand"]);
+        expect(toolsToUseRef.value).toHaveLength(1);
+    });
+    it("should fall back to all tool names when allowedTools is not set", async () => {
+        const allNames = ["execCommand", "readFile", "writeFileChunk"];
+        jest.spyOn(Tools, "getToolNames").mockReturnValue(allNames);
+        const toolsByNamesSpy = jest
+            .spyOn(Tools, "getToolsByNames")
+            .mockReturnValue([]);
+        const getConfig = jest.fn().mockResolvedValue({});
+        await simulateReloadConfig(Mcp, Tools, mcpServer, getConfig, toolsToUseRef);
+        expect(toolsByNamesSpy).toHaveBeenCalledWith(allNames);
+    });
+    it("should call mcpServer.withTools() with the rebuilt tool list", async () => {
+        const fakeTools = [{ function: { name: "readFile" } }];
+        jest.spyOn(Tools, "getToolsByNames").mockReturnValue(fakeTools);
+        const getConfig = jest
+            .fn()
+            .mockResolvedValue({ worker: { allowedTools: ["readFile"] } });
+        await simulateReloadConfig(Mcp, Tools, mcpServer, getConfig, toolsToUseRef);
+        expect(mcpServer.withTools).toHaveBeenCalledWith(fakeTools);
+    });
+    it("should re-read the config on every reload (not use stale config)", async () => {
+        const getConfig = jest
+            .fn()
+            .mockResolvedValueOnce({ worker: { allowedTools: ["execCommand"] } })
+            .mockResolvedValueOnce({ worker: { allowedTools: ["readFile", "writeFileChunk"] } });
+        jest.spyOn(Tools, "getToolsByNames").mockReturnValue([]);
+        await simulateReloadConfig(Mcp, Tools, mcpServer, getConfig, toolsToUseRef);
+        await simulateReloadConfig(Mcp, Tools, mcpServer, getConfig, toolsToUseRef);
+        expect(getConfig).toHaveBeenCalledTimes(2);
+        expect(Mcp.closeAll).toHaveBeenCalledTimes(2);
+        expect(Mcp.connectToConfigured).toHaveBeenCalledTimes(2);
+    });
+});
+//# sourceMappingURL=WorkerReloadConfig.test.js.map

package/ts_build/tests/services/WorkerReloadConfig.test.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"WorkerReloadConfig.test.js","sourceRoot":"","sources":["../../../tests/services/WorkerReloadConfig.test.ts"],"names":[],"mappings":";;AAQA,gDAAoD;AACpD,oDAAwD;AAOxD,KAAK,UAAU,oBAAoB,CACjC,GAAe,EACf,KAAmB,EACnB,SAAoD,EACpD,SAAkE,EAClE,aAAmC;IAGnC,MAAM,WAAW,GAAG,MAAM,SAAS,EAAE,CAAC;IACtC,MAAM,GAAG,CAAC,QAAQ,EAAE,CAAC;IACrB,MAAM,GAAG,CAAC,mBAAmB,CAAC,KAAK,CAAC,CAAC;IACrC,MAAM,gBAAgB,GACpB,WAAW,CAAC,MAAM,EAAE,YAAY,IAAI,KAAK,CAAC,YAAY,EAAE,CAAC;IAC3D,aAAa,CAAC,KAAK,GAAG,KAAK,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC;IAC9D,SAAS,CAAC,SAAS,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC;AAC3C,CAAC;AAMD,QAAQ,CAAC,6BAA6B,EAAE,GAAG,EAAE;IAC3C,IAAI,GAAe,CAAC;IACpB,IAAI,KAAmB,CAAC;IACxB,IAAI,SAAmC,CAAC;IACxC,IAAI,aAAmC,CAAC;IAExC,UAAU,CAAC,GAAG,EAAE;QACd,GAAG,GAAG,IAAI,gBAAU,EAAE,CAAC;QACvB,KAAK,GAAG,IAAI,oBAAY,EAAE,CAAC;QAC3B,SAAS,GAAG,EAAE,SAAS,EAAE,IAAI,CAAC,EAAE,EAAE,EAAE,CAAC;QACrC,aAAa,GAAG,EAAE,KAAK,EAAE,EAAE,EAAE,CAAC;QAG9B,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,UAAU,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;QACzD,IAAI,CAAC,KAAK,CAAC,GAAG,EAAE,qBAAqB,CAAC,CAAC,iBAAiB,CAAC,SAAS,CAAC,CAAC;IACtE,CAAC,CAAC,CAAC;IAEH,SAAS,CAAC,GAAG,EAAE;QACb,IAAI,CAAC,eAAe,EAAE,CAAC;IACzB,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,SAAS,GAAG,IAAI;aACnB,EAAE,EAAE;aACJ,iBAAiB,CAAC,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QAEvD,MAAM,oBAAoB,CAAC,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QAE5E,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAChD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,uEAAuE,EAAE,KAAK,IAAI,EAAE;QACrF,MAAM,SAAS,GAAG,IAAI;aACnB,EAAE,EAAE;aACJ,iBAAiB,CAAC,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,EAAE,EAAE,EAAE,CAAC,CAAC;QAEvD,MAAM,oBAAoB,CAAC,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QAE5E,MAAM,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC,oBAAoB,CAAC,KAAK,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,gEAAgE,EAAE,KAAK,IAAI,EAAE;QAE9E,MAAM,eAAe,GAAG,IAAI;aACzB,KAAK,CAAC,KAAK,EAAE,iBAAiB,CAAC;aAC/B,eAAe,CAAC,CAAC,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,CAAgD,CAAC,CAAC;QAE3G,MAAM,SAAS,GAAG,IAAI;aACnB,EAAE,EAAE;aACJ,iBAAiB,CAAC,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE,CAAC,CAAC;QAEpE,MAAM,oBAAoB,CAAC,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QAE5E,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,CAAC,aAAa,CAAC,CAAC,CAAC;QAC9D,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;IAC9C,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,iEAAiE,EAAE,KAAK,IAAI,EAAE;QAC/E,MAAM,QAAQ,GAAG,CAAC,aAAa,EAAE,UAAU,EAAE,gBAAgB,CAAC,CAAC;QAC/D,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,cAAc,CAAC,CAAC,eAAe,CAAC,QAAQ,CAAC,CAAC;QAC5D,MAAM,eAAe,GAAG,IAAI;aACzB,KAAK,CAAC,KAAK,EAAE,iBAAiB,CAAC;aAC/B,eAAe,CAAC,EAAiD,CAAC,CAAC;QAGtE,MAAM,SAAS,GAAG,IAAI,CAAC,EAAE,EAAE,CAAC,iBAAiB,CAAC,EAAE,CAAC,CAAC;QAElD,MAAM,oBAAoB,CAAC,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QAE5E,MAAM,CAAC,eAAe,CAAC,CAAC,oBAAoB,CAAC,QAAQ,CAAC,CAAC;IACzD,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,8DAA8D,EAAE,KAAK,IAAI,EAAE;QAC5E,MAAM,SAAS,GAAG,CAAC,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,CAAgD,CAAC;QACtG,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,iBAAiB,CAAC,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;QAEhE,MAAM,SAAS,GAAG,IAAI;aACnB,EAAE,EAAE;aACJ,iBAAiB,CAAC,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,CAAC,UAAU,CAAC,EAAE,EAAE,CAAC,CAAC;QAEjE,MAAM,oBAAoB,CAAC,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QAE5E,MAAM,CAAC,SAAS,CAAC,SAAS,CAAC,CAAC,oBAAoB,CAAC,SAAS,CAAC,CAAC;IAC9D,CAAC,CAAC,CAAC;IAEH,EAAE,CAAC,kEAAkE,EAAE,KAAK,IAAI,EAAE;QAChF,MAAM,SAAS,GAAG,IAAI;aACnB,EAAE,EAAE;aACJ,qBAAqB,CAAC,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,CAAC,aAAa,CAAC,EAAE,EAAE,CAAC;aACpE,qBAAqB,CAAC,EAAE,MAAM,EAAE,EAAE,YAAY,EAAE,CAAC,UAAU,EAAE,gBAAgB,CAAC,EAAE,EAAE,CAAC,CAAC;QAEvF,IAAI,CAAC,KAAK,CAAC,KAAK,EAAE,iBAAiB,CAAC,CAAC,eAAe,CAAC,EAAE,CAAC,CAAC;QAGzD,MAAM,oBAAoB,CAAC,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QAE5E,MAAM,oBAAoB,CAAC,GAAG,EAAE,KAAK,EAAE,SAAS,EAAE,SAAS,EAAE,aAAa,CAAC,CAAC;QAE5E,MAAM,CAAC,SAAS,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAE3C,MAAM,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;QAC9C,MAAM,CAAC,GAAG,CAAC,mBAAmB,CAAC,CAAC,qBAAqB,CAAC,CAAC,CAAC,CAAC;IAC3D,CAAC,CAAC,CAAC;AACL,CAAC,CAAC,CAAC"}

package/ts_build/tests/unit/clients/AIClient.test.d.ts

@@ -0,0 +1 @@
+export {};

package/ts_build/tests/unit/clients/AIClient.test.js

@@ -0,0 +1,339 @@
+"use strict";
+Object.defineProperty(exports, "__esModule", { value: true });
+jest.mock("../../../src/config", () => ({
+    getConfig: jest.fn().mockResolvedValue({ modules: [] }),
+    getGlobalConfig: jest.fn().mockResolvedValue({ modules: [] }),
+    getConfigSync: jest.fn().mockReturnValue({}),
+}));
+jest.mock("../../../src/services/KnowhowClient", () => ({
+    loadKnowhowJwt: jest.fn().mockReturnValue(null),
+    KNOWHOW_API_URL: "https://mock.local",
+}));
+const index_1 = require("../../../src/clients/index");
+const mockCompletion = () => ({
+    choices: [{ message: { role: "assistant", content: "hello" } }],
+    model: "mock-model",
+    usage: { prompt_tokens: 10, completion_tokens: 5 },
+});
+const mockImage = () => ({
+    created: Date.now(),
+    data: [{ url: "https://mock.local/image.png" }],
+});
+const mockEmbedding = () => ({
+    data: [{ object: "embedding", embedding: [0.1, 0.2], index: 0 }],
+    model: "mock-embed",
+    usage: { prompt_tokens: 5, total_tokens: 5 },
+});
+const mockAudio = () => ({
+    audio: Buffer.from("fake-audio"),
+    format: "mp3",
+});
+function setupClient(overrides = {}) {
+    const mockGenericClient = {
+        setKey: jest.fn(),
+        createChatCompletion: jest
+            .fn()
+            .mockResolvedValue(mockCompletion()),
+        createEmbedding: jest.fn().mockResolvedValue(mockEmbedding()),
+        createImageGeneration: jest.fn().mockResolvedValue(mockImage()),
+        createAudioGeneration: jest.fn().mockResolvedValue(mockAudio()),
+        createAudioTranscription: jest
+            .fn()
+            .mockResolvedValue({ text: "transcribed" }),
+        createVideoGeneration: jest.fn().mockResolvedValue({
+            created: Date.now(),
+            data: [{ url: "https://mock.local/video.mp4" }],
+        }),
+        getModels: jest.fn().mockResolvedValue([]),
+        ...overrides,
+    };
+    const aiClient = new index_1.AIClient();
+    aiClient.registerClient("mock", mockGenericClient);
+    aiClient.registerModels("mock", ["mock-model", "mock-embed"]);
+    return { aiClient, mockGenericClient };
+}
+describe("AIClient — retry / timeout / AbortSignal", () => {
+    afterEach(() => {
+        jest.useRealTimers();
+    });
+    describe("createCompletion", () => {
+        it("returns a completion on success", async () => {
+            const { aiClient } = setupClient();
+            const result = await aiClient.createCompletion("mock", {
+                model: "mock-model",
+                messages: [{ role: "user", content: "hi" }],
+            });
+            expect(result.choices[0].message.content).toBe("hello");
+        });
+        it("forwards the AbortSignal to createChatCompletion", async () => {
+            const { aiClient, mockGenericClient } = setupClient();
+            const controller = new AbortController();
+            await aiClient.createCompletion("mock", {
+                model: "mock-model",
+                messages: [],
+                signal: controller.signal,
+            });
+            const callArgs = mockGenericClient.createChatCompletion
+                .mock.calls[0][0];
+            expect(callArgs.signal).toBeInstanceOf(AbortSignal);
+        });
+        it("retries on 429 and succeeds", async () => {
+            jest.useFakeTimers();
+            const { aiClient, mockGenericClient } = setupClient({
+                createChatCompletion: jest
+                    .fn()
+                    .mockRejectedValueOnce(new Error("429 rate limited"))
+                    .mockResolvedValueOnce(mockCompletion()),
+            });
+            const promise = aiClient.createCompletion("mock", {
+                model: "mock-model",
+                messages: [],
+                maxRetries: 2,
+                backoffMs: 50,
+            });
+            await jest.runAllTimersAsync();
+            const result = await promise;
+            expect(result.choices[0].message.content).toBe("hello");
+            expect(mockGenericClient.createChatCompletion).toHaveBeenCalledTimes(2);
+        });
+        it("aborts immediately when external signal is pre-aborted", async () => {
+            const controller = new AbortController();
+            controller.abort();
+            const { aiClient, mockGenericClient } = setupClient();
+            await expect(aiClient.createCompletion("mock", {
+                model: "mock-model",
+                messages: [],
+                signal: controller.signal,
+            })).rejects.toMatchObject({ name: "AbortError" });
+            expect(mockGenericClient.createChatCompletion).not.toHaveBeenCalled();
+        });
+        it("cancels in-flight request when external signal is aborted", async () => {
+            const controller = new AbortController();
+            const { aiClient, mockGenericClient } = setupClient({
+                createChatCompletion: jest.fn().mockImplementation((opts) => {
+                    return new Promise((_, reject) => {
+                        opts.signal?.addEventListener("abort", () => reject(opts.signal.reason));
+                    });
+                }),
+            });
+            const promise = aiClient.createCompletion("mock", {
+                model: "mock-model",
+                messages: [],
+                signal: controller.signal,
+            });
+            setImmediate(() => controller.abort(new DOMException("User cancelled", "AbortError")));
+            await expect(promise).rejects.toMatchObject({ name: "AbortError" });
+            expect(mockGenericClient.createChatCompletion).toHaveBeenCalledTimes(1);
+        });
+        it("times out per-attempt and retries", async () => {
+            jest.useFakeTimers();
+            const { aiClient, mockGenericClient } = setupClient({
+                createChatCompletion: jest
+                    .fn()
+                    .mockImplementationOnce((opts) => {
+                    return new Promise((_, reject) => {
+                        opts.signal?.addEventListener("abort", () => reject(opts.signal.reason));
+                    });
+                })
+                    .mockResolvedValueOnce(mockCompletion()),
+            });
+            const promise = aiClient.createCompletion("mock", {
+                model: "mock-model",
+                messages: [],
+                timeout: 1000,
+                maxRetries: 2,
+                backoffMs: 10,
+            });
+            await jest.runAllTimersAsync();
+            const result = await promise;
+            expect(result.choices[0].message.content).toBe("hello");
+            expect(mockGenericClient.createChatCompletion).toHaveBeenCalledTimes(2);
+        });
+    });
+    describe("createEmbedding", () => {
+        it("forwards the AbortSignal to createEmbedding on the client", async () => {
+            const { aiClient, mockGenericClient } = setupClient();
+            const controller = new AbortController();
+            await aiClient.createEmbedding("mock", {
+                input: "test text",
+                model: "mock-embed",
+                signal: controller.signal,
+            });
+            const callArgs = mockGenericClient.createEmbedding.mock
+                .calls[0][0];
+            expect(callArgs.signal).toBeInstanceOf(AbortSignal);
+        });
+        it("retries on 500 and succeeds", async () => {
+            jest.useFakeTimers();
+            const { aiClient, mockGenericClient } = setupClient({
+                createEmbedding: jest
+                    .fn()
+                    .mockRejectedValueOnce(new Error("500 Internal Server Error"))
+                    .mockResolvedValueOnce(mockEmbedding()),
+            });
+            const promise = aiClient.createEmbedding("mock", {
+                input: "test",
+                model: "mock-embed",
+                maxRetries: 2,
+                backoffMs: 50,
+            });
+            await jest.runAllTimersAsync();
+            const result = await promise;
+            expect(result.data[0].embedding).toEqual([0.1, 0.2]);
+            expect(mockGenericClient.createEmbedding).toHaveBeenCalledTimes(2);
+        });
+    });
+    describe("createImageGeneration", () => {
+        it("forwards the AbortSignal to createImageGeneration on the client", async () => {
+            const { aiClient, mockGenericClient } = setupClient();
+            const controller = new AbortController();
+            await aiClient.createImageGeneration("mock", {
+                model: "mock-model",
+                prompt: "a cat",
+                signal: controller.signal,
+            });
+            const callArgs = mockGenericClient.createImageGeneration
+                .mock.calls[0][0];
+            expect(callArgs.signal).toBeInstanceOf(AbortSignal);
+        });
+        it("retries on 429 and succeeds", async () => {
+            jest.useFakeTimers();
+            const { aiClient, mockGenericClient } = setupClient({
+                createImageGeneration: jest
+                    .fn()
+                    .mockRejectedValueOnce(new Error("429 Too Many Requests"))
+                    .mockResolvedValueOnce(mockImage()),
+            });
+            const promise = aiClient.createImageGeneration("mock", {
+                model: "mock-model",
+                prompt: "a cat",
+                maxRetries: 2,
+                backoffMs: 50,
+            });
+            await jest.runAllTimersAsync();
+            const result = await promise;
+            expect(result.data[0].url).toBe("https://mock.local/image.png");
+            expect(mockGenericClient.createImageGeneration).toHaveBeenCalledTimes(2);
+        });
+        it("aborts when external signal fires mid-request", async () => {
+            const controller = new AbortController();
+            const { aiClient, mockGenericClient } = setupClient({
+                createImageGeneration: jest.fn().mockImplementation((opts) => {
+                    return new Promise((_, reject) => {
+                        opts.signal?.addEventListener("abort", () => reject(opts.signal.reason));
+                    });
+                }),
+            });
+            const promise = aiClient.createImageGeneration("mock", {
+                model: "mock-model",
+                prompt: "a cat",
+                signal: controller.signal,
+            });
+            setImmediate(() => controller.abort(new DOMException("User cancelled", "AbortError")));
+            await expect(promise).rejects.toMatchObject({ name: "AbortError" });
+            expect(mockGenericClient.createImageGeneration).toHaveBeenCalledTimes(1);
+        });
+    });
+    describe("createAudioGeneration", () => {
+        it("forwards the AbortSignal to createAudioGeneration on the client", async () => {
+            const { aiClient, mockGenericClient } = setupClient();
+            const controller = new AbortController();
+            await aiClient.createAudioGeneration("mock", {
+                model: "mock-model",
+                input: "Hello world",
+                voice: "alloy",
+                signal: controller.signal,
+            });
+            const callArgs = mockGenericClient.createAudioGeneration
+                .mock.calls[0][0];
+            expect(callArgs.signal).toBeInstanceOf(AbortSignal);
+        });
+        it("retries on ECONNRESET and succeeds", async () => {
+            jest.useFakeTimers();
+            const { aiClient, mockGenericClient } = setupClient({
+                createAudioGeneration: jest
+                    .fn()
+                    .mockRejectedValueOnce(new Error("ECONNRESET"))
+                    .mockResolvedValueOnce(mockAudio()),
+            });
+            const promise = aiClient.createAudioGeneration("mock", {
+                model: "mock-model",
+                input: "Hello",
+                voice: "alloy",
+                maxRetries: 2,
+                backoffMs: 50,
+            });
+            await jest.runAllTimersAsync();
+            const result = await promise;
+            expect(result.format).toBe("mp3");
+            expect(mockGenericClient.createAudioGeneration).toHaveBeenCalledTimes(2);
+        });
+    });
+    describe("createVideoGeneration", () => {
+        it("forwards the AbortSignal to createVideoGeneration on the client", async () => {
+            const { aiClient, mockGenericClient } = setupClient();
+            const controller = new AbortController();
+            await aiClient.createVideoGeneration("mock", {
+                model: "mock-model",
+                prompt: "a sunset",
+                signal: controller.signal,
+            });
+            const callArgs = mockGenericClient.createVideoGeneration
+                .mock.calls[0][0];
+            expect(callArgs.signal).toBeInstanceOf(AbortSignal);
+        });
+        it("retries on 503 and succeeds", async () => {
+            jest.useFakeTimers();
+            const { aiClient, mockGenericClient } = setupClient({
+                createVideoGeneration: jest
+                    .fn()
+                    .mockRejectedValueOnce(new Error("503 Service Unavailable"))
+                    .mockResolvedValueOnce({
+                    created: Date.now(),
+                    data: [{ url: "https://mock.local/video.mp4" }],
+                }),
+            });
+            const promise = aiClient.createVideoGeneration("mock", {
+                model: "mock-model",
+                prompt: "a sunset",
+                maxRetries: 2,
+                backoffMs: 50,
+            });
+            await jest.runAllTimersAsync();
+            const result = await promise;
+            expect(result.data[0].url).toBe("https://mock.local/video.mp4");
+            expect(mockGenericClient.createVideoGeneration).toHaveBeenCalledTimes(2);
+        });
+    });
+    describe("createAudioTranscription", () => {
+        it("forwards the AbortSignal to createAudioTranscription on the client", async () => {
+            const { aiClient, mockGenericClient } = setupClient();
+            const controller = new AbortController();
+            await aiClient.createAudioTranscription("mock", {
+                file: Buffer.from("fake-audio"),
+                signal: controller.signal,
+            });
+            const callArgs = mockGenericClient.createAudioTranscription.mock.calls[0][0];
+            expect(callArgs.signal).toBeInstanceOf(AbortSignal);
+        });
+        it("retries on timeout error and succeeds", async () => {
+            jest.useFakeTimers();
+            const { aiClient, mockGenericClient } = setupClient({
+                createAudioTranscription: jest
+                    .fn()
+                    .mockRejectedValueOnce(new Error("timeout"))
+                    .mockResolvedValueOnce({ text: "hello world" }),
+            });
+            const promise = aiClient.createAudioTranscription("mock", {
+                file: Buffer.from("fake-audio"),
+                maxRetries: 2,
+                backoffMs: 50,
+            });
+            await jest.runAllTimersAsync();
+            const result = await promise;
+            expect(result.text).toBe("hello world");
+            expect(mockGenericClient.createAudioTranscription).toHaveBeenCalledTimes(2);
+        });
+    });
+});
+//# sourceMappingURL=AIClient.test.js.map