@loopstack/ai-module 0.19.0 → 0.20.0

package/dist/documents/ai-message-document.d.ts

@@ -1,3 +1,12 @@
+import { z } from 'zod';
+export declare const AiMessageDocumentSchema: z.ZodObject<{
+    id: z.ZodOptional<z.ZodString>;
+    role: z.ZodUnion<readonly [z.ZodLiteral<"system">, z.ZodLiteral<"user">, z.ZodLiteral<"assistant">]>;
+    metadata: z.ZodOptional<z.ZodAny>;
+    parts: z.ZodArray<z.ZodAny>;
+}, z.core.$strip>;
+export type AiMessageDocumentContentType = z.infer<typeof AiMessageDocumentSchema>;
 export declare class AiMessageDocument {
+    content: z.infer<typeof AiMessageDocumentSchema>;
 }
 //# sourceMappingURL=ai-message-document.d.ts.map

package/dist/documents/ai-message-document.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ai-message-document.d.ts","sourceRoot":"","sources":["../../src/documents/ai-message-document.ts"],"names":[],"mappings":"AAWA,qBASa,iBAAiB;CAAG"}
+{"version":3,"file":"ai-message-document.d.ts","sourceRoot":"","sources":["../../src/documents/ai-message-document.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAGxB,eAAO,MAAM,uBAAuB;;;;;iBAKlC,CAAC;AAEH,MAAM,MAAM,4BAA4B,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;AAEnF,qBAQa,iBAAiB;IAI5B,OAAO,EAAE,CAAC,CAAC,KAAK,CAAC,OAAO,uBAAuB,CAAC,CAAC;CAClD"}

package/dist/documents/ai-message-document.js

@@ -5,20 +5,30 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
     else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
     return c > 3 && r && Object.defineProperty(target, key, r), r;
 };
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.AiMessageDocument = void 0;
+exports.AiMessageDocument = exports.AiMessageDocumentSchema = void 0;
 const common_1 = require("@nestjs/common");
 const zod_1 = require("zod");
 const common_2 = require("@loopstack/common");
-const AiMessageDocumentSchema = zod_1.z.object({
+exports.AiMessageDocumentSchema = zod_1.z.object({
     id: zod_1.z.string().optional(),
     role: zod_1.z.union([zod_1.z.literal('system'), zod_1.z.literal('user'), zod_1.z.literal('assistant')]),
     metadata: zod_1.z.any().optional(),
     parts: zod_1.z.array(zod_1.z.any()),
 });
 let AiMessageDocument = class AiMessageDocument {
+    content;
 };
 exports.AiMessageDocument = AiMessageDocument;
+__decorate([
+    (0, common_2.Input)({
+        schema: exports.AiMessageDocumentSchema,
+    }),
+    __metadata("design:type", Object)
+], AiMessageDocument.prototype, "content", void 0);
 exports.AiMessageDocument = AiMessageDocument = __decorate([
     (0, common_1.Injectable)(),
     (0, common_2.Document)({
@@ -27,7 +37,6 @@ exports.AiMessageDocument = AiMessageDocument = __decorate([
             description: 'Ai Message Document.',
         },
         configFile: __dirname + '/ai-message-document.yaml',
-    }),
-    (0, common_2.WithArguments)(AiMessageDocumentSchema)
+    })
 ], AiMessageDocument);
 //# sourceMappingURL=ai-message-document.js.map

package/dist/documents/ai-message-document.js.map

@@ -1 +1 @@
-{"version":3,"file":"ai-message-document.js","sourceRoot":"","sources":["../../src/documents/ai-message-document.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA4C;AAC5C,6BAAwB;AACxB,8CAA4D;AAE5D,MAAM,uBAAuB,GAAG,OAAC,CAAC,MAAM,CAAC;IACvC,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACzB,IAAI,EAAE,OAAC,CAAC,KAAK,CAAC,CAAC,OAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,OAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,OAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;IAC/E,QAAQ,EAAE,OAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;IAC5B,KAAK,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC;CACxB,CAAC,CAAC;AAWI,IAAM,iBAAiB,GAAvB,MAAM,iBAAiB;CAAG,CAAA;AAApB,8CAAiB;4BAAjB,iBAAiB;IAT7B,IAAA,mBAAU,GAAE;IACZ,IAAA,iBAAQ,EAAC;QACR,MAAM,EAAE;YACN,IAAI,EAAE,UAAU;YAChB,WAAW,EAAE,sBAAsB;SACpC;QACD,UAAU,EAAE,SAAS,GAAG,2BAA2B;KACpD,CAAC;IACD,IAAA,sBAAa,EAAC,uBAAuB,CAAC;GAC1B,iBAAiB,CAAG"}
+{"version":3,"file":"ai-message-document.js","sourceRoot":"","sources":["../../src/documents/ai-message-document.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA4C;AAC5C,6BAAwB;AACxB,8CAAoD;AAEvC,QAAA,uBAAuB,GAAG,OAAC,CAAC,MAAM,CAAC;IAC9C,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACzB,IAAI,EAAE,OAAC,CAAC,KAAK,CAAC,CAAC,OAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,OAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,OAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;IAC/E,QAAQ,EAAE,OAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;IAC5B,KAAK,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC;CACxB,CAAC,CAAC;AAYI,IAAM,iBAAiB,GAAvB,MAAM,iBAAiB;IAI5B,OAAO,CAA0C;CAClD,CAAA;AALY,8CAAiB;AAI5B;IAHC,IAAA,cAAK,EAAC;QACL,MAAM,EAAE,+BAAuB;KAChC,CAAC;;kDAC+C;4BAJtC,iBAAiB;IAR7B,IAAA,mBAAU,GAAE;IACZ,IAAA,iBAAQ,EAAC;QACR,MAAM,EAAE;YACN,IAAI,EAAE,UAAU;YAChB,WAAW,EAAE,sBAAsB;SACpC;QACD,UAAU,EAAE,SAAS,GAAG,2BAA2B;KACpD,CAAC;GACW,iBAAiB,CAK7B"}

package/dist/tools/ai-generate-document.tool.d.ts

@@ -1,9 +1,10 @@
-import { ToolInterface, ToolResult, WorkflowExecution, WorkflowInterface } from '@loopstack/common';
+import { RunContext, ToolInterface, ToolResult, WorkflowInterface, WorkflowMetadataInterface } from '@loopstack/common';
 import { AiGenerateObjectArgsType } from './ai-generate-object.tool';
 export declare class AiGenerateDocument implements ToolInterface<AiGenerateObjectArgsType> {
     private aiGenerateObject;
     private createDocument;
+    args: AiGenerateObjectArgsType;
     private getRequiredTool;
-    execute(args: AiGenerateObjectArgsType, ctx: WorkflowExecution<any>, parent: WorkflowInterface): Promise<ToolResult>;
+    execute(args: AiGenerateObjectArgsType, ctx: RunContext, parent: WorkflowInterface, metadata: WorkflowMetadataInterface): Promise<ToolResult>;
 }
 //# sourceMappingURL=ai-generate-document.tool.d.ts.map

package/dist/tools/ai-generate-document.tool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ai-generate-document.tool.d.ts","sourceRoot":"","sources":["../../src/tools/ai-generate-document.tool.ts"],"names":[],"mappings":"AACA,OAAO,EAGL,aAAa,EACb,UAAU,EAEV,iBAAiB,EACjB,iBAAiB,EAElB,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EAAoB,wBAAwB,EAA0B,MAAM,2BAA2B,CAAC;AAE/G,qBAOa,kBAAmB,YAAW,aAAa,CAAC,wBAAwB,CAAC;IAClE,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,cAAc,CAAkB;IAEtD,OAAO,CAAC,eAAe;IAQjB,OAAO,CACX,IAAI,EAAE,wBAAwB,EAC9B,GAAG,EAAE,iBAAiB,CAAC,GAAG,CAAC,EAC3B,MAAM,EAAE,iBAAiB,GACxB,OAAO,CAAC,UAAU,CAAC;CAcvB"}
+{"version":3,"file":"ai-generate-document.tool.d.ts","sourceRoot":"","sources":["../../src/tools/ai-generate-document.tool.ts"],"names":[],"mappings":"AACA,OAAO,EAGL,UAAU,EAEV,aAAa,EACb,UAAU,EACV,iBAAiB,EACjB,yBAAyB,EAE1B,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EAAoB,wBAAwB,EAA0B,MAAM,2BAA2B,CAAC;AAE/G,qBAMa,kBAAmB,YAAW,aAAa,CAAC,wBAAwB,CAAC;IAClE,OAAO,CAAC,gBAAgB,CAAoB;IAC5C,OAAO,CAAC,cAAc,CAAkB;IAKtD,IAAI,EAAE,wBAAwB,CAAC;IAE/B,OAAO,CAAC,eAAe;IAQjB,OAAO,CACX,IAAI,EAAE,wBAAwB,EAC9B,GAAG,EAAE,UAAU,EACf,MAAM,EAAE,iBAAiB,EACzB,QAAQ,EAAE,yBAAyB,GAClC,OAAO,CAAC,UAAU,CAAC;CAevB"}

package/dist/tools/ai-generate-document.tool.js

@@ -17,6 +17,7 @@ const ai_generate_object_tool_1 = require("./ai-generate-object.tool");
 let AiGenerateDocument = class AiGenerateDocument {
     aiGenerateObject;
     createDocument;
+    args;
     getRequiredTool(name) {
         const tool = (0, common_2.getBlockTool)(this, name);
         if (tool === undefined) {
@@ -24,15 +25,15 @@ let AiGenerateDocument = class AiGenerateDocument {
         }
         return tool;
     }
-    async execute(args, ctx, parent) {
-        const result = await this.getRequiredTool('aiGenerateObject').execute(args, ctx, parent);
+    async execute(args, ctx, parent, metadata) {
+        const result = await this.getRequiredTool('aiGenerateObject').execute(args, ctx, parent, metadata);
         return this.getRequiredTool('createDocument').execute({
             id: args.response.id,
             document: args.response.document,
             update: {
                 content: result.data,
             },
-        }, ctx, parent);
+        }, ctx, parent, metadata);
     }
 };
 exports.AiGenerateDocument = AiGenerateDocument;
@@ -44,13 +45,18 @@ __decorate([
     (0, common_2.InjectTool)(),
     __metadata("design:type", core_ui_module_1.CreateDocument)
 ], AiGenerateDocument.prototype, "createDocument", void 0);
+__decorate([
+    (0, common_2.Input)({
+        schema: ai_generate_object_tool_1.AiGenerateObjectSchema,
+    }),
+    __metadata("design:type", Object)
+], AiGenerateDocument.prototype, "args", void 0);
 exports.AiGenerateDocument = AiGenerateDocument = __decorate([
     (0, common_1.Injectable)(),
     (0, common_2.Tool)({
         config: {
             description: 'Generates a structured object using a LLM and creates it as document',
         },
-    }),
-    (0, common_2.WithArguments)(ai_generate_object_tool_1.AiGenerateObjectSchema)
+    })
 ], AiGenerateDocument);
 //# sourceMappingURL=ai-generate-document.tool.js.map

package/dist/tools/ai-generate-document.tool.js.map

@@ -1 +1 @@
-{"version":3,"file":"ai-generate-document.tool.js","sourceRoot":"","sources":["../../src/tools/ai-generate-document.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA4C;AAC5C,8CAS2B;AAC3B,8DAA2D;AAC3D,uEAA+G;AASxG,IAAM,kBAAkB,GAAxB,MAAM,kBAAkB;IACP,gBAAgB,CAAoB;IACpC,cAAc,CAAkB;IAE9C,eAAe,CAAC,IAAY;QAClC,MAAM,IAAI,GAAG,IAAA,qBAAY,EAAgB,IAAI,EAAE,IAAI,CAAC,CAAC;QACrD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,SAAS,IAAI,oBAAoB,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,OAAO,CACX,IAA8B,EAC9B,GAA2B,EAC3B,MAAyB;QAEzB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACzF,OAAO,IAAI,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC,OAAO,CACnD;YACE,EAAE,EAAE,IAAI,CAAC,QAAQ,CAAC,EAAE;YACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ;YAChC,MAAM,EAAE;gBACN,OAAO,EAAE,MAAM,CAAC,IAAe;aAChC;SACF,EACD,GAAG,EACH,MAAM,CACP,CAAC;IACJ,CAAC;CACF,CAAA;AA9BY,gDAAkB;AACP;IAArB,IAAA,mBAAU,GAAE;8BAA4B,0CAAgB;4DAAC;AACpC;IAArB,IAAA,mBAAU,GAAE;8BAA0B,+BAAc;0DAAC;6BAF3C,kBAAkB;IAP9B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,MAAM,EAAE;YACN,WAAW,EAAE,sEAAsE;SACpF;KACF,CAAC;IACD,IAAA,sBAAa,EAAC,gDAAsB,CAAC;GACzB,kBAAkB,CA8B9B"}
+{"version":3,"file":"ai-generate-document.tool.js","sourceRoot":"","sources":["../../src/tools/ai-generate-document.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA4C;AAC5C,8CAU2B;AAC3B,8DAA2D;AAC3D,uEAA+G;AAQxG,IAAM,kBAAkB,GAAxB,MAAM,kBAAkB;IACP,gBAAgB,CAAoB;IACpC,cAAc,CAAkB;IAKtD,IAAI,CAA2B;IAEvB,eAAe,CAAC,IAAY;QAClC,MAAM,IAAI,GAAG,IAAA,qBAAY,EAAgB,IAAI,EAAE,IAAI,CAAC,CAAC;QACrD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,SAAS,IAAI,oBAAoB,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,OAAO,CACX,IAA8B,EAC9B,GAAe,EACf,MAAyB,EACzB,QAAmC;QAEnC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QACnG,OAAO,IAAI,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC,OAAO,CACnD;YACE,EAAE,EAAE,IAAI,CAAC,QAAQ,CAAC,EAAE;YACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ;YAChC,MAAM,EAAE;gBACN,OAAO,EAAE,MAAM,CAAC,IAAe;aAChC;SACF,EACD,GAAG,EACH,MAAM,EACN,QAAQ,CACT,CAAC;IACJ,CAAC;CACF,CAAA;AArCY,gDAAkB;AACP;IAArB,IAAA,mBAAU,GAAE;8BAA4B,0CAAgB;4DAAC;AACpC;IAArB,IAAA,mBAAU,GAAE;8BAA0B,+BAAc;0DAAC;AAKtD;IAHC,IAAA,cAAK,EAAC;QACL,MAAM,EAAE,gDAAsB;KAC/B,CAAC;;gDAC6B;6BAPpB,kBAAkB;IAN9B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,MAAM,EAAE;YACN,WAAW,EAAE,sEAAsE;SACpF;KACF,CAAC;GACW,kBAAkB,CAqC9B"}

package/dist/tools/ai-generate-object.tool.d.ts

@@ -1,7 +1,5 @@
 import { z } from 'zod';
-import { ToolInterface, ToolResult, WorkflowExecution, WorkflowInterface } from '@loopstack/common';
-import { AiMessagesHelperService } from '../services';
-import { AiProviderModelHelperService } from '../services';
+import { RunContext, ToolInterface, ToolResult, WorkflowInterface, WorkflowMetadataInterface } from '@loopstack/common';
 export declare const AiGenerateObjectSchema: z.ZodObject<{
     llm: z.ZodOptional<z.ZodObject<{
         model: z.ZodOptional<z.ZodString>;
@@ -29,8 +27,8 @@ export type AiGenerateObjectArgsType = z.infer<typeof AiGenerateObjectSchema>;
 export declare class AiGenerateObject implements ToolInterface<AiGenerateObjectArgsType> {
     private readonly aiMessagesHelperService;
     private readonly aiProviderModelHelperService;
-    constructor(aiMessagesHelperService: AiMessagesHelperService, aiProviderModelHelperService: AiProviderModelHelperService);
-    execute(args: AiGenerateObjectArgsType, ctx: WorkflowExecution<any>, parent: WorkflowInterface): Promise<ToolResult>;
+    args: AiGenerateObjectArgsType;
+    execute(args: AiGenerateObjectArgsType, ctx: RunContext, parent: WorkflowInterface, metadata: WorkflowMetadataInterface): Promise<ToolResult>;
     private handleGenerateObject;
 }
 //# sourceMappingURL=ai-generate-object.tool.d.ts.map

package/dist/tools/ai-generate-object.tool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ai-generate-object.tool.d.ts","sourceRoot":"","sources":["../../src/tools/ai-generate-object.tool.ts"],"names":[],"mappings":"AAWA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAGL,aAAa,EACb,UAAU,EAEV,iBAAiB,EACjB,iBAAiB,EAGlB,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EAAE,uBAAuB,EAAE,MAAM,aAAa,CAAC;AACtD,OAAO,EAAE,4BAA4B,EAAE,MAAM,aAAa,CAAC;AAE3D,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;kBAKxB,CAAC;AAEZ,MAAM,MAAM,wBAAwB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAE9E,qBAOa,gBAAiB,YAAW,aAAa,CAAC,wBAAwB,CAAC;IAE5E,OAAO,CAAC,QAAQ,CAAC,uBAAuB;IACxC,OAAO,CAAC,QAAQ,CAAC,4BAA4B;gBAD5B,uBAAuB,EAAE,uBAAuB,EAChD,4BAA4B,EAAE,4BAA4B;IAGvE,OAAO,CACX,IAAI,EAAE,wBAAwB,EAC9B,GAAG,EAAE,iBAAiB,CAAC,GAAG,CAAC,EAC3B,MAAM,EAAE,iBAAiB,GACxB,OAAO,CAAC,UAAU,CAAC;YA4CR,oBAAoB;CAsBnC"}
+{"version":3,"file":"ai-generate-object.tool.d.ts","sourceRoot":"","sources":["../../src/tools/ai-generate-object.tool.ts"],"names":[],"mappings":"AAWA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAGL,UAAU,EAEV,aAAa,EACb,UAAU,EACV,iBAAiB,EACjB,yBAAyB,EAG1B,MAAM,mBAAmB,CAAC;AAK3B,eAAO,MAAM,sBAAsB;;;;;;;;;;;;;;;;;;;;;;kBAKxB,CAAC;AAEZ,MAAM,MAAM,wBAAwB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAE9E,qBAMa,gBAAiB,YAAW,aAAa,CAAC,wBAAwB,CAAC;IAE9E,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAA0B;IAGlE,OAAO,CAAC,QAAQ,CAAC,4BAA4B,CAA+B;IAK5E,IAAI,EAAE,wBAAwB,CAAC;IAEzB,OAAO,CACX,IAAI,EAAE,wBAAwB,EAC9B,GAAG,EAAE,UAAU,EACf,MAAM,EAAE,iBAAiB,EACzB,QAAQ,EAAE,yBAAyB,GAClC,OAAO,CAAC,UAAU,CAAC;YA0CR,oBAAoB;CAsBnC"}

package/dist/tools/ai-generate-object.tool.js

@@ -26,11 +26,8 @@ exports.AiGenerateObjectSchema = ai_generate_tool_base_schema_1.AiGenerateToolBa
 let AiGenerateObject = class AiGenerateObject {
     aiMessagesHelperService;
     aiProviderModelHelperService;
-    constructor(aiMessagesHelperService, aiProviderModelHelperService) {
-        this.aiMessagesHelperService = aiMessagesHelperService;
-        this.aiProviderModelHelperService = aiProviderModelHelperService;
-    }
-    async execute(args, ctx, parent) {
+    args;
+    async execute(args, ctx, parent, metadata) {
         const model = this.aiProviderModelHelperService.getProviderModel(args.llm);
         const options = {
             messages: [],
@@ -42,7 +39,7 @@ let AiGenerateObject = class AiGenerateObject {
             });
         }
         else {
-            const messages = this.aiMessagesHelperService.getMessages(ctx.state.getMetadata('documents'), {
+            const messages = this.aiMessagesHelperService.getMessages(metadata.documents, {
                 messages: args.messages,
                 messagesSearchTag: args.messagesSearchTag,
             });
@@ -58,7 +55,6 @@ let AiGenerateObject = class AiGenerateObject {
         }
         options.schema = responseSchema;
         const response = await this.handleGenerateObject(model, options);
-        console.log(response);
         return {
             data: response.output,
         };
@@ -81,15 +77,26 @@ let AiGenerateObject = class AiGenerateObject {
     }
 };
 exports.AiGenerateObject = AiGenerateObject;
+__decorate([
+    (0, common_1.Inject)(),
+    __metadata("design:type", services_1.AiMessagesHelperService)
+], AiGenerateObject.prototype, "aiMessagesHelperService", void 0);
+__decorate([
+    (0, common_1.Inject)(),
+    __metadata("design:type", services_2.AiProviderModelHelperService)
+], AiGenerateObject.prototype, "aiProviderModelHelperService", void 0);
+__decorate([
+    (0, common_2.Input)({
+        schema: exports.AiGenerateObjectSchema,
+    }),
+    __metadata("design:type", Object)
+], AiGenerateObject.prototype, "args", void 0);
 exports.AiGenerateObject = AiGenerateObject = __decorate([
     (0, common_1.Injectable)(),
     (0, common_2.Tool)({
         config: {
             description: 'Generates a structured object using a LLM',
         },
-    }),
-    (0, common_2.WithArguments)(exports.AiGenerateObjectSchema),
-    __metadata("design:paramtypes", [services_1.AiMessagesHelperService,
-        services_2.AiProviderModelHelperService])
+    })
 ], AiGenerateObject);
 //# sourceMappingURL=ai-generate-object.tool.js.map

package/dist/tools/ai-generate-object.tool.js.map

@@ -1 +1 @@
-{"version":3,"file":"ai-generate-object.tool.js","sourceRoot":"","sources":["../../src/tools/ai-generate-object.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,2CAA4C;AAC5C,2BAQY;AACZ,6BAAwB;AACxB,8CAU2B;AAC3B,0FAAmF;AACnF,0CAAsD;AACtD,0CAA2D;AAE9C,QAAA,sBAAsB,GAAG,uDAAwB,CAAC,MAAM,CAAC;IACpE,QAAQ,EAAE,OAAC,CAAC,MAAM,CAAC;QACjB,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACzB,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;KACrB,CAAC;CACH,CAAC,CAAC,MAAM,EAAE,CAAC;AAWL,IAAM,gBAAgB,GAAtB,MAAM,gBAAgB;IAER;IACA;IAFnB,YACmB,uBAAgD,EAChD,4BAA0D;QAD1D,4BAAuB,GAAvB,uBAAuB,CAAyB;QAChD,iCAA4B,GAA5B,4BAA4B,CAA8B;IAC1E,CAAC;IAEJ,KAAK,CAAC,OAAO,CACX,IAA8B,EAC9B,GAA2B,EAC3B,MAAyB;QAEzB,MAAM,KAAK,GAAG,IAAI,CAAC,4BAA4B,CAAC,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE3E,MAAM,OAAO,GAGT;YACF,QAAQ,EAAE,EAAE;SACb,CAAC;QAEF,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACpB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,IAAI,CAAC,MAAM;aACL,CAAC,CAAC;QACrB,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;gBAC5F,QAAQ,EAAE,IAAI,CAAC,QAAkC;gBACjD,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;aAC1C,CAAC,CAAC;YAEH,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAA,2BAAsB,EAAC,QAAQ,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,QAAQ,GAAG,IAAA,yBAAgB,EAAoB,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACrF,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,IAAI,CAAC,QAAQ,CAAC,QAAQ,wCAAwC,CAAC,CAAC;QACzG,CAAC;QACD,MAAM,cAAc,GAAG,IAAA,2BAAkB,EAAC,QAAQ,CAAC,CAAC;QACpD,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAC9E,CAAC;QAED,OAAO,CAAC,MAAM,GAAG,cAAc,CAAC;QAEhC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAEjE,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAEtB,OAAO;YACL,IAAI,EAAE,QAAQ,CAAC,MAAM;SACtB,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAChC,KAAoB,EACpB,OAGC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAE5B,OAAO,IAAA,iBAAY,EAAC;gBAClB,KAAK;gBACL,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,WAAM,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACjF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACxD,OAAO,CAAC,KAAK,CAAC,wBAAwB,iBAAiB,KAAK,EAAE,KAAK,CAAC,CAAC;YACrE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAA;AA5EY,4CAAgB;2BAAhB,gBAAgB;IAP5B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,MAAM,EAAE;YACN,WAAW,EAAE,2CAA2C;SACzD;KACF,CAAC;IACD,IAAA,sBAAa,EAAC,8BAAsB,CAAC;qCAGQ,kCAAuB;QAClB,uCAA4B;GAHlE,gBAAgB,CA4E5B"}
+{"version":3,"file":"ai-generate-object.tool.js","sourceRoot":"","sources":["../../src/tools/ai-generate-object.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,2CAAoD;AACpD,2BAQY;AACZ,6BAAwB;AACxB,8CAW2B;AAC3B,0FAAmF;AACnF,0CAAsD;AACtD,0CAA2D;AAE9C,QAAA,sBAAsB,GAAG,uDAAwB,CAAC,MAAM,CAAC;IACpE,QAAQ,EAAE,OAAC,CAAC,MAAM,CAAC;QACjB,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACzB,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;KACrB,CAAC;CACH,CAAC,CAAC,MAAM,EAAE,CAAC;AAUL,IAAM,gBAAgB,GAAtB,MAAM,gBAAgB;IAEV,uBAAuB,CAA0B;IAGjD,4BAA4B,CAA+B;IAK5E,IAAI,CAA2B;IAE/B,KAAK,CAAC,OAAO,CACX,IAA8B,EAC9B,GAAe,EACf,MAAyB,EACzB,QAAmC;QAEnC,MAAM,KAAK,GAAG,IAAI,CAAC,4BAA4B,CAAC,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE3E,MAAM,OAAO,GAGT;YACF,QAAQ,EAAE,EAAE;SACb,CAAC;QAEF,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACpB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,IAAI,CAAC,MAAM;aACL,CAAC,CAAC;QACrB,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,WAAW,CAAC,QAAQ,CAAC,SAAS,EAAE;gBAC5E,QAAQ,EAAE,IAAI,CAAC,QAAkC;gBACjD,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;aAC1C,CAAC,CAAC;YAEH,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAA,2BAAsB,EAAC,QAAQ,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,QAAQ,GAAG,IAAA,yBAAgB,EAAoB,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACrF,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,IAAI,CAAC,QAAQ,CAAC,QAAQ,wCAAwC,CAAC,CAAC;QACzG,CAAC;QACD,MAAM,cAAc,GAAG,IAAA,2BAAkB,EAAC,QAAQ,CAAC,CAAC;QACpD,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAC9E,CAAC;QAED,OAAO,CAAC,MAAM,GAAG,cAAc,CAAC;QAEhC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAEjE,OAAO;YACL,IAAI,EAAE,QAAQ,CAAC,MAAM;SACtB,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAChC,KAAoB,EACpB,OAGC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAE5B,OAAO,IAAA,iBAAY,EAAC;gBAClB,KAAK;gBACL,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,WAAM,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACjF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACxD,OAAO,CAAC,KAAK,CAAC,wBAAwB,iBAAiB,KAAK,EAAE,KAAK,CAAC,CAAC;YACrE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAA;AAjFY,4CAAgB;AAEV;IADhB,IAAA,eAAM,GAAE;8BACiC,kCAAuB;iEAAC;AAGjD;IADhB,IAAA,eAAM,GAAE;8BACsC,uCAA4B;sEAAC;AAK5E;IAHC,IAAA,cAAK,EAAC;QACL,MAAM,EAAE,8BAAsB;KAC/B,CAAC;;8CAC6B;2BAVpB,gBAAgB;IAN5B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,MAAM,EAAE;YACN,WAAW,EAAE,2CAA2C;SACzD;KACF,CAAC;GACW,gBAAgB,CAiF5B"}

package/dist/tools/ai-generate-text.tool.d.ts

@@ -1,8 +1,5 @@
 import { z } from 'zod';
-import { ToolInterface, ToolResult, WorkflowExecution, WorkflowInterface } from '@loopstack/common';
-import { AiMessagesHelperService } from '../services';
-import { AiProviderModelHelperService } from '../services';
-import { AiToolsHelperService } from '../services';
+import { RunContext, ToolInterface, ToolResult, WorkflowInterface, WorkflowMetadataInterface } from '@loopstack/common';
 export declare const AiGenerateTextSchema: z.ZodObject<{
     llm: z.ZodOptional<z.ZodObject<{
         model: z.ZodOptional<z.ZodString>;
@@ -28,8 +25,8 @@ export declare class AiGenerateText implements ToolInterface<AiGenerateTextArgsT
     private readonly aiMessagesHelperService;
     private readonly aiToolsHelperService;
     private readonly aiProviderModelHelperService;
-    constructor(aiMessagesHelperService: AiMessagesHelperService, aiToolsHelperService: AiToolsHelperService, aiProviderModelHelperService: AiProviderModelHelperService);
-    execute(args: AiGenerateTextArgsType, ctx: WorkflowExecution<any>, parent: WorkflowInterface): Promise<ToolResult>;
+    args: AiGenerateTextArgsType;
+    execute(args: AiGenerateTextArgsType, ctx: RunContext, parent: WorkflowInterface, runtime: WorkflowMetadataInterface): Promise<ToolResult>;
     private handleGenerateText;
 }
 export {};

package/dist/tools/ai-generate-text.tool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"ai-generate-text.tool.d.ts","sourceRoot":"","sources":["../../src/tools/ai-generate-text.tool.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAEL,aAAa,EACb,UAAU,EAEV,iBAAiB,EACjB,iBAAiB,EAClB,MAAM,mBAAmB,CAAC;AAE3B,OAAO,EAAE,uBAAuB,EAAE,MAAM,aAAa,CAAC;AACtD,OAAO,EAAE,4BAA4B,EAAE,MAAM,aAAa,CAAC;AAC3D,OAAO,EAAE,oBAAoB,EAAE,MAAM,aAAa,CAAC;AAEnD,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;kBAEtB,CAAC;AAEZ,KAAK,sBAAsB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAEnE,qBAOa,cAAe,YAAW,aAAa,CAAC,sBAAsB,CAAC;IAExE,OAAO,CAAC,QAAQ,CAAC,uBAAuB;IACxC,OAAO,CAAC,QAAQ,CAAC,oBAAoB;IACrC,OAAO,CAAC,QAAQ,CAAC,4BAA4B;gBAF5B,uBAAuB,EAAE,uBAAuB,EAChD,oBAAoB,EAAE,oBAAoB,EAC1C,4BAA4B,EAAE,4BAA4B;IAGvE,OAAO,CACX,IAAI,EAAE,sBAAsB,EAC5B,GAAG,EAAE,iBAAiB,CAAC,GAAG,CAAC,EAC3B,MAAM,EAAE,iBAAiB,GACxB,OAAO,CAAC,UAAU,CAAC;YAoCR,kBAAkB;CAgDjC"}
+{"version":3,"file":"ai-generate-text.tool.d.ts","sourceRoot":"","sources":["../../src/tools/ai-generate-text.tool.ts"],"names":[],"mappings":"AAGA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAEL,UAAU,EAEV,aAAa,EACb,UAAU,EACV,iBAAiB,EACjB,yBAAyB,EAC1B,MAAM,mBAAmB,CAAC;AAM3B,eAAO,MAAM,oBAAoB;;;;;;;;;;;;;;;;;;;kBAEtB,CAAC;AAEZ,KAAK,sBAAsB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,oBAAoB,CAAC,CAAC;AAEnE,qBAMa,cAAe,YAAW,aAAa,CAAC,sBAAsB,CAAC;IAE1E,OAAO,CAAC,QAAQ,CAAC,uBAAuB,CAA0B;IAElE,OAAO,CAAC,QAAQ,CAAC,oBAAoB,CAAuB;IAE5D,OAAO,CAAC,QAAQ,CAAC,4BAA4B,CAA+B;IAK5E,IAAI,EAAE,sBAAsB,CAAC;IAEvB,OAAO,CACX,IAAI,EAAE,sBAAsB,EAC5B,GAAG,EAAE,UAAU,EACf,MAAM,EAAE,iBAAiB,EACzB,OAAO,EAAE,yBAAyB,GACjC,OAAO,CAAC,UAAU,CAAC;YAoCR,kBAAkB;CAgDjC"}

package/dist/tools/ai-generate-text.tool.js

@@ -25,12 +25,8 @@ let AiGenerateText = class AiGenerateText {
     aiMessagesHelperService;
     aiToolsHelperService;
     aiProviderModelHelperService;
-    constructor(aiMessagesHelperService, aiToolsHelperService, aiProviderModelHelperService) {
-        this.aiMessagesHelperService = aiMessagesHelperService;
-        this.aiToolsHelperService = aiToolsHelperService;
-        this.aiProviderModelHelperService = aiProviderModelHelperService;
-    }
-    async execute(args, ctx, parent) {
+    args;
+    async execute(args, ctx, parent, runtime) {
         const model = this.aiProviderModelHelperService.getProviderModel(args.llm);
         const options = {
             messages: [],
@@ -43,7 +39,7 @@ let AiGenerateText = class AiGenerateText {
             });
         }
         else {
-            const messages = this.aiMessagesHelperService.getMessages(ctx.state.getMetadata('documents'), {
+            const messages = this.aiMessagesHelperService.getMessages(runtime.documents, {
                 messages: args.messages,
                 messagesSearchTag: args.messagesSearchTag,
             });
@@ -97,16 +93,30 @@ let AiGenerateText = class AiGenerateText {
     }
 };
 exports.AiGenerateText = AiGenerateText;
+__decorate([
+    (0, common_1.Inject)(),
+    __metadata("design:type", services_1.AiMessagesHelperService)
+], AiGenerateText.prototype, "aiMessagesHelperService", void 0);
+__decorate([
+    (0, common_1.Inject)(),
+    __metadata("design:type", services_3.AiToolsHelperService)
+], AiGenerateText.prototype, "aiToolsHelperService", void 0);
+__decorate([
+    (0, common_1.Inject)(),
+    __metadata("design:type", services_2.AiProviderModelHelperService)
+], AiGenerateText.prototype, "aiProviderModelHelperService", void 0);
+__decorate([
+    (0, common_2.Input)({
+        schema: exports.AiGenerateTextSchema,
+    }),
+    __metadata("design:type", Object)
+], AiGenerateText.prototype, "args", void 0);
 exports.AiGenerateText = AiGenerateText = __decorate([
     (0, common_1.Injectable)(),
     (0, common_2.Tool)({
         config: {
             description: 'Generates text using a LLM',
         },
-    }),
-    (0, common_2.WithArguments)(exports.AiGenerateTextSchema),
-    __metadata("design:paramtypes", [services_1.AiMessagesHelperService,
-        services_3.AiToolsHelperService,
-        services_2.AiProviderModelHelperService])
+    })
 ], AiGenerateText);
 //# sourceMappingURL=ai-generate-text.tool.js.map

package/dist/tools/ai-generate-text.tool.js.map

@@ -1 +1 @@
-{"version":3,"file":"ai-generate-text.tool.js","sourceRoot":"","sources":["../../src/tools/ai-generate-text.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,2CAA4C;AAC5C,2BAAkH;AAClH,6BAAwB;AACxB,8CAO2B;AAC3B,0FAAmF;AACnF,0CAAsD;AACtD,0CAA2D;AAC3D,0CAAmD;AAEtC,QAAA,oBAAoB,GAAG,uDAAwB,CAAC,MAAM,CAAC;IAClE,KAAK,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;CACtC,CAAC,CAAC,MAAM,EAAE,CAAC;AAWL,IAAM,cAAc,GAApB,MAAM,cAAc;IAEN;IACA;IACA;IAHnB,YACmB,uBAAgD,EAChD,oBAA0C,EAC1C,4BAA0D;QAF1D,4BAAuB,GAAvB,uBAAuB,CAAyB;QAChD,yBAAoB,GAApB,oBAAoB,CAAsB;QAC1C,iCAA4B,GAA5B,4BAA4B,CAA8B;IAC1E,CAAC;IAEJ,KAAK,CAAC,OAAO,CACX,IAA4B,EAC5B,GAA2B,EAC3B,MAAyB;QAEzB,MAAM,KAAK,GAAG,IAAI,CAAC,4BAA4B,CAAC,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE3E,MAAM,OAAO,GAIT;YACF,QAAQ,EAAE,EAAE;SACb,CAAC;QAEF,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEhG,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACpB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,IAAI,CAAC,MAAM;aACL,CAAC,CAAC;QACrB,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;gBAC5F,QAAQ,EAAE,IAAI,CAAC,QAAkC;gBACjD,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;aAC1C,CAAC,CAAC;YAEH,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAA,2BAAsB,EAAC,QAAQ,EAAE;gBACxD,KAAK,EAAE,OAAO,CAAC,KAAgB;aAChC,CAAC,CAAC;QACL,CAAC;QAED,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAEhE,OAAO;YACL,IAAI,EAAE,SAAS;SAChB,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAC9B,KAAoB,EACpB,OAIC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAA,eAAU,EAAC;gBACxB,KAAK;gBACL,GAAG,OAAO;aACyB,CAAC,CAAC;YAEvC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACrC,MAAM,MAAM,GAAG,IAAA,0BAAqB,EAAC;oBACnC,OAAO,CAAC,EAAE,MAAM,EAAE;wBAChB,MAAM,CAAC,KAAK,CACV,MAAM,CAAC,iBAAiB,CAAC;4BACvB,aAAa,EAAE,IAAI;yBACpB,CAAC,CACH,CAAC;oBACJ,CAAC;oBACD,QAAQ,EAAE,CAAC,IAAI,EAAE,EAAE;wBACjB,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;oBAChC,CAAC;iBACF,CAAC,CAAC;gBAGH,KAAK,CAAC,KAAK,IAAI,EAAE;oBACf,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;wBAClC,OAAO,IAAI,EAAE,CAAC;4BACZ,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;4BACrC,IAAI,IAAI;gCAAE,MAAM;wBAClB,CAAC;oBACH,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,MAAM,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBACpE,CAAC;gBACH,CAAC,CAAC,EAAE,CAAC;YACP,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACxD,OAAO,CAAC,KAAK,CAAC,wBAAwB,iBAAiB,KAAK,EAAE,KAAK,CAAC,CAAC;YACrE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAA;AA/FY,wCAAc;yBAAd,cAAc;IAP1B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,MAAM,EAAE;YACN,WAAW,EAAE,4BAA4B;SAC1C;KACF,CAAC;IACD,IAAA,sBAAa,EAAC,4BAAoB,CAAC;qCAGU,kCAAuB;QAC1B,+BAAoB;QACZ,uCAA4B;GAJlE,cAAc,CA+F1B"}
+{"version":3,"file":"ai-generate-text.tool.js","sourceRoot":"","sources":["../../src/tools/ai-generate-text.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,2CAAoD;AACpD,2BAAkH;AAClH,6BAAwB;AACxB,8CAQ2B;AAC3B,0FAAmF;AACnF,0CAAsD;AACtD,0CAA2D;AAC3D,0CAAmD;AAEtC,QAAA,oBAAoB,GAAG,uDAAwB,CAAC,MAAM,CAAC;IAClE,KAAK,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,MAAM,EAAE,CAAC,CAAC,QAAQ,EAAE;CACtC,CAAC,CAAC,MAAM,EAAE,CAAC;AAUL,IAAM,cAAc,GAApB,MAAM,cAAc;IAER,uBAAuB,CAA0B;IAEjD,oBAAoB,CAAuB;IAE3C,4BAA4B,CAA+B;IAK5E,IAAI,CAAyB;IAE7B,KAAK,CAAC,OAAO,CACX,IAA4B,EAC5B,GAAe,EACf,MAAyB,EACzB,OAAkC;QAElC,MAAM,KAAK,GAAG,IAAI,CAAC,4BAA4B,CAAC,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE3E,MAAM,OAAO,GAIT;YACF,QAAQ,EAAE,EAAE;SACb,CAAC;QAEF,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC,CAAC,CAAC,CAAC,SAAS,CAAC;QAEhG,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACpB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,IAAI,CAAC,MAAM;aACL,CAAC,CAAC;QACrB,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,WAAW,CAAC,OAAO,CAAC,SAAS,EAAE;gBAC3E,QAAQ,EAAE,IAAI,CAAC,QAAkC;gBACjD,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;aAC1C,CAAC,CAAC;YAEH,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAA,2BAAsB,EAAC,QAAQ,EAAE;gBACxD,KAAK,EAAE,OAAO,CAAC,KAAgB;aAChC,CAAC,CAAC;QACL,CAAC;QAED,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,kBAAkB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAEhE,OAAO;YACL,IAAI,EAAE,SAAS;SAChB,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,kBAAkB,CAC9B,KAAoB,EACpB,OAIC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAA,eAAU,EAAC;gBACxB,KAAK;gBACL,GAAG,OAAO;aACyB,CAAC,CAAC;YAEvC,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACrC,MAAM,MAAM,GAAG,IAAA,0BAAqB,EAAC;oBACnC,OAAO,CAAC,EAAE,MAAM,EAAE;wBAChB,MAAM,CAAC,KAAK,CACV,MAAM,CAAC,iBAAiB,CAAC;4BACvB,aAAa,EAAE,IAAI;yBACpB,CAAC,CACH,CAAC;oBACJ,CAAC;oBACD,QAAQ,EAAE,CAAC,IAAI,EAAE,EAAE;wBACjB,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;oBAChC,CAAC;iBACF,CAAC,CAAC;gBAGH,KAAK,CAAC,KAAK,IAAI,EAAE;oBACf,IAAI,CAAC;wBACH,MAAM,MAAM,GAAG,MAAM,CAAC,SAAS,EAAE,CAAC;wBAClC,OAAO,IAAI,EAAE,CAAC;4BACZ,MAAM,EAAE,IAAI,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;4BACrC,IAAI,IAAI;gCAAE,MAAM;wBAClB,CAAC;oBACH,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBACf,MAAM,CAAC,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC,CAAC;oBACpE,CAAC;gBACH,CAAC,CAAC,EAAE,CAAC;YACP,CAAC,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACxD,OAAO,CAAC,KAAK,CAAC,wBAAwB,iBAAiB,KAAK,EAAE,KAAK,CAAC,CAAC;YACrE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAA;AAtGY,wCAAc;AAER;IADhB,IAAA,eAAM,GAAE;8BACiC,kCAAuB;+DAAC;AAEjD;IADhB,IAAA,eAAM,GAAE;8BAC8B,+BAAoB;4DAAC;AAE3C;IADhB,IAAA,eAAM,GAAE;8BACsC,uCAA4B;oEAAC;AAK5E;IAHC,IAAA,cAAK,EAAC;QACL,MAAM,EAAE,4BAAoB;KAC7B,CAAC;;4CAC2B;yBAXlB,cAAc;IAN1B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,MAAM,EAAE;YACN,WAAW,EAAE,4BAA4B;SAC1C;KACF,CAAC;GACW,cAAc,CAsG1B"}

package/dist/tools/delegate-tool-call.tool.d.ts

@@ -1,5 +1,5 @@
 import { z } from 'zod';
-import { ToolInterface, ToolResult, WorkflowExecution, WorkflowInterface } from '@loopstack/common';
+import { RunContext, ToolInterface, ToolResult, WorkflowInterface, WorkflowMetadataInterface } from '@loopstack/common';
 declare const DelegateToolCallsToolSchema: z.ZodObject<{
     message: z.ZodObject<{
         id: z.ZodString;
@@ -12,7 +12,8 @@ declare const DelegateToolCallsToolSchema: z.ZodObject<{
 }, z.core.$strip>;
 type DelegateToolCallsToolArgs = z.infer<typeof DelegateToolCallsToolSchema>;
 export declare class DelegateToolCall implements ToolInterface<DelegateToolCallsToolArgs> {
-    execute(args: DelegateToolCallsToolArgs, ctx: WorkflowExecution<any>, parent: WorkflowInterface): Promise<ToolResult>;
+    args: DelegateToolCallsToolArgs;
+    execute(args: DelegateToolCallsToolArgs, ctx: RunContext, parent: WorkflowInterface, runtime: WorkflowMetadataInterface): Promise<ToolResult>;
 }
 export {};
 //# sourceMappingURL=delegate-tool-call.tool.d.ts.map

package/dist/tools/delegate-tool-call.tool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"delegate-tool-call.tool.d.ts","sourceRoot":"","sources":["../../src/tools/delegate-tool-call.tool.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAEL,aAAa,EACb,UAAU,EAEV,iBAAiB,EACjB,iBAAiB,EAElB,MAAM,mBAAmB,CAAC;AAE3B,QAAA,MAAM,2BAA2B;;;;;;;;;iBAW/B,CAAC;AAEH,KAAK,yBAAyB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,2BAA2B,CAAC,CAAC;AAE7E,qBAOa,gBAAiB,YAAW,aAAa,CAAC,yBAAyB,CAAC;IACzE,OAAO,CACX,IAAI,EAAE,yBAAyB,EAC/B,GAAG,EAAE,iBAAiB,CAAC,GAAG,CAAC,EAC3B,MAAM,EAAE,iBAAiB,GACxB,OAAO,CAAC,UAAU,CAAC;CA2CvB"}
+{"version":3,"file":"delegate-tool-call.tool.d.ts","sourceRoot":"","sources":["../../src/tools/delegate-tool-call.tool.ts"],"names":[],"mappings":"AAEA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAEL,UAAU,EAEV,aAAa,EACb,UAAU,EACV,iBAAiB,EACjB,yBAAyB,EAE1B,MAAM,mBAAmB,CAAC;AAE3B,QAAA,MAAM,2BAA2B;;;;;;;;;iBAW/B,CAAC;AAEH,KAAK,yBAAyB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,2BAA2B,CAAC,CAAC;AAE7E,qBAMa,gBAAiB,YAAW,aAAa,CAAC,yBAAyB,CAAC;IAI/E,IAAI,EAAE,yBAAyB,CAAC;IAE1B,OAAO,CACX,IAAI,EAAE,yBAAyB,EAC/B,GAAG,EAAE,UAAU,EACf,MAAM,EAAE,iBAAiB,EACzB,OAAO,EAAE,yBAAyB,GACjC,OAAO,CAAC,UAAU,CAAC;CA2CvB"}

package/dist/tools/delegate-tool-call.tool.js

@@ -5,6 +5,9 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
     else for (var i = decorators.length - 1; i >= 0; i--) if (d = decorators[i]) r = (c < 3 ? d(r) : c > 3 ? d(target, key, r) : d(target, key)) || r;
     return c > 3 && r && Object.defineProperty(target, key, r), r;
 };
+var __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DelegateToolCall = void 0;
 const common_1 = require("@nestjs/common");
@@ -21,7 +24,8 @@ const DelegateToolCallsToolSchema = zod_1.z.object({
     }),
 });
 let DelegateToolCall = class DelegateToolCall {
-    async execute(args, ctx, parent) {
+    args;
+    async execute(args, ctx, parent, runtime) {
         const parts = args.message.parts;
         const resultParts = [];
         for (const part of parts) {
@@ -36,7 +40,7 @@ let DelegateToolCall = class DelegateToolCall {
             if (!tool) {
                 throw new Error(`Tool ${toolName} not found.`);
             }
-            const result = await tool.execute(part.input, ctx, parent);
+            const result = await tool.execute(part.input, ctx, parent, runtime);
             resultParts.push({
                 type: part.type,
                 toolCallId: part.toolCallId,
@@ -59,13 +63,18 @@ let DelegateToolCall = class DelegateToolCall {
     }
 };
 exports.DelegateToolCall = DelegateToolCall;
+__decorate([
+    (0, common_2.Input)({
+        schema: DelegateToolCallsToolSchema,
+    }),
+    __metadata("design:type", Object)
+], DelegateToolCall.prototype, "args", void 0);
 exports.DelegateToolCall = DelegateToolCall = __decorate([
     (0, common_1.Injectable)(),
     (0, common_2.Tool)({
         config: {
             description: 'Delegate a tool call.',
         },
-    }),
-    (0, common_2.WithArguments)(DelegateToolCallsToolSchema)
+    })
 ], DelegateToolCall);
 //# sourceMappingURL=delegate-tool-call.tool.js.map

package/dist/tools/delegate-tool-call.tool.js.map

@@ -1 +1 @@
-{"version":3,"file":"delegate-tool-call.tool.js","sourceRoot":"","sources":["../../src/tools/delegate-tool-call.tool.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA4C;AAE5C,6BAAwB;AACxB,8CAQ2B;AAE3B,MAAM,2BAA2B,GAAG,OAAC,CAAC,MAAM,CAAC;IAC3C,OAAO,EAAE,OAAC,CAAC,MAAM,CAAC;QAChB,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE;QACd,KAAK,EAAE,OAAC,CAAC,KAAK,CACZ,OAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;YAChB,KAAK,EAAE,OAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;YACzB,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;SAClC,CAAC,CACH;KACF,CAAC;CACH,CAAC,CAAC;AAWI,IAAM,gBAAgB,GAAtB,MAAM,gBAAgB;IAC3B,KAAK,CAAC,OAAO,CACX,IAA+B,EAC/B,GAA2B,EAC3B,MAAyB;QAEzB,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;QACjC,MAAM,WAAW,GAAiB,EAAE,CAAC;QAErC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;gBACnC,SAAS;YACX,CAAC;YAED,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;gBACrB,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;YAC5C,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YAEjD,MAAM,IAAI,GAAG,IAAA,qBAAY,EAAgB,MAAM,EAAE,QAAQ,CAAC,CAAC;YAC3D,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,IAAI,KAAK,CAAC,QAAQ,QAAQ,aAAa,CAAC,CAAC;YACjD,CAAC;YACD,MAAM,MAAM,GAAe,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAgC,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;YAElG,WAAW,CAAC,IAAI,CAAC;gBACf,IAAI,EAAE,IAAI,CAAC,IAA0B;gBACrC,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,MAAM,EAAE;oBACN,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,MAAM;oBAC3B,KAAK,EAAE,MAAM,CAAC,IAAe;iBAC9B;gBACD,KAAK,EAAE,IAAI,CAAC,KAAgC;gBAC5C,KAAK,EAAE,kBAAkB;aACL,CAAC,CAAC;QAC1B,CAAC;QAED,MAAM,aAAa,GAAc;YAC/B,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,EAAE;YACnB,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,WAAW;SACnB,CAAC;QAEF,OAAO;YACL,IAAI,EAAE,aAAa;SACpB,CAAC;IACJ,CAAC;CACF,CAAA;AAhDY,4CAAgB;2BAAhB,gBAAgB;IAP5B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,MAAM,EAAE;YACN,WAAW,EAAE,uBAAuB;SACrC;KACF,CAAC;IACD,IAAA,sBAAa,EAAC,2BAA2B,CAAC;GAC9B,gBAAgB,CAgD5B"}
+{"version":3,"file":"delegate-tool-call.tool.js","sourceRoot":"","sources":["../../src/tools/delegate-tool-call.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA4C;AAE5C,6BAAwB;AACxB,8CAS2B;AAE3B,MAAM,2BAA2B,GAAG,OAAC,CAAC,MAAM,CAAC;IAC3C,OAAO,EAAE,OAAC,CAAC,MAAM,CAAC;QAChB,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE;QACd,KAAK,EAAE,OAAC,CAAC,KAAK,CACZ,OAAC,CAAC,MAAM,CAAC;YACP,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE;YAChB,KAAK,EAAE,OAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;YACzB,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;SAClC,CAAC,CACH;KACF,CAAC;CACH,CAAC,CAAC;AAUI,IAAM,gBAAgB,GAAtB,MAAM,gBAAgB;IAI3B,IAAI,CAA4B;IAEhC,KAAK,CAAC,OAAO,CACX,IAA+B,EAC/B,GAAe,EACf,MAAyB,EACzB,OAAkC;QAElC,MAAM,KAAK,GAAG,IAAI,CAAC,OAAO,CAAC,KAAK,CAAC;QACjC,MAAM,WAAW,GAAiB,EAAE,CAAC;QAErC,KAAK,MAAM,IAAI,IAAI,KAAK,EAAE,CAAC;YACzB,IAAI,CAAC,IAAI,CAAC,IAAI,CAAC,UAAU,CAAC,OAAO,CAAC,EAAE,CAAC;gBACnC,SAAS;YACX,CAAC;YAED,IAAI,CAAC,IAAI,CAAC,UAAU,EAAE,CAAC;gBACrB,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;YAC5C,CAAC;YAED,MAAM,QAAQ,GAAG,IAAI,CAAC,IAAI,CAAC,OAAO,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;YAEjD,MAAM,IAAI,GAAG,IAAA,qBAAY,EAAgB,MAAM,EAAE,QAAQ,CAAC,CAAC;YAC3D,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,IAAI,KAAK,CAAC,QAAQ,QAAQ,aAAa,CAAC,CAAC;YACjD,CAAC;YACD,MAAM,MAAM,GAAe,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAgC,EAAE,GAAG,EAAE,MAAM,EAAE,OAAO,CAAC,CAAC;YAE3G,WAAW,CAAC,IAAI,CAAC;gBACf,IAAI,EAAE,IAAI,CAAC,IAA0B;gBACrC,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,MAAM,EAAE;oBACN,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,MAAM;oBAC3B,KAAK,EAAE,MAAM,CAAC,IAAe;iBAC9B;gBACD,KAAK,EAAE,IAAI,CAAC,KAAgC;gBAC5C,KAAK,EAAE,kBAAkB;aACL,CAAC,CAAC;QAC1B,CAAC;QAED,MAAM,aAAa,GAAc;YAC/B,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,EAAE;YACnB,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,WAAW;SACnB,CAAC;QAEF,OAAO;YACL,IAAI,EAAE,aAAa;SACpB,CAAC;IACJ,CAAC;CACF,CAAA;AAtDY,4CAAgB;AAI3B;IAHC,IAAA,cAAK,EAAC;QACL,MAAM,EAAE,2BAA2B;KACpC,CAAC;;8CAC8B;2BAJrB,gBAAgB;IAN5B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,MAAM,EAAE;YACN,WAAW,EAAE,uBAAuB;SACrC;KACF,CAAC;GACW,gBAAgB,CAsD5B"}

package/package.json

@@ -10,7 +10,7 @@
     "openai",
     "tool"
   ],
-  "version": "0.19.0",
+  "version": "0.20.0",
   "license": "Apache-2.0",
   "author": {
     "name": "Jakob Klippel",
@@ -34,8 +34,8 @@
     "@ai-sdk/anthropic": "^3.0.23",
     "@ai-sdk/openai": "^3.0.19",
     "@ai-sdk/provider-utils": "^4.0.9",
-    "@loopstack/common": "^0.19.0",
-    "@loopstack/core-ui-module": "^0.19.0",
+    "@loopstack/common": "^0.20.0",
+    "@loopstack/core-ui-module": "^0.20.0",
     "@nestjs/common": "^11.1.12",
     "@nestjs/config": "^4.0.2",
     "@nestjs/core": "^11.1.12",

package/src/documents/ai-message-document.ts

@@ -1,14 +1,16 @@
 import { Injectable } from '@nestjs/common';
 import { z } from 'zod';
-import { Document, WithArguments } from '@loopstack/common';
+import { Document, Input } from '@loopstack/common';
 
-const AiMessageDocumentSchema = z.object({
+export const AiMessageDocumentSchema = z.object({
   id: z.string().optional(),
   role: z.union([z.literal('system'), z.literal('user'), z.literal('assistant')]),
   metadata: z.any().optional(),
   parts: z.array(z.any()),
 });
 
+export type AiMessageDocumentContentType = z.infer<typeof AiMessageDocumentSchema>;
+
 @Injectable()
 @Document({
   config: {
@@ -17,5 +19,9 @@ const AiMessageDocumentSchema = z.object({
   },
   configFile: __dirname + '/ai-message-document.yaml',
 })
-@WithArguments(AiMessageDocumentSchema)
-export class AiMessageDocument {}
+export class AiMessageDocument {
+  @Input({
+    schema: AiMessageDocumentSchema,
+  })
+  content: z.infer<typeof AiMessageDocumentSchema>;
+}

package/src/tools/ai-generate-document.tool.ts

@@ -1,12 +1,13 @@
 import { Injectable } from '@nestjs/common';
 import {
   InjectTool,
+  Input,
+  RunContext,
   Tool,
   ToolInterface,
   ToolResult,
-  WithArguments,
-  WorkflowExecution,
   WorkflowInterface,
+  WorkflowMetadataInterface,
   getBlockTool,
 } from '@loopstack/common';
 import { CreateDocument } from '@loopstack/core-ui-module';
@@ -18,11 +19,15 @@ import { AiGenerateObject, AiGenerateObjectArgsType, AiGenerateObjectSchema } fr
     description: 'Generates a structured object using a LLM and creates it as document',
   },
 })
-@WithArguments(AiGenerateObjectSchema)
 export class AiGenerateDocument implements ToolInterface<AiGenerateObjectArgsType> {
   @InjectTool() private aiGenerateObject!: AiGenerateObject;
   @InjectTool() private createDocument!: CreateDocument;
 
+  @Input({
+    schema: AiGenerateObjectSchema,
+  })
+  args: AiGenerateObjectArgsType;
+
   private getRequiredTool(name: string): ToolInterface {
     const tool = getBlockTool<ToolInterface>(this, name);
     if (tool === undefined) {
@@ -33,10 +38,11 @@ export class AiGenerateDocument implements ToolInterface<AiGenerateObjectArgsTyp
 
   async execute(
     args: AiGenerateObjectArgsType,
-    ctx: WorkflowExecution<any>,
+    ctx: RunContext,
     parent: WorkflowInterface,
+    metadata: WorkflowMetadataInterface,
   ): Promise<ToolResult> {
-    const result = await this.getRequiredTool('aiGenerateObject').execute(args, ctx, parent);
+    const result = await this.getRequiredTool('aiGenerateObject').execute(args, ctx, parent, metadata);
     return this.getRequiredTool('createDocument').execute(
       {
         id: args.response.id,
@@ -47,6 +53,7 @@ export class AiGenerateDocument implements ToolInterface<AiGenerateObjectArgsTyp
       },
       ctx,
       parent,
+      metadata,
     );
   }
 }

package/src/tools/ai-generate-object.tool.ts

@@ -1,5 +1,5 @@
 import { ModelMessage } from '@ai-sdk/provider-utils';
-import { Injectable } from '@nestjs/common';
+import { Inject, Injectable } from '@nestjs/common';
 import {
   GenerateTextResult,
   LanguageModel,
@@ -12,12 +12,13 @@ import {
 import { z } from 'zod';
 import {
   DocumentInterface,
+  Input,
+  RunContext,
   Tool,
   ToolInterface,
   ToolResult,
-  WithArguments,
-  WorkflowExecution,
   WorkflowInterface,
+  WorkflowMetadataInterface,
   getBlockArgsSchema,
   getBlockDocument,
 } from '@loopstack/common';
@@ -40,17 +41,23 @@ export type AiGenerateObjectArgsType = z.infer<typeof AiGenerateObjectSchema>;
     description: 'Generates a structured object using a LLM',
   },
 })
-@WithArguments(AiGenerateObjectSchema)
 export class AiGenerateObject implements ToolInterface<AiGenerateObjectArgsType> {
-  constructor(
-    private readonly aiMessagesHelperService: AiMessagesHelperService,
-    private readonly aiProviderModelHelperService: AiProviderModelHelperService,
-  ) {}
+  @Inject()
+  private readonly aiMessagesHelperService: AiMessagesHelperService;
+
+  @Inject()
+  private readonly aiProviderModelHelperService: AiProviderModelHelperService;
+
+  @Input({
+    schema: AiGenerateObjectSchema,
+  })
+  args: AiGenerateObjectArgsType;
 
   async execute(
     args: AiGenerateObjectArgsType,
-    ctx: WorkflowExecution<any>,
+    ctx: RunContext,
     parent: WorkflowInterface,
+    metadata: WorkflowMetadataInterface,
   ): Promise<ToolResult> {
     const model = this.aiProviderModelHelperService.getProviderModel(args.llm);
 
@@ -67,7 +74,7 @@ export class AiGenerateObject implements ToolInterface<AiGenerateObjectArgsType>
         content: args.prompt,
       } as ModelMessage);
     } else {
-      const messages = this.aiMessagesHelperService.getMessages(ctx.state.getMetadata('documents'), {
+      const messages = this.aiMessagesHelperService.getMessages(metadata.documents, {
         messages: args.messages as unknown as UIMessage[],
         messagesSearchTag: args.messagesSearchTag,
       });
@@ -88,8 +95,6 @@ export class AiGenerateObject implements ToolInterface<AiGenerateObjectArgsType>
 
     const response = await this.handleGenerateObject(model, options);
 
-    console.log(response);
-
     return {
       data: response.output,
     };

package/src/tools/ai-generate-text.tool.ts

@@ -1,14 +1,15 @@
 import { ModelMessage } from '@ai-sdk/provider-utils';
-import { Injectable } from '@nestjs/common';
+import { Inject, Injectable } from '@nestjs/common';
 import { LanguageModel, ToolSet, UIMessage, convertToModelMessages, createUIMessageStream, streamText } from 'ai';
 import { z } from 'zod';
 import {
+  Input,
+  RunContext,
   Tool,
   ToolInterface,
   ToolResult,
-  WithArguments,
-  WorkflowExecution,
   WorkflowInterface,
+  WorkflowMetadataInterface,
 } from '@loopstack/common';
 import { AiGenerateToolBaseSchema } from '../schemas/ai-generate-tool-base.schema';
 import { AiMessagesHelperService } from '../services';
@@ -27,18 +28,24 @@ type AiGenerateTextArgsType = z.infer<typeof AiGenerateTextSchema>;
     description: 'Generates text using a LLM',
   },
 })
-@WithArguments(AiGenerateTextSchema)
 export class AiGenerateText implements ToolInterface<AiGenerateTextArgsType> {
-  constructor(
-    private readonly aiMessagesHelperService: AiMessagesHelperService,
-    private readonly aiToolsHelperService: AiToolsHelperService,
-    private readonly aiProviderModelHelperService: AiProviderModelHelperService,
-  ) {}
+  @Inject()
+  private readonly aiMessagesHelperService: AiMessagesHelperService;
+  @Inject()
+  private readonly aiToolsHelperService: AiToolsHelperService;
+  @Inject()
+  private readonly aiProviderModelHelperService: AiProviderModelHelperService;
+
+  @Input({
+    schema: AiGenerateTextSchema,
+  })
+  args: AiGenerateTextArgsType;
 
   async execute(
     args: AiGenerateTextArgsType,
-    ctx: WorkflowExecution<any>,
+    ctx: RunContext,
     parent: WorkflowInterface,
+    runtime: WorkflowMetadataInterface,
   ): Promise<ToolResult> {
     const model = this.aiProviderModelHelperService.getProviderModel(args.llm);
 
@@ -58,7 +65,7 @@ export class AiGenerateText implements ToolInterface<AiGenerateTextArgsType> {
         content: args.prompt,
       } as ModelMessage);
     } else {
-      const messages = this.aiMessagesHelperService.getMessages(ctx.state.getMetadata('documents'), {
+      const messages = this.aiMessagesHelperService.getMessages(runtime.documents, {
         messages: args.messages as unknown as UIMessage[],
         messagesSearchTag: args.messagesSearchTag,
       });

package/src/tools/delegate-tool-call.tool.ts

@@ -2,12 +2,13 @@ import { Injectable } from '@nestjs/common';
 import { ToolUIPart, UIMessage } from 'ai';
 import { z } from 'zod';
 import {
+  Input,
+  RunContext,
   Tool,
   ToolInterface,
   ToolResult,
-  WithArguments,
-  WorkflowExecution,
   WorkflowInterface,
+  WorkflowMetadataInterface,
   getBlockTool,
 } from '@loopstack/common';
 
@@ -32,12 +33,17 @@ type DelegateToolCallsToolArgs = z.infer<typeof DelegateToolCallsToolSchema>;
     description: 'Delegate a tool call.',
   },
 })
-@WithArguments(DelegateToolCallsToolSchema)
 export class DelegateToolCall implements ToolInterface<DelegateToolCallsToolArgs> {
+  @Input({
+    schema: DelegateToolCallsToolSchema,
+  })
+  args: DelegateToolCallsToolArgs;
+
   async execute(
     args: DelegateToolCallsToolArgs,
-    ctx: WorkflowExecution<any>,
+    ctx: RunContext,
     parent: WorkflowInterface,
+    runtime: WorkflowMetadataInterface,
   ): Promise<ToolResult> {
     const parts = args.message.parts;
     const resultParts: ToolUIPart[] = [];
@@ -57,7 +63,7 @@ export class DelegateToolCall implements ToolInterface<DelegateToolCallsToolArgs
       if (!tool) {
         throw new Error(`Tool ${toolName} not found.`);
       }
-      const result: ToolResult = await tool.execute(part.input as Record<string, unknown>, ctx, parent);
+      const result: ToolResult = await tool.execute(part.input as Record<string, unknown>, ctx, parent, runtime);
 
       resultParts.push({
         type: part.type as ToolUIPart['type'],