@workglow/ai 0.2.6 → 0.2.8

package/dist/task/ToolCallingTask.d.ts

@@ -4,9 +4,10 @@
  * SPDX-License-Identifier: Apache-2.0
  */
 import { CreateWorkflow } from "@workglow/task-graph";
-import type { TaskConfig } from "@workglow/task-graph";
-import { DataPortSchema, FromSchema } from "@workglow/util/schema";
+import type { IExecuteContext, StreamEvent, TaskConfig } from "@workglow/task-graph";
 import { ServiceRegistry } from "@workglow/util";
+import { DataPortSchema, FromSchema } from "@workglow/util/schema";
+import type { AiJobInput } from "../job/AiJob";
 import { StreamingAiTask } from "./base/StreamingAiTask";
 import type { ToolDefinition } from "./ToolCallingUtils";
 export interface ToolDefinitionWithTaskType extends ToolDefinition {
@@ -109,6 +110,14 @@ export declare const ToolCallingInputSchema: {
                                 readonly format: "credential";
                                 readonly "x-ui-hidden": true;
                             };
+                            readonly native_dimensions: {
+                                readonly type: "integer";
+                                readonly description: "Native output vector dimensions for embedding models";
+                            };
+                            readonly mrl: {
+                                readonly type: "boolean";
+                                readonly description: "Whether the model supports Matryoshka Representation Learning";
+                            };
                         };
                         readonly additionalProperties: true;
                         readonly default: {};
@@ -316,6 +325,7 @@ export type ToolCallingTaskInput = Omit<FromSchema<typeof ToolCallingInputSchema
         readonly role: "user" | "assistant" | "tool";
         readonly content: unknown;
     }>;
+    readonly sessionId?: string;
 };
 export type ToolCallingTaskOutput = FromSchema<typeof ToolCallingOutputSchema>;
 export type ToolCallingTaskConfig = TaskConfig<ToolCallingTaskInput>;
@@ -326,6 +336,17 @@ export declare class ToolCallingTask extends StreamingAiTask<ToolCallingTaskInpu
     static description: string;
     static inputSchema(): DataPortSchema;
     static outputSchema(): DataPortSchema;
+    /** Session ID computed during getJobInput, used to register cleanup. */
+    private _computedSessionId;
+    /**
+     * Override to auto-compute a prefix-rewind session ID from tools + systemPrompt
+     * + runnerId when no explicit sessionId is provided. The runnerId scopes the
+     * cache to the current graph run so it's cleaned up via ResourceScope.
+     */
+    protected getJobInput(input: ToolCallingTaskInput): Promise<AiJobInput<ToolCallingTaskInput>>;
+    private registerSessionDispose;
+    execute(input: ToolCallingTaskInput, executeContext: IExecuteContext): Promise<ToolCallingTaskOutput | undefined>;
+    executeStream(input: ToolCallingTaskInput, context: IExecuteContext): AsyncIterable<StreamEvent<ToolCallingTaskOutput>>;
 }
 /**
  * Convenience function to run a tool calling task.

package/dist/task/ToolCallingTask.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ToolCallingTask.d.ts","sourceRoot":"","sources":["../../src/task/ToolCallingTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAiC,MAAM,sBAAsB,CAAC;AAErF,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAEjD,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAMzD,MAAM,WAAW,0BAA2B,SAAQ,cAAc;IAChE,6DAA6D;IAC7D,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;CAC3B;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,gBAAgB,CAC9B,SAAS,EAAE,aAAa,CAAC,MAAM,CAAC,EAChC,QAAQ,CAAC,EAAE,eAAe,GACzB,0BAA0B,EAAE,CAsB9B;AAMD,eAAO,MAAM,oBAAoB;aAC/B,IAAI,EAAE,QAAQ;aACd,UAAU;iBACR,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,MAAM;qBACb,WAAW,EAAE,eAAe;;iBAE9B,WAAW;qBACT,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,qCAAqC;;iBAEpD,WAAW;qBACT,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,cAAc;qBACrB,WAAW,EAAE,oDAAoD;qBACjE,oBAAoB;;iBAEtB,YAAY;qBACV,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,8CAA8C;qBAC3D,oBAAoB;;iBAEtB,YAAY;qBACV,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,+EAA+E;qBAC5F,oBAAoB;;iBAEtB,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,QAAQ;qBACf,WAAW,EAAE,0EAA0E;qBACvF,oBAAoB;;;aAGxB,QAAQ,YAAG,MAAM,EAAE,aAAa,EAAE,aAAa;aAC/C,oBAAoB;CACZ,CAAC;AA4BX,eAAO,MAAM,sBAAsB;mBAC3B,QAAQ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;+BAKA,QAAQ;gCAAS,QAAQ;sCAAe,iCAAiC;;+BAEzE,OAAO;gCACN,QAAQ;sCACF,qDAAqD;;;uCAGtD,QAAQ;;uCAER,QAAQ;;qCAEZ,IAAI;yCAAI,IAAI,EAAE,QAAQ;yCAAE,IAAI;;;;;;;;4BASjC,QAAQ;kCACF,iCAAiC;;;qBAG9C,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,4CAA4C;;;qBAGzD,IAAI,EAAE,OAAO;qBACb,KAAK,EAAE,UAAU;qBACjB,WAAW,EACT,oJAAoJ;qBACtJ,KAAK;yBACH,IAAI,EAAE,QAAQ;yBACd,UAAU;6BACR,IAAI;iCAAI,IAAI,EAAE,QAAQ;iCAAE,IAAI;;6BAC5B,OAAO;;yBAET,QAAQ;yBACR,oBAAoB;;;;2BAIhB,OAAO;6BACL,OAAO;4BACR,OAAO;kCACD,kDAAkD;;;mCAGnD,QAAQ;qCAAU,OAAO;0CAAe,gBAAgB;;mCA3HlE,QAAQ;;iCAEZ,IAAI;qCACF,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,MAAM;qCACb,WAAW,EAAE,eAAe;;iCAE9B,WAAW;qCACT,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,aAAa;qCACpB,WAAW,EAAE,qCAAqC;;iCAEpD,WAAW;qCACT,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,cAAc;qCACrB,WAAW,EAAE,oDAAoD;qCACjE,oBAAoB;;iCAEtB,YAAY;qCACV,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,eAAe;qCACtB,WAAW,EAAE,8CAA8C;qCAC3D,oBAAoB;;iCAEtB,YAAY;qCACV,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,eAAe;qCACtB,WAAW,EAAE,+EAA+E;qCAC5F,oBAAoB;;iCAEtB,MAAM;qCACJ,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,QAAQ;qCACf,WAAW,EAAE,0EAA0E;qCACvF,oBAAoB;;;iDAGb,MAAM,EAAE,aAAa,EAAE,aAAa;;;;;;qBA4F3C,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EACT,4HAA4H;qBAC9H,YAAY,EAAE,eAAe;;;qBAG7B,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,YAAY;qBACnB,WAAW,EAAE,0CAA0C;qBACvD,OAAO,EAAE,CAAC;qBACV,YAAY,EAAE,eAAe;;;qBAG7B,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,qCAAqC;qBAClD,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;qBACV,YAAY,EAAE,eAAe;;;;;CAKA,CAAC;AAEpC,eAAO,MAAM,uBAAuB;mBAC5B,QAAQ;;iBAEZ,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,MAAM;qBACb,WAAW,EAAE,yCAAyC;qBACtD,UAAU,EAAE,QAAQ;;iBAEtB,SAAS;qBACP,IAAI,EAAE,OAAO;qBACb,KAAK;+BA5HH,QAAQ;;6BAEZ,EAAE;iCACA,IAAI,EAAE,QAAQ;iCACd,KAAK,EAAE,IAAI;iCACX,WAAW,EAAE,sCAAsC;;6BAErD,IAAI;iCACF,IAAI,EAAE,QAAQ;iCACd,KAAK,EAAE,MAAM;iCACb,WAAW,EAAE,gCAAgC;;6BAE/C,KAAK;iCACH,IAAI,EAAE,QAAQ;iCACd,KAAK,EAAE,OAAO;iCACd,WAAW,EAAE,uCAAuC;iCACpD,oBAAoB;;;6CAGb,IAAI,EAAE,MAAM,EAAE,OAAO;;;qBA0G5B,KAAK,EAAE,YAAY;qBACnB,WAAW,EAAE,mCAAmC;qBAChD,UAAU,EAAE,QAAQ;;;;;CAKS,CAAC;AAEpC;;;;;;;;;;;GAWG;AACH,MAAM,MAAM,oBAAoB,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,sBAAsB,CAAC,EAAE,OAAO,CAAC,GAAG;IAC5F,QAAQ,CAAC,KAAK,EAAE,cAAc,EAAE,CAAC;IACjC,QAAQ,CAAC,QAAQ,CAAC,EAAE,aAAa,CAAC;QAChC,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;QAC7C,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;KAC3B,CAAC,CAAC;CACJ,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAC/E,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,oBAAoB,CAAC,CAAC;AAMrE,qBAAa,eAAgB,SAAQ,eAAe,CAClD,oBAAoB,EACpB,qBAAqB,EACrB,qBAAqB,CACtB;IACC,OAAuB,IAAI,SAAqB;IAChD,OAAuB,QAAQ,SAAmB;IAClD,OAAuB,KAAK,SAAkB;IAC9C,OAAuB,WAAW,SACyF;IAC3H,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;CACF;AAED;;GAEG;AACH,eAAO,MAAM,WAAW,UAAW,oBAAoB,WAAW,qBAAqB;;;;;;;;;EAEtF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CAAC,oBAAoB,EAAE,qBAAqB,EAAE,qBAAqB,CAAC,CAAC;KACjG;CACF"}
+{"version":3,"file":"ToolCallingTask.d.ts","sourceRoot":"","sources":["../../src/task/ToolCallingTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAiC,MAAM,sBAAsB,CAAC;AAErF,OAAO,KAAK,EAAE,eAAe,EAAE,WAAW,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACrF,OAAO,EAAmB,eAAe,EAAE,MAAM,gBAAgB,CAAC;AAClE,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,cAAc,CAAC;AAI/C,OAAO,EAAE,eAAe,EAAE,MAAM,wBAAwB,CAAC;AACzD,OAAO,KAAK,EAAE,cAAc,EAAE,MAAM,oBAAoB,CAAC;AAMzD,MAAM,WAAW,0BAA2B,SAAQ,cAAc;IAChE,6DAA6D;IAC7D,QAAQ,CAAC,QAAQ,EAAE,MAAM,CAAC;CAC3B;AAED;;;;;;;;;;;;GAYG;AACH,wBAAgB,gBAAgB,CAC9B,SAAS,EAAE,aAAa,CAAC,MAAM,CAAC,EAChC,QAAQ,CAAC,EAAE,eAAe,GACzB,0BAA0B,EAAE,CAsB9B;AAMD,eAAO,MAAM,oBAAoB;aAC/B,IAAI,EAAE,QAAQ;aACd,UAAU;iBACR,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,MAAM;qBACb,WAAW,EAAE,eAAe;;iBAE9B,WAAW;qBACT,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,qCAAqC;;iBAEpD,WAAW;qBACT,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,cAAc;qBACrB,WAAW,EAAE,oDAAoD;qBACjE,oBAAoB;;iBAEtB,YAAY;qBACV,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,8CAA8C;qBAC3D,oBAAoB;;iBAEtB,YAAY;qBACV,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,+EAA+E;qBAC5F,oBAAoB;;iBAEtB,MAAM;qBACJ,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,QAAQ;qBACf,WAAW,EAAE,0EAA0E;qBACvF,oBAAoB;;;aAGxB,QAAQ,YAAG,MAAM,EAAE,aAAa,EAAE,aAAa;aAC/C,oBAAoB;CACZ,CAAC;AA4BX,eAAO,MAAM,sBAAsB;mBAC3B,QAAQ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;+BAKA,QAAQ;gCAAS,QAAQ;sCAAe,iCAAiC;;+BAEzE,OAAO;gCACN,QAAQ;sCACF,qDAAqD;;;uCAGtD,QAAQ;;uCAER,QAAQ;;qCAEZ,IAAI;yCAAI,IAAI,EAAE,QAAQ;yCAAE,IAAI;;;;;;;;4BASjC,QAAQ;kCACF,iCAAiC;;;qBAG9C,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,eAAe;qBACtB,WAAW,EAAE,4CAA4C;;;qBAGzD,IAAI,EAAE,OAAO;qBACb,KAAK,EAAE,UAAU;qBACjB,WAAW,EACT,oJAAoJ;qBACtJ,KAAK;yBACH,IAAI,EAAE,QAAQ;yBACd,UAAU;6BACR,IAAI;iCAAI,IAAI,EAAE,QAAQ;iCAAE,IAAI;;6BAC5B,OAAO;;yBAET,QAAQ;yBACR,oBAAoB;;;;2BAIhB,OAAO;6BACL,OAAO;4BACR,OAAO;kCACD,kDAAkD;;;mCAGnD,QAAQ;qCAAU,OAAO;0CAAe,gBAAgB;;mCA3HlE,QAAQ;;iCAEZ,IAAI;qCACF,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,MAAM;qCACb,WAAW,EAAE,eAAe;;iCAE9B,WAAW;qCACT,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,aAAa;qCACpB,WAAW,EAAE,qCAAqC;;iCAEpD,WAAW;qCACT,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,cAAc;qCACrB,WAAW,EAAE,oDAAoD;qCACjE,oBAAoB;;iCAEtB,YAAY;qCACV,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,eAAe;qCACtB,WAAW,EAAE,8CAA8C;qCAC3D,oBAAoB;;iCAEtB,YAAY;qCACV,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,eAAe;qCACtB,WAAW,EAAE,+EAA+E;qCAC5F,oBAAoB;;iCAEtB,MAAM;qCACJ,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,QAAQ;qCACf,WAAW,EAAE,0EAA0E;qCACvF,oBAAoB;;;iDAGb,MAAM,EAAE,aAAa,EAAE,aAAa;;;;;;qBA4F3C,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EACT,4HAA4H;qBAC9H,YAAY,EAAE,eAAe;;;qBAG7B,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,YAAY;qBACnB,WAAW,EAAE,0CAA0C;qBACvD,OAAO,EAAE,CAAC;qBACV,YAAY,EAAE,eAAe;;;qBAG7B,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,aAAa;qBACpB,WAAW,EAAE,qCAAqC;qBAClD,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;qBACV,YAAY,EAAE,eAAe;;;;;CAKA,CAAC;AAEpC,eAAO,MAAM,uBAAuB;mBAC5B,QAAQ;;iBAEZ,IAAI;qBACF,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,MAAM;qBACb,WAAW,EAAE,yCAAyC;qBACtD,UAAU,EAAE,QAAQ;;iBAEtB,SAAS;qBACP,IAAI,EAAE,OAAO;qBACb,KAAK;+BA5HH,QAAQ;;6BAEZ,EAAE;iCACA,IAAI,EAAE,QAAQ;iCACd,KAAK,EAAE,IAAI;iCACX,WAAW,EAAE,sCAAsC;;6BAErD,IAAI;iCACF,IAAI,EAAE,QAAQ;iCACd,KAAK,EAAE,MAAM;iCACb,WAAW,EAAE,gCAAgC;;6BAE/C,KAAK;iCACH,IAAI,EAAE,QAAQ;iCACd,KAAK,EAAE,OAAO;iCACd,WAAW,EAAE,uCAAuC;iCACpD,oBAAoB;;;6CAGb,IAAI,EAAE,MAAM,EAAE,OAAO;;;qBA0G5B,KAAK,EAAE,YAAY;qBACnB,WAAW,EAAE,mCAAmC;qBAChD,UAAU,EAAE,QAAQ;;;;;CAKS,CAAC;AAEpC;;;;;;;;;;;GAWG;AACH,MAAM,MAAM,oBAAoB,GAAG,IAAI,CAAC,UAAU,CAAC,OAAO,sBAAsB,CAAC,EAAE,OAAO,CAAC,GAAG;IAC5F,QAAQ,CAAC,KAAK,EAAE,cAAc,EAAE,CAAC;IACjC,QAAQ,CAAC,QAAQ,CAAC,EAAE,aAAa,CAAC;QAChC,QAAQ,CAAC,IAAI,EAAE,MAAM,GAAG,WAAW,GAAG,MAAM,CAAC;QAC7C,QAAQ,CAAC,OAAO,EAAE,OAAO,CAAC;KAC3B,CAAC,CAAC;IACH,QAAQ,CAAC,SAAS,CAAC,EAAE,MAAM,CAAC;CAC7B,CAAC;AAEF,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAC/E,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,oBAAoB,CAAC,CAAC;AAMrE,qBAAa,eAAgB,SAAQ,eAAe,CAClD,oBAAoB,EACpB,qBAAqB,EACrB,qBAAqB,CACtB;IACC,OAAuB,IAAI,SAAqB;IAChD,OAAuB,QAAQ,SAAmB;IAClD,OAAuB,KAAK,SAAkB;IAC9C,OAAuB,WAAW,SACyF;IAC3H,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAED,wEAAwE;IACxE,OAAO,CAAC,kBAAkB,CAAqB;IAE/C;;;;OAIG;IACH,UAAyB,WAAW,CAClC,KAAK,EAAE,oBAAoB,GAC1B,OAAO,CAAC,UAAU,CAAC,oBAAoB,CAAC,CAAC,CAa3C;IAED,OAAO,CAAC,sBAAsB;IAaf,OAAO,CACpB,KAAK,EAAE,oBAAoB,EAC3B,cAAc,EAAE,eAAe,GAC9B,OAAO,CAAC,qBAAqB,GAAG,SAAS,CAAC,CAI5C;IAEe,aAAa,CAC3B,KAAK,EAAE,oBAAoB,EAC3B,OAAO,EAAE,eAAe,GACvB,aAAa,CAAC,WAAW,CAAC,qBAAqB,CAAC,CAAC,CAGnD;CACF;AAED;;GAEG;AACH,eAAO,MAAM,WAAW,UAAW,oBAAoB,WAAW,qBAAqB;;;;;;;;;EAEtF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CAAC,oBAAoB,EAAE,qBAAqB,EAAE,qBAAqB,CAAC,CAAC;KACjG;CACF"}

package/dist/task/UnloadModelTask.d.ts

@@ -48,6 +48,14 @@ declare const UnloadModelInputSchema: {
                                 readonly format: "credential";
                                 readonly "x-ui-hidden": true;
                             };
+                            readonly native_dimensions: {
+                                readonly type: "integer";
+                                readonly description: "Native output vector dimensions for embedding models";
+                            };
+                            readonly mrl: {
+                                readonly type: "boolean";
+                                readonly description: "Whether the model supports Matryoshka Representation Learning";
+                            };
                         };
                         readonly additionalProperties: true;
                         readonly default: {};
@@ -112,6 +120,14 @@ declare const UnloadModelOutputSchema: {
                                 readonly format: "credential";
                                 readonly "x-ui-hidden": true;
                             };
+                            readonly native_dimensions: {
+                                readonly type: "integer";
+                                readonly description: "Native output vector dimensions for embedding models";
+                            };
+                            readonly mrl: {
+                                readonly type: "boolean";
+                                readonly description: "Whether the model supports Matryoshka Representation Learning";
+                            };
                         };
                         readonly additionalProperties: true;
                         readonly default: {};
@@ -171,6 +187,8 @@ export declare const unloadModel: (input: UnloadModelTaskRunInput, config?: Unlo
         provider_config: {
             [x: string]: unknown;
             credential_key?: string | undefined;
+            mrl?: boolean | undefined;
+            native_dimensions?: number | undefined;
         };
         tasks?: string[] | undefined;
         title?: string | undefined;

package/dist/task/UnloadModelTask.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"UnloadModelTask.d.ts","sourceRoot":"","sources":["../../src/task/UnloadModelTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAY,MAAM,sBAAsB,CAAC;AAChE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,QAAA,MAAM,sBAAsB;mBACpB,QAAQ;;iBAEZ,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAI0B,CAAC;AAEpC,QAAA,MAAM,uBAAuB;mBACrB,QAAQ;;iBAEZ,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAI0B,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAChF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAClF,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,uBAAuB,CAAC,CAAC;AAExE;;;;;GAKG;AACH,qBAAa,eAAgB,SAAQ,MAAM,CACzC,uBAAuB,EACvB,wBAAwB,EACxB,qBAAqB,CACtB;IACC,OAAuB,IAAI,SAAqB;IAChD,OAAuB,QAAQ,SAAc;IAC7C,OAAuB,KAAK,SAAkB;IAC9C,OAAuB,WAAW,SAC8B;IAChE,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;IACD,OAAuB,SAAS,UAAS;CAC1C;AAED;;;;;GAKG;AACH,eAAO,MAAM,WAAW,UAAW,uBAAuB,WAAW,qBAAqB;;;;;;;;;;;;;;;;EAEzF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CACzB,uBAAuB,EACvB,wBAAwB,EACxB,qBAAqB,CACtB,CAAC;KACH;CACF"}
+{"version":3,"file":"UnloadModelTask.d.ts","sourceRoot":"","sources":["../../src/task/UnloadModelTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,cAAc,EAAY,MAAM,sBAAsB,CAAC;AAChE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,sBAAsB,CAAC;AACvD,OAAO,EAAE,cAAc,EAAE,UAAU,EAAE,MAAM,uBAAuB,CAAC;AACnE,OAAO,EAAE,MAAM,EAAE,MAAM,eAAe,CAAC;AAKvC,QAAA,MAAM,sBAAsB;mBACpB,QAAQ;;iBAEZ,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAI0B,CAAC;AAEpC,QAAA,MAAM,uBAAuB;mBACrB,QAAQ;;iBAEZ,KAAK;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;CAI0B,CAAC;AAEpC,MAAM,MAAM,uBAAuB,GAAG,UAAU,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAChF,MAAM,MAAM,wBAAwB,GAAG,UAAU,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAClF,MAAM,MAAM,qBAAqB,GAAG,UAAU,CAAC,uBAAuB,CAAC,CAAC;AAExE;;;;;GAKG;AACH,qBAAa,eAAgB,SAAQ,MAAM,CACzC,uBAAuB,EACvB,wBAAwB,EACxB,qBAAqB,CACtB;IACC,OAAuB,IAAI,SAAqB;IAChD,OAAuB,QAAQ,SAAc;IAC7C,OAAuB,KAAK,SAAkB;IAC9C,OAAuB,WAAW,SAC8B;IAChE,OAAuB,WAAW,IAAI,cAAc,CAEnD;IACD,OAAuB,YAAY,IAAI,cAAc,CAEpD;IACD,OAAuB,SAAS,UAAS;CAC1C;AAED;;;;;GAKG;AACH,eAAO,MAAM,WAAW,UAAW,uBAAuB,WAAW,qBAAqB;;;;;;;;;;;;;;;;;;EAEzF,CAAC;AAEF,OAAO,QAAQ,sBAAsB,CAAC,CAAC;IACrC,UAAU,QAAQ;QAChB,WAAW,EAAE,cAAc,CACzB,uBAAuB,EACvB,wBAAwB,EACxB,qBAAqB,CACtB,CAAC;KACH;CACF"}

package/dist/task/base/AiTask.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AiTask.d.ts","sourceRoot":"","sources":["../../../src/task/base/AiTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;GAGG;AAEH,OAAO,EAAE,GAAG,EAAE,MAAM,qBAAqB,CAAC;AAC1C,OAAO,KAAK,EACV,eAAe,EACf,uBAAuB,EACvB,UAAU,EAEV,gBAAgB,EAChB,UAAU,EACX,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAEL,IAAI,EAGJ,SAAS,EAEV,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACtD,OAAO,KAAK,EAAE,cAAc,EAAc,MAAM,uBAAuB,CAAC;AAExE,OAAO,EAAS,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAGpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAiB3D,MAAM,WAAW,WAAY,SAAQ,SAAS;IAC5C,KAAK,EAAE,MAAM,GAAG,WAAW,CAAC;CAC7B;AAED;;;;;;GAMG;AACH,qBAAa,MAAM,CACjB,KAAK,SAAS,WAAW,GAAG,WAAW,EACvC,MAAM,SAAS,UAAU,GAAG,UAAU,EACtC,MAAM,SAAS,UAAU,CAAC,KAAK,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC,CACpD,SAAQ,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;IACnC,OAAuB,IAAI,EAAE,MAAM,CAAY;IAC/C,OAAuB,sBAAsB,EAAE,OAAO,CAAQ;IAE9D,OAAuB,YAAY,IAAI,gBAAgB,CAItD;IAEe,YAAY,IAAI,gBAAgB,CAoB/C;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAMc,OAAO,CACpB,KAAK,EAAE,KAAK,EACZ,cAAc,EAAE,eAAe,GAC9B,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAa7B;IAMD;;;OAGG;IACH,UAAgB,WAAW,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAiCpE;IAED;;OAEG;IACG,SAAS,CAAC,KAAK,EAAE,KAAK,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,CAAC,CAYzF;IAED;;OAEG;IACH,UAAgB,mBAAmB,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAG7E;IAMD;;;OAGG;IACY,eAAe,CAC5B,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,uBAAuB,GAC/B,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAa7B;IAMD;;OAEG;IACmB,aAAa,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,CA+ClE;IAEqB,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,eAAe,GAAG,OAAO,CAAC,KAAK,CAAC,CAqCzF;CACF"}
+{"version":3,"file":"AiTask.d.ts","sourceRoot":"","sources":["../../../src/task/base/AiTask.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH;;;GAGG;AAEH,OAAO,EAAE,GAAG,EAAE,MAAM,qBAAqB,CAAC;AAC1C,OAAO,KAAK,EACV,eAAe,EACf,uBAAuB,EACvB,UAAU,EAEV,gBAAgB,EAChB,UAAU,EACX,MAAM,sBAAsB,CAAC;AAC9B,OAAO,EAEL,IAAI,EAGJ,SAAS,EAEV,MAAM,sBAAsB,CAAC;AAC9B,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gBAAgB,CAAC;AACtD,OAAO,KAAK,EAAE,cAAc,EAAc,MAAM,uBAAuB,CAAC;AAExE,OAAO,EAAS,UAAU,EAAE,MAAM,iBAAiB,CAAC;AAGpD,OAAO,KAAK,EAAE,WAAW,EAAE,MAAM,yBAAyB,CAAC;AAiB3D,MAAM,WAAW,WAAY,SAAQ,SAAS;IAC5C,KAAK,EAAE,MAAM,GAAG,WAAW,CAAC;CAC7B;AAED;;;;;;GAMG;AACH,qBAAa,MAAM,CACjB,KAAK,SAAS,WAAW,GAAG,WAAW,EACvC,MAAM,SAAS,UAAU,GAAG,UAAU,EACtC,MAAM,SAAS,UAAU,CAAC,KAAK,CAAC,GAAG,UAAU,CAAC,KAAK,CAAC,CACpD,SAAQ,IAAI,CAAC,KAAK,EAAE,MAAM,EAAE,MAAM,CAAC;IACnC,OAAuB,IAAI,EAAE,MAAM,CAAY;IAC/C,OAAuB,sBAAsB,EAAE,OAAO,CAAQ;IAE9D,OAAuB,YAAY,IAAI,gBAAgB,CAItD;IAEe,YAAY,IAAI,gBAAgB,CAoB/C;IAED,OAAuB,YAAY,IAAI,cAAc,CAEpD;IAMc,OAAO,CACpB,KAAK,EAAE,KAAK,EACZ,cAAc,EAAE,eAAe,GAC9B,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CA4B7B;IAMD;;;OAGG;IACH,UAAgB,WAAW,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,UAAU,CAAC,KAAK,CAAC,CAAC,CAsCpE;IAED;;OAEG;IACG,SAAS,CAAC,KAAK,EAAE,KAAK,EAAE,SAAS,CAAC,EAAE,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,UAAU,CAAC,KAAK,CAAC,EAAE,MAAM,CAAC,CAAC,CAYzF;IAED;;OAEG;IACH,UAAgB,mBAAmB,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAG7E;IAMD;;;OAGG;IACY,eAAe,CAC5B,KAAK,EAAE,KAAK,EACZ,MAAM,EAAE,MAAM,EACd,OAAO,EAAE,uBAAuB,GAC/B,OAAO,CAAC,MAAM,GAAG,SAAS,CAAC,CAa7B;IAMD;;OAEG;IACmB,aAAa,CAAC,KAAK,EAAE,KAAK,GAAG,OAAO,CAAC,OAAO,CAAC,CA+ClE;IAEqB,WAAW,CAAC,KAAK,EAAE,KAAK,EAAE,QAAQ,EAAE,eAAe,GAAG,OAAO,CAAC,KAAK,CAAC,CAqCzF;CACF"}

package/dist/task/base/AiTaskSchemas.d.ts

@@ -54,6 +54,14 @@ export declare function TypeModelByDetail<S extends TypeModelSemantic = "model",
                     readonly format: "credential";
                     readonly "x-ui-hidden": true;
                 };
+                readonly native_dimensions: {
+                    readonly type: "integer";
+                    readonly description: "Native output vector dimensions for embedding models";
+                };
+                readonly mrl: {
+                    readonly type: "boolean";
+                    readonly description: "Whether the model supports Matryoshka Representation Learning";
+                };
             };
             readonly additionalProperties: true;
             readonly default: {};
@@ -108,6 +116,14 @@ export declare function TypeModel<S extends TypeModelSemantic = "model", O exten
                         readonly format: "credential";
                         readonly "x-ui-hidden": true;
                     };
+                    readonly native_dimensions: {
+                        readonly type: "integer";
+                        readonly description: "Native output vector dimensions for embedding models";
+                    };
+                    readonly mrl: {
+                        readonly type: "boolean";
+                        readonly description: "Whether the model supports Matryoshka Representation Learning";
+                    };
                 };
                 readonly additionalProperties: true;
                 readonly default: {};

package/dist/task/base/AiTaskSchemas.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"AiTaskSchemas.d.ts","sourceRoot":"","sources":["../../../src/task/base/AiTaskSchemas.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,wBAAwB,EAAc,MAAM,uBAAuB,CAAC;AAG7E,eAAO,MAAM,YAAY,iBAAiB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;mBAEvD,QAAQ;oBACP,UAAU;0BACJ,qBAAqB;wBACvB,CAAC;wBACD,CAAC;CAEH,CAAC;AAEd,MAAM,MAAM,iBAAiB,GAAG,OAAO,GAAG,SAAS,MAAM,EAAE,CAAC;AAE5D,MAAM,MAAM,UAAU,GAAG,wBAAwB,GAAG;IAClD,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC;IACxB,QAAQ,CAAC,MAAM,EAAE,iBAAiB,CAAC;CACpC,CAAC;AAEF,wBAAgB,iBAAiB,CAC/B,CAAC,SAAS,iBAAiB,GAAG,OAAO,EACrC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,EACtC,QAAQ,GAAE,CAAgB,EAAE,OAAO,GAAE,CAAW;oBAYvC,OAAO;;;;mBAIR,QAAQ;EAEjB;AAED,wBAAgB,iBAAiB,CAC/B,CAAC,SAAS,iBAAiB,GAAG,OAAO,EACrC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,EACtC,QAAQ,GAAE,CAAgB,EAAE,OAAO,GAAE,CAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EASjD;AAED,wBAAgB,SAAS,CACvB,CAAC,SAAS,iBAAiB,GAAG,OAAO,EACrC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,EACtC,QAAQ,GAAE,CAAgB,EAAE,OAAO,GAAE,CAAW;;wBAzBvC,OAAO;;;;uBAIR,QAAQ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA2BjB;AAED,wBAAgB,iBAAiB,CAAC,KAAK,CAAC,CAAC,SAAS,wBAAwB,EAAE,IAAI,EAAE,CAAC;;uBAEzD,OAAO;;;EAEhC;AAED,MAAM,MAAM,WAAW,GAAG,WAAW,GAAG,eAAe,GAAG,UAAU,CAAC;AAErE;;GAEG;AACH,eAAO,MAAM,cAAc;;uBAGf,QAAQ;wBACP,YAAY;8BACN,mBAAmB;yBACxB,gBAAgB;;uBAGlB,QAAQ;;;;;mCAMA,QAAQ;qCACN,mBAAmB;oCACpB,aAAa;;mCAGd,QAAQ;qCACN,uBAAuB;oCACxB,iBAAiB;;mCAGlB,QAAQ;qCACN,kBAAkB;oCACnB,YAAY;;mCAGb,QAAQ;;iCAEZ,IAAI;qCACF,IAAI,EAAE,OAAO;qCACb,KAAK;yCAAI,IAAI,EAAE,QAAQ;yCAAE,MAAM,EAAE,cAAc;;qCAC/C,MAAM,EAAE,mBAAmB;qCAC3B,KAAK,EAAE,MAAM;qCACb,WAAW,EAAE,mBAAmB;;iCAElC,KAAK;qCAAI,IAAI,EAAE,QAAQ;qCAAE,KAAK,EAAE,OAAO;qCAAE,WAAW,EAAE,oBAAoB;;iCAC1E,MAAM;qCAAI,IAAI,EAAE,QAAQ;qCAAE,KAAK,EAAE,QAAQ;qCAAE,WAAW,EAAE,qBAAqB;;iCAC7E,QAAQ;qCACN,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,UAAU;qCACjB,WAAW,EAAE,uBAAuB;;iCAEtC,WAAW;qCACT,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,cAAc;qCACrB,WAAW,EAAE,2BAA2B;;;;;qCAKpC,mBAAmB;oCACpB,aAAa;;;;yBAIjB,IAAI,EAAE,QAAQ;yBAAE,KAAK,EAAE,OAAO;yBAAE,WAAW,EAAE,oBAAoB;;;yBAChE,IAAI,EAAE,QAAQ;yBAAE,KAAK,EAAE,QAAQ;yBAAE,WAAW,EAAE,qBAAqB;;;yBAE3E,IAAI,EAAE,QAAQ;yBACd,KAAK,EAAE,UAAU;yBACjB,WAAW,EAAE,uBAAuB;yBACpC,OAAO,EAAE,CAAC;yBACV,OAAO,EAAE,CAAC;;;;;oBAMX,OAAO;qBACN,OAAO;0BACF,qCAAqC;CACrB,CAAC;AAEhC;;GAEG;AACH,eAAO,MAAM,cAAc;mBACnB,QAAQ;oBACP,OAAO;qBACN,gBAAgB;0BACX,4BAA4B;CACZ,CAAC;AAEhC;;GAEG;AACH,eAAO,MAAM,eAAe;mBACpB,QAAQ;;iBAEZ,CAAC;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,cAAc;qBAAE,WAAW,EAAE,+BAA+B;;iBACxF,CAAC;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,cAAc;qBAAE,WAAW,EAAE,8BAA8B;;iBACvF,KAAK;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,OAAO;qBAAE,WAAW,EAAE,2BAA2B;;iBACjF,MAAM;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,QAAQ;qBAAE,WAAW,EAAE,4BAA4B;;;;;oBAI/E,cAAc;0BACR,0BAA0B;CACV,CAAC;AAEhC;;GAEG;AACH,eAAO,MAAM,YAAY;mBACjB,QAAQ;;iBAEZ,KAAK;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,OAAO;qBAAE,WAAW,EAAE,gBAAgB;;iBACtE,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,kBAAkB;qBACzB,WAAW,EAAE,kCAAkC;qBAC/C,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;;;;;oBAKP,UAAU;0BACJ,8CAA8C;CAC9B,CAAC"}
+{"version":3,"file":"AiTaskSchemas.d.ts","sourceRoot":"","sources":["../../../src/task/base/AiTaskSchemas.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,wBAAwB,EAAc,MAAM,uBAAuB,CAAC;AAG7E,eAAO,MAAM,YAAY,iBAAiB,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;mBAEvD,QAAQ;oBACP,UAAU;0BACJ,qBAAqB;wBACvB,CAAC;wBACD,CAAC;CAEH,CAAC;AAEd,MAAM,MAAM,iBAAiB,GAAG,OAAO,GAAG,SAAS,MAAM,EAAE,CAAC;AAE5D,MAAM,MAAM,UAAU,GAAG,wBAAwB,GAAG;IAClD,QAAQ,CAAC,IAAI,EAAE,QAAQ,CAAC;IACxB,QAAQ,CAAC,MAAM,EAAE,iBAAiB,CAAC;CACpC,CAAC;AAEF,wBAAgB,iBAAiB,CAC/B,CAAC,SAAS,iBAAiB,GAAG,OAAO,EACrC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,EACtC,QAAQ,GAAE,CAAgB,EAAE,OAAO,GAAE,CAAW;oBAYvC,OAAO;;;;mBAIR,QAAQ;EAEjB;AAED,wBAAgB,iBAAiB,CAC/B,CAAC,SAAS,iBAAiB,GAAG,OAAO,EACrC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,EACtC,QAAQ,GAAE,CAAgB,EAAE,OAAO,GAAE,CAAW;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EASjD;AAED,wBAAgB,SAAS,CACvB,CAAC,SAAS,iBAAiB,GAAG,OAAO,EACrC,CAAC,SAAS,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,EAAE,EACtC,QAAQ,GAAE,CAAgB,EAAE,OAAO,GAAE,CAAW;;wBAzBvC,OAAO;;;;uBAIR,QAAQ;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;;EA2BjB;AAED,wBAAgB,iBAAiB,CAAC,KAAK,CAAC,CAAC,SAAS,wBAAwB,EAAE,IAAI,EAAE,CAAC;;uBAEzD,OAAO;;;EAEhC;AAED,MAAM,MAAM,WAAW,GAAG,WAAW,GAAG,eAAe,GAAG,UAAU,CAAC;AAErE;;GAEG;AACH,eAAO,MAAM,cAAc;;uBAGf,QAAQ;wBACP,YAAY;8BACN,mBAAmB;yBACxB,gBAAgB;;uBAGlB,QAAQ;;;;;mCAMA,QAAQ;qCACN,mBAAmB;oCACpB,aAAa;;mCAGd,QAAQ;qCACN,uBAAuB;oCACxB,iBAAiB;;mCAGlB,QAAQ;qCACN,kBAAkB;oCACnB,YAAY;;mCAGb,QAAQ;;iCAEZ,IAAI;qCACF,IAAI,EAAE,OAAO;qCACb,KAAK;yCAAI,IAAI,EAAE,QAAQ;yCAAE,MAAM,EAAE,cAAc;;qCAC/C,MAAM,EAAE,mBAAmB;qCAC3B,KAAK,EAAE,MAAM;qCACb,WAAW,EAAE,mBAAmB;;iCAElC,KAAK;qCAAI,IAAI,EAAE,QAAQ;qCAAE,KAAK,EAAE,OAAO;qCAAE,WAAW,EAAE,oBAAoB;;iCAC1E,MAAM;qCAAI,IAAI,EAAE,QAAQ;qCAAE,KAAK,EAAE,QAAQ;qCAAE,WAAW,EAAE,qBAAqB;;iCAC7E,QAAQ;qCACN,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,UAAU;qCACjB,WAAW,EAAE,uBAAuB;;iCAEtC,WAAW;qCACT,IAAI,EAAE,QAAQ;qCACd,KAAK,EAAE,cAAc;qCACrB,WAAW,EAAE,2BAA2B;;;;;qCAKpC,mBAAmB;oCACpB,aAAa;;;;yBAIjB,IAAI,EAAE,QAAQ;yBAAE,KAAK,EAAE,OAAO;yBAAE,WAAW,EAAE,oBAAoB;;;yBAChE,IAAI,EAAE,QAAQ;yBAAE,KAAK,EAAE,QAAQ;yBAAE,WAAW,EAAE,qBAAqB;;;yBAE3E,IAAI,EAAE,QAAQ;yBACd,KAAK,EAAE,UAAU;yBACjB,WAAW,EAAE,uBAAuB;yBACpC,OAAO,EAAE,CAAC;yBACV,OAAO,EAAE,CAAC;;;;;oBAMX,OAAO;qBACN,OAAO;0BACF,qCAAqC;CACrB,CAAC;AAEhC;;GAEG;AACH,eAAO,MAAM,cAAc;mBACnB,QAAQ;oBACP,OAAO;qBACN,gBAAgB;0BACX,4BAA4B;CACZ,CAAC;AAEhC;;GAEG;AACH,eAAO,MAAM,eAAe;mBACpB,QAAQ;;iBAEZ,CAAC;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,cAAc;qBAAE,WAAW,EAAE,+BAA+B;;iBACxF,CAAC;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,cAAc;qBAAE,WAAW,EAAE,8BAA8B;;iBACvF,KAAK;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,OAAO;qBAAE,WAAW,EAAE,2BAA2B;;iBACjF,MAAM;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,QAAQ;qBAAE,WAAW,EAAE,4BAA4B;;;;;oBAI/E,cAAc;0BACR,0BAA0B;CACV,CAAC;AAEhC;;GAEG;AACH,eAAO,MAAM,YAAY;mBACjB,QAAQ;;iBAEZ,KAAK;qBAAI,IAAI,EAAE,QAAQ;qBAAE,KAAK,EAAE,OAAO;qBAAE,WAAW,EAAE,gBAAgB;;iBACtE,KAAK;qBACH,IAAI,EAAE,QAAQ;qBACd,KAAK,EAAE,kBAAkB;qBACzB,WAAW,EAAE,kCAAkC;qBAC/C,OAAO,EAAE,CAAC;qBACV,OAAO,EAAE,CAAC;;;;;oBAKP,UAAU;0BACJ,8CAA8C;CAC9B,CAAC"}

package/dist/task/index.d.ts

@@ -3,7 +3,6 @@
  * Copyright 2025 Steven Roussey <sroussey@gmail.com>
  * SPDX-License-Identifier: Apache-2.0
  */
-import { AgentTask } from "./AgentTask";
 import { BackgroundRemovalTask } from "./BackgroundRemovalTask";
 import { ChunkRetrievalTask } from "./ChunkRetrievalTask";
 import { ChunkToVectorTask } from "./ChunkToVectorTask";
@@ -19,6 +18,7 @@ import { FaceLandmarkerTask } from "./FaceLandmarkerTask";
 import { HandLandmarkerTask } from "./HandLandmarkerTask";
 import { HierarchicalChunkerTask } from "./HierarchicalChunkerTask";
 import { HierarchyJoinTask } from "./HierarchyJoinTask";
+import { KbToDocumentsTask } from "./KbToDocumentsTask";
 import { ImageClassificationTask } from "./ImageClassificationTask";
 import { ImageEmbeddingTask } from "./ImageEmbeddingTask";
 import { ImageSegmentationTask } from "./ImageSegmentationTask";
@@ -44,10 +44,7 @@ import { TopicSegmenterTask } from "./TopicSegmenterTask";
 import { UnloadModelTask } from "./UnloadModelTask";
 import { VectorQuantizeTask } from "./VectorQuantizeTask";
 import { VectorSimilarityTask } from "./VectorSimilarityTask";
-export declare const registerAiTasks: () => (typeof ToolCallingTask | typeof AgentTask | typeof BackgroundRemovalTask | typeof TextEmbeddingTask | typeof ChunkRetrievalTask | typeof ChunkToVectorTask | typeof ChunkVectorHybridSearchTask | typeof ChunkVectorSearchTask | typeof ChunkVectorUpsertTask | typeof CountTokensTask | typeof ContextBuilderTask | typeof TextNamedEntityRecognitionTask | typeof TextSummaryTask | typeof DocumentEnricherTask | typeof DocumentUpsertTask | typeof FaceDetectorTask | typeof FaceLandmarkerTask | typeof HandLandmarkerTask | typeof HierarchicalChunkerTask | typeof HierarchyJoinTask | typeof ImageClassificationTask | typeof ImageEmbeddingTask | typeof ImageSegmentationTask | typeof ImageToTextTask | typeof ModelSearchTask | typeof ObjectDetectionTask | typeof PoseLandmarkerTask | typeof QueryExpanderTask | typeof TextClassificationTask | typeof RerankerTask | typeof StructuralParserTask | typeof StructuredGenerationTask | typeof TextChunkerTask | typeof TextFillMaskTask | typeof TextGenerationTask | typeof TextLanguageDetectionTask | typeof TextQuestionAnswerTask | typeof TopicSegmenterTask | typeof UnloadModelTask | typeof VectorQuantizeTask | typeof VectorSimilarityTask)[];
-export * from "./AgentTask";
-export * from "./AgentTypes";
-export * from "./AgentUtils";
+export declare const registerAiTasks: () => (typeof ToolCallingTask | typeof BackgroundRemovalTask | typeof TextEmbeddingTask | typeof ChunkRetrievalTask | typeof ChunkToVectorTask | typeof ChunkVectorHybridSearchTask | typeof ChunkVectorSearchTask | typeof ChunkVectorUpsertTask | typeof CountTokensTask | typeof ContextBuilderTask | typeof TextNamedEntityRecognitionTask | typeof TextSummaryTask | typeof DocumentEnricherTask | typeof DocumentUpsertTask | typeof FaceDetectorTask | typeof FaceLandmarkerTask | typeof HandLandmarkerTask | typeof HierarchicalChunkerTask | typeof HierarchyJoinTask | typeof KbToDocumentsTask | typeof ImageClassificationTask | typeof ImageEmbeddingTask | typeof ImageSegmentationTask | typeof ImageToTextTask | typeof ModelSearchTask | typeof ObjectDetectionTask | typeof PoseLandmarkerTask | typeof QueryExpanderTask | typeof TextClassificationTask | typeof RerankerTask | typeof StructuralParserTask | typeof StructuredGenerationTask | typeof TextChunkerTask | typeof TextFillMaskTask | typeof TextGenerationTask | typeof TextLanguageDetectionTask | typeof TextQuestionAnswerTask | typeof TopicSegmenterTask | typeof UnloadModelTask | typeof VectorQuantizeTask | typeof VectorSimilarityTask)[];
 export * from "./MessageConversion";
 export * from "./ToolCallingUtils";
 export * from "./BackgroundRemovalTask";
@@ -70,6 +67,7 @@ export * from "./GestureRecognizerTask";
 export * from "./HandLandmarkerTask";
 export * from "./HierarchicalChunkerTask";
 export * from "./HierarchyJoinTask";
+export * from "./KbToDocumentsTask";
 export * from "./ImageClassificationTask";
 export * from "./ImageEmbeddingTask";
 export * from "./ImageSegmentationTask";

package/dist/task/index.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/task/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,SAAS,EAAE,MAAM,aAAa,CAAC;AACxC,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,2BAA2B,EAAE,MAAM,+BAA+B,CAAC;AAC5E,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAC9D,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAE1D,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAE1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AACpE,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AACpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpD,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAC9D,OAAO,EAAE,wBAAwB,EAAE,MAAM,4BAA4B,CAAC;AACtE,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,yBAAyB,EAAE,MAAM,6BAA6B,CAAC;AACxE,OAAO,EAAE,8BAA8B,EAAE,MAAM,kCAAkC,CAAC;AAClF,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAElE,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpD,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAK9D,eAAO,MAAM,eAAe,gqCAmD3B,CAAC;AAEF,cAAc,aAAa,CAAC;AAC5B,cAAc,cAAc,CAAC;AAC7B,cAAc,cAAc,CAAC;AAC7B,cAAc,qBAAqB,CAAC;AACpC,cAAc,oBAAoB,CAAC;AACnC,cAAc,yBAAyB,CAAC;AACxC,cAAc,eAAe,CAAC;AAC9B,cAAc,sBAAsB,CAAC;AACrC,cAAc,wBAAwB,CAAC;AACvC,cAAc,sBAAsB,CAAC;AACrC,cAAc,qBAAqB,CAAC;AACpC,cAAc,+BAA+B,CAAC;AAC9C,cAAc,yBAAyB,CAAC;AACxC,cAAc,yBAAyB,CAAC;AACxC,cAAc,sBAAsB,CAAC;AACrC,cAAc,mBAAmB,CAAC;AAClC,cAAc,wBAAwB,CAAC;AACvC,cAAc,sBAAsB,CAAC;AACrC,cAAc,qBAAqB,CAAC;AACpC,cAAc,oBAAoB,CAAC;AACnC,cAAc,sBAAsB,CAAC;AACrC,cAAc,yBAAyB,CAAC;AACxC,cAAc,sBAAsB,CAAC;AACrC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,qBAAqB,CAAC;AACpC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,sBAAsB,CAAC;AACrC,cAAc,yBAAyB,CAAC;AACxC,cAAc,mBAAmB,CAAC;AAClC,cAAc,iBAAiB,CAAC;AAChC,cAAc,mBAAmB,CAAC;AAClC,cAAc,uBAAuB,CAAC;AACtC,cAAc,sBAAsB,CAAC;AACrC,cAAc,qBAAqB,CAAC;AACpC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,wBAAwB,CAAC;AACvC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,mBAAmB,CAAC;AAClC,cAAc,0BAA0B,CAAC;AACzC,cAAc,qBAAqB,CAAC;AACpC,cAAc,oBAAoB,CAAC;AACnC,cAAc,sBAAsB,CAAC;AACrC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,kCAAkC,CAAC;AACjD,cAAc,0BAA0B,CAAC;AACzC,cAAc,oBAAoB,CAAC;AACnC,cAAc,mBAAmB,CAAC;AAClC,cAAc,uBAAuB,CAAC;AACtC,cAAc,mBAAmB,CAAC;AAClC,cAAc,sBAAsB,CAAC;AACrC,cAAc,mBAAmB,CAAC;AAClC,cAAc,sBAAsB,CAAC;AACrC,cAAc,wBAAwB,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/task/index.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAGH,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,2BAA2B,EAAE,MAAM,+BAA+B,CAAC;AAC5E,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAC9D,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAE1D,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAE1D,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AACpE,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,uBAAuB,EAAE,MAAM,2BAA2B,CAAC;AACpE,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,qBAAqB,EAAE,MAAM,yBAAyB,CAAC;AAChE,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpD,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,mBAAmB,EAAE,MAAM,uBAAuB,CAAC;AAC5D,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAC9D,OAAO,EAAE,wBAAwB,EAAE,MAAM,4BAA4B,CAAC;AACtE,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAClE,OAAO,EAAE,iBAAiB,EAAE,MAAM,qBAAqB,CAAC;AACxD,OAAO,EAAE,gBAAgB,EAAE,MAAM,oBAAoB,CAAC;AACtD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,yBAAyB,EAAE,MAAM,6BAA6B,CAAC;AACxE,OAAO,EAAE,8BAA8B,EAAE,MAAM,kCAAkC,CAAC;AAClF,OAAO,EAAE,sBAAsB,EAAE,MAAM,0BAA0B,CAAC;AAElE,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AAEpD,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAE,kBAAkB,EAAE,MAAM,sBAAsB,CAAC;AAC1D,OAAO,EAAE,oBAAoB,EAAE,MAAM,wBAAwB,CAAC;AAK9D,eAAO,MAAM,eAAe,wqCAmD3B,CAAC;AAEF,cAAc,qBAAqB,CAAC;AACpC,cAAc,oBAAoB,CAAC;AACnC,cAAc,yBAAyB,CAAC;AACxC,cAAc,eAAe,CAAC;AAC9B,cAAc,sBAAsB,CAAC;AACrC,cAAc,wBAAwB,CAAC;AACvC,cAAc,sBAAsB,CAAC;AACrC,cAAc,qBAAqB,CAAC;AACpC,cAAc,+BAA+B,CAAC;AAC9C,cAAc,yBAAyB,CAAC;AACxC,cAAc,yBAAyB,CAAC;AACxC,cAAc,sBAAsB,CAAC;AACrC,cAAc,mBAAmB,CAAC;AAClC,cAAc,wBAAwB,CAAC;AACvC,cAAc,sBAAsB,CAAC;AACrC,cAAc,qBAAqB,CAAC;AACpC,cAAc,oBAAoB,CAAC;AACnC,cAAc,sBAAsB,CAAC;AACrC,cAAc,yBAAyB,CAAC;AACxC,cAAc,sBAAsB,CAAC;AACrC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,qBAAqB,CAAC;AACpC,cAAc,qBAAqB,CAAC;AACpC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,sBAAsB,CAAC;AACrC,cAAc,yBAAyB,CAAC;AACxC,cAAc,mBAAmB,CAAC;AAClC,cAAc,iBAAiB,CAAC;AAChC,cAAc,mBAAmB,CAAC;AAClC,cAAc,uBAAuB,CAAC;AACtC,cAAc,sBAAsB,CAAC;AACrC,cAAc,qBAAqB,CAAC;AACpC,cAAc,gBAAgB,CAAC;AAC/B,cAAc,wBAAwB,CAAC;AACvC,cAAc,4BAA4B,CAAC;AAC3C,cAAc,mBAAmB,CAAC;AAClC,cAAc,0BAA0B,CAAC;AACzC,cAAc,qBAAqB,CAAC;AACpC,cAAc,oBAAoB,CAAC;AACnC,cAAc,sBAAsB,CAAC;AACrC,cAAc,6BAA6B,CAAC;AAC5C,cAAc,kCAAkC,CAAC;AACjD,cAAc,0BAA0B,CAAC;AACzC,cAAc,oBAAoB,CAAC;AACnC,cAAc,mBAAmB,CAAC;AAClC,cAAc,uBAAuB,CAAC;AACtC,cAAc,mBAAmB,CAAC;AAClC,cAAc,sBAAsB,CAAC;AACrC,cAAc,mBAAmB,CAAC;AAClC,cAAc,sBAAsB,CAAC;AACrC,cAAc,wBAAwB,CAAC"}

package/dist/worker.js

@@ -92,7 +92,7 @@ class AiJob extends Job {
         const timeoutMs = resolveAiJobTimeoutMs(input.aiProvider, input.timeoutMs);
         const timeoutSignal = AbortSignal.timeout(timeoutMs);
         const combinedSignal = AbortSignal.any([context.signal, timeoutSignal]);
-        return await fn(input.taskInput, model, context.updateProgress, combinedSignal, input.outputSchema);
+        return await fn(input.taskInput, model, context.updateProgress, combinedSignal, input.outputSchema, input.sessionId);
       };
       const runFnPromise = runFn();
       return await Promise.race([runFnPromise, abortPromise]);
@@ -120,7 +120,7 @@ class AiJob extends Job {
     const timeoutSignal = AbortSignal.timeout(timeoutMs);
     const combinedSignal = AbortSignal.any([context.signal, timeoutSignal]);
     try {
-      for await (const event of streamFn(input.taskInput, model, combinedSignal, input.outputSchema)) {
+      for await (const event of streamFn(input.taskInput, model, combinedSignal, input.outputSchema, input.sessionId)) {
         if (event.type === "finish") {
           lastFinishData = event.data;
         }
@@ -216,6 +216,19 @@ class AiProviderRegistry {
     }
     return this.defaultStrategy;
   }
+  createSession(providerName, model) {
+    const provider = this.providers.get(providerName);
+    if (!provider) {
+      throw new Error(`No provider found for "${providerName}". Register the provider before creating sessions.`);
+    }
+    return provider.createSession(model);
+  }
+  async disposeSession(providerName, sessionId) {
+    const provider = this.providers.get(providerName);
+    if (provider) {
+      await provider.disposeSession(sessionId);
+    }
+  }
   getProviderIdsForTask(taskType) {
     const taskMap = this.runFnRegistry.get(taskType);
     if (!taskMap)
@@ -229,9 +242,9 @@ class AiProviderRegistry {
     this.runFnRegistry.get(taskType).set(modelProvider, runFn);
   }
   registerAsWorkerRunFn(modelProvider, taskType) {
-    const workerFn = async (input, model, update_progress, signal, outputSchema) => {
+    const workerFn = async (input, model, update_progress, signal, outputSchema, sessionId) => {
       const workerManager = globalServiceRegistry.get(WORKER_MANAGER);
-      const result = await workerManager.callWorkerFunction(modelProvider, taskType, [input, model, outputSchema], {
+      const result = await workerManager.callWorkerFunction(modelProvider, taskType, [input, model, outputSchema, sessionId], {
         signal,
         onProgress: update_progress
       });
@@ -246,9 +259,9 @@ class AiProviderRegistry {
     this.streamFnRegistry.get(taskType).set(modelProvider, streamFn);
   }
   registerAsWorkerStreamFn(modelProvider, taskType) {
-    const streamFn = async function* (input, model, signal, outputSchema) {
+    const streamFn = async function* (input, model, signal, outputSchema, sessionId) {
       const workerManager = globalServiceRegistry.get(WORKER_MANAGER);
-      yield* workerManager.callWorkerStreamFunction(modelProvider, taskType, [input, model, outputSchema], { signal });
+      yield* workerManager.callWorkerStreamFunction(modelProvider, taskType, [input, model, outputSchema, sessionId], { signal });
     };
     this.registerStreamFn(modelProvider, taskType, streamFn);
   }
@@ -298,6 +311,7 @@ function setAiProviderRegistry(pr) {
 }
 
 // src/provider/AiProvider.ts
+var DEFAULT_AI_PROVIDER_WORKER_IDLE_TIMEOUT_MS = 15 * 60 * 1000;
 function resolveAiProviderGpuQueueConcurrency(concurrency) {
   if (concurrency === undefined) {
     return 1;
@@ -345,7 +359,13 @@ class AiProvider {
     const registry = getAiProviderRegistry();
     if (!isInline && options.worker) {
       const workerManager = globalServiceRegistry2.get(WORKER_MANAGER2);
-      workerManager.registerWorker(this.name, options.worker);
+      if (typeof options.worker === "function") {
+        workerManager.registerWorker(this.name, options.worker, {
+          idleTimeoutMs: options.workerIdleTimeoutMs ?? DEFAULT_AI_PROVIDER_WORKER_IDLE_TIMEOUT_MS
+        });
+      } else {
+        workerManager.registerWorker(this.name, options.worker);
+      }
       for (const taskType of this.taskTypes) {
         registry.registerAsWorkerRunFn(this.name, taskType);
         registry.registerAsWorkerStreamFn(this.name, taskType);
@@ -394,6 +414,10 @@ class AiProvider {
   }
   async onInitialize(_options) {}
   async dispose() {}
+  createSession(_model) {
+    return crypto.randomUUID();
+  }
+  async disposeSession(_sessionId) {}
   async afterRegister(_options) {}
 }
 // src/task/ToolCallingUtils.ts
@@ -630,7 +654,15 @@ var ModelConfigSchema = {
     provider_config: {
       type: "object",
       properties: {
-        credential_key: { type: "string", format: "credential", "x-ui-hidden": true }
+        credential_key: { type: "string", format: "credential", "x-ui-hidden": true },
+        native_dimensions: {
+          type: "integer",
+          description: "Native output vector dimensions for embedding models"
+        },
+        mrl: {
+          type: "boolean",
+          description: "Whether the model supports Matryoshka Representation Learning"
+        }
       },
       additionalProperties: true,
       default: {}
@@ -671,8 +703,9 @@ export {
   ModelRecordSchema,
   ModelPrimaryKeyNames,
   ModelConfigSchema,
+  DEFAULT_AI_PROVIDER_WORKER_IDLE_TIMEOUT_MS,
   AiProviderRegistry,
   AiProvider
 };
 
-//# debugId=D34633C7196C310D64756E2164756E21
+//# debugId=A123B823D13D61C764756E2164756E21