@loopstack/claude-module 0.21.1 → 0.22.1

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

@@ -5,22 +5,20 @@ 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);
-};
 var DelegateToolCalls_1;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DelegateToolCalls = void 0;
 const common_1 = require("@nestjs/common");
 const zod_1 = require("zod");
 const common_2 = require("@loopstack/common");
-const core_1 = require("@loopstack/core");
 const DelegateToolCallsSchema = zod_1.z.object({
     message: zod_1.z.object({
         id: zod_1.z.string().optional(),
         content: zod_1.z.array(zod_1.z.any()),
     }),
-    document: zod_1.z.string().optional(),
+    document: zod_1.z
+        .custom((val) => typeof val === 'function' && (0, common_2.getBlockTypeFromMetadata)(val) === 'document')
+        .optional(),
     skipResponseMessage: zod_1.z.boolean().optional(),
     callback: zod_1.z
         .object({
@@ -28,41 +26,36 @@ const DelegateToolCallsSchema = zod_1.z.object({
     })
         .optional(),
 });
-let DelegateToolCalls = DelegateToolCalls_1 = class DelegateToolCalls {
-    toolCallProcessor;
-    eventSubscriberService;
+let DelegateToolCalls = DelegateToolCalls_1 = class DelegateToolCalls extends common_2.BaseTool {
     logger = new common_1.Logger(DelegateToolCalls_1.name);
-    createDocument;
-    constructor(toolCallProcessor, eventSubscriberService) {
-        this.toolCallProcessor = toolCallProcessor;
-        this.eventSubscriberService = eventSubscriberService;
-    }
-    args;
-    async execute(args, ctx, parent, metadata) {
+    async call(args) {
         const contentBlocks = args.message.content;
         const toolUseBlocks = contentBlocks.filter((b) => b.type === 'tool_use');
         if (toolUseBlocks.length === 0) {
             return {
-                data: { allCompleted: true, toolResults: [], message: args.message, pendingCount: 0 },
+                data: {
+                    allCompleted: true,
+                    toolResults: [],
+                    message: args.message,
+                    pendingCount: 0,
+                },
             };
         }
-        const results = await Promise.all(toolUseBlocks.map((block) => this.executeTool(block, ctx, parent, metadata)));
+        const results = await Promise.all(toolUseBlocks.map((block) => {
+            const toolCallback = args.callback
+                ? {
+                    transition: args.callback.transition,
+                    metadata: { toolUseId: block.id, toolName: block.name },
+                }
+                : undefined;
+            return this.executeTool(block, toolCallback ? { callback: toolCallback } : undefined);
+        }));
         let pendingCount = 0;
         const toolResults = [];
-        const toolEffects = [];
         for (let i = 0; i < toolUseBlocks.length; i++) {
             const block = toolUseBlocks[i];
             const result = results[i];
-            if (result.effects) {
-                toolEffects.push(...result.effects);
-            }
-            const resultData = result.data;
-            if (resultData?.mode === 'async') {
-                if (!args.callback?.transition) {
-                    throw new Error(`Tool "${block.name}" returned an async result but no callback.transition was provided. ` +
-                        `Add a callback configuration to the delegateToolCalls args.`);
-                }
-                await this.eventSubscriberService.registerSubscriber(ctx.pipelineId, ctx.workflowId, args.callback.transition, resultData.correlationId, resultData.eventName, ctx.userId, ctx.workspaceId, { toolUseId: block.id, toolName: block.name });
+            if (result.pending) {
                 pendingCount++;
             }
             else {
@@ -74,14 +67,9 @@ let DelegateToolCalls = DelegateToolCalls_1 = class DelegateToolCalls {
                 });
             }
         }
-        const effects = [];
         if (args.document && !args.skipResponseMessage) {
-            const docResult = await this.createResponseDocument(args, toolResults, ctx, parent, metadata);
-            if (docResult.effects) {
-                effects.push(...docResult.effects);
-            }
+            await this.createResponseDocument(args.document, args, toolResults);
         }
-        effects.push(...toolEffects);
         return {
             data: {
                 allCompleted: pendingCount === 0,
@@ -89,14 +77,15 @@ let DelegateToolCalls = DelegateToolCalls_1 = class DelegateToolCalls {
                 message: args.message,
                 pendingCount,
             },
-            effects,
         };
     }
-    async executeTool(block, ctx, parent, metadata) {
-        const tool = this.toolCallProcessor.getTool(parent, block.name);
-        const parsedArgs = this.toolCallProcessor.parseArgs(tool, block.input, metadata.transition);
+    async executeTool(block, options) {
         try {
-            return await this.toolCallProcessor.executeToolCall(tool, parsedArgs, ctx, parent, metadata);
+            const tool = (0, common_2.getBlockTool)(this.ctx.parent, block.name);
+            if (!tool) {
+                throw new Error(`Tool with name ${block.name} not found.`);
+            }
+            return await tool.call(block.input, options);
         }
         catch (error) {
             const errorMessage = error instanceof Error ? error.message : String(error);
@@ -107,40 +96,22 @@ let DelegateToolCalls = DelegateToolCalls_1 = class DelegateToolCalls {
             };
         }
     }
-    async createResponseDocument(args, toolResults, ctx, parent, metadata) {
-        return this.createDocument.execute({
-            id: args.message.id,
-            document: args.document,
-            validate: 'skip',
-            update: {
-                content: {
-                    role: 'assistant',
-                    content: args.message.content,
-                    toolResults,
-                },
-            },
-        }, ctx, parent, metadata);
+    async createResponseDocument(document, args, toolResults) {
+        await this.repository.save(document, {
+            role: 'assistant',
+            content: args.message.content,
+            toolResults,
+        }, { id: args.message.id, validate: 'skip' });
     }
 };
 exports.DelegateToolCalls = DelegateToolCalls;
-__decorate([
-    (0, common_2.InjectTool)(),
-    __metadata("design:type", core_1.CreateDocument)
-], DelegateToolCalls.prototype, "createDocument", void 0);
-__decorate([
-    (0, common_2.Input)({
-        schema: DelegateToolCallsSchema,
-    }),
-    __metadata("design:type", Object)
-], DelegateToolCalls.prototype, "args", void 0);
 exports.DelegateToolCalls = DelegateToolCalls = DelegateToolCalls_1 = __decorate([
     (0, common_1.Injectable)(),
     (0, common_2.Tool)({
-        config: {
+        uiConfig: {
             description: 'Delegate tool calls from an LLM response. Handles both sync tools and async task tools.',
         },
-    }),
-    __metadata("design:paramtypes", [core_1.StateMachineToolCallProcessorService,
-        core_1.EventSubscriberService])
+        schema: DelegateToolCallsSchema,
+    })
 ], DelegateToolCalls);
 //# sourceMappingURL=delegate-tool-calls.tool.js.map

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

@@ -1 +1 @@
-{"version":3,"file":"delegate-tool-calls.tool.js","sourceRoot":"","sources":["../../src/tools/delegate-tool-calls.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;;AACA,2CAAoD;AACpD,6BAAwB;AACxB,8CAU2B;AAC3B,0CAA+G;AAE/G,MAAM,uBAAuB,GAAG,OAAC,CAAC,MAAM,CAAC;IACvC,OAAO,EAAE,OAAC,CAAC,MAAM,CAAC;QAChB,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACzB,OAAO,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC;KAC1B,CAAC;IACF,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IAC/B,mBAAmB,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IAC3C,QAAQ,EAAE,OAAC;SACR,MAAM,CAAC;QACN,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE;KACvB,CAAC;SACD,QAAQ,EAAE;CACd,CAAC,CAAC;AAiBI,IAAM,iBAAiB,yBAAvB,MAAM,iBAAiB;IAMT;IACA;IANF,MAAM,GAAG,IAAI,eAAM,CAAC,mBAAiB,CAAC,IAAI,CAAC,CAAC;IAEvC,cAAc,CAAiB;IAErD,YACmB,iBAAuD,EACvD,sBAA8C;QAD9C,sBAAiB,GAAjB,iBAAiB,CAAsC;QACvD,2BAAsB,GAAtB,sBAAsB,CAAwB;IAC9D,CAAC;IAKJ,IAAI,CAAwB;IAE5B,KAAK,CAAC,OAAO,CACX,IAA2B,EAC3B,GAAe,EACf,MAAyB,EACzB,QAAmC;QAEnC,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,OAAmC,CAAC;QACvE,MAAM,aAAa,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAA+B,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;QAEtG,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO;gBACL,IAAI,EAAE,EAAE,YAAY,EAAE,IAAI,EAAE,WAAW,EAAE,EAAE,EAAE,OAAO,EAAE,IAAI,CAAC,OAAO,EAAE,YAAY,EAAE,CAAC,EAAE;aACtF,CAAC;QACJ,CAAC;QAGD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,GAAG,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC,CAAC,CAAC;QAGhH,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,MAAM,WAAW,GAAsB,EAAE,CAAC;QAC1C,MAAM,WAAW,GAAsB,EAAE,CAAC;QAE1C,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9C,MAAM,KAAK,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAG1B,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,WAAW,CAAC,IAAI,CAAC,GAAG,MAAM,CAAC,OAAO,CAAC,CAAC;YACtC,CAAC;YAED,MAAM,UAAU,GAAG,MAAM,CAAC,IAA2C,CAAC;YACtE,IAAI,UAAU,EAAE,IAAI,KAAK,OAAO,EAAE,CAAC;gBAEjC,IAAI,CAAC,IAAI,CAAC,QAAQ,EAAE,UAAU,EAAE,CAAC;oBAC/B,MAAM,IAAI,KAAK,CACb,SAAS,KAAK,CAAC,IAAI,sEAAsE;wBACvF,6DAA6D,CAChE,CAAC;gBACJ,CAAC;gBAGD,MAAM,IAAI,CAAC,sBAAsB,CAAC,kBAAkB,CAClD,GAAG,CAAC,UAAU,EACd,GAAG,CAAC,UAAW,EACf,IAAI,CAAC,QAAQ,CAAC,UAAU,EACxB,UAAU,CAAC,aAAuB,EAClC,UAAU,CAAC,SAAmB,EAC9B,GAAG,CAAC,MAAM,EACV,GAAG,CAAC,WAAW,EACf,EAAE,SAAS,EAAE,KAAK,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,IAAI,EAAE,CAC9C,CAAC;gBAEF,YAAY,EAAE,CAAC;YACjB,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,IAAI,CAAC;oBACf,IAAI,EAAE,aAAa;oBACnB,WAAW,EAAE,KAAK,CAAC,EAAE;oBACrB,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;oBAChE,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK;iBACzB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAID,MAAM,OAAO,GAAsB,EAAE,CAAC;QACtC,IAAI,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC/C,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,sBAAsB,CAAC,IAAI,EAAE,WAAW,EAAE,GAAG,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;YAC9F,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;gBACtB,OAAO,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC;YACrC,CAAC;QACH,CAAC;QACD,OAAO,CAAC,IAAI,CAAC,GAAG,WAAW,CAAC,CAAC;QAE7B,OAAO;YACL,IAAI,EAAE;gBACJ,YAAY,EAAE,YAAY,KAAK,CAAC;gBAChC,WAAW;gBACX,OAAO,EAAE,IAAI,CAAC,OAAO;gBACrB,YAAY;aACb;YACD,OAAO;SACR,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,WAAW,CACvB,KAA6B,EAC7B,GAAe,EACf,MAAyB,EACzB,QAAmC;QAEnC,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;QAChE,MAAM,UAAU,GAAG,IAAI,CAAC,iBAAiB,CAAC,SAAS,CACjD,IAAI,EACJ,KAAK,CAAC,KAA4C,EAClD,QAAQ,CAAC,UAAW,CACrB,CAAC;QAEF,IAAI,CAAC;YACH,OAAO,MAAM,IAAI,CAAC,iBAAiB,CAAC,eAAe,CAAC,IAAI,EAAE,UAAU,EAAE,GAAG,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC/F,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,KAAK,CAAC,IAAI,aAAa,YAAY,EAAE,CAAC,CAAC;YAClE,OAAO;gBACL,IAAI,EAAE,YAAY;gBAClB,KAAK,EAAE,YAAY;aACpB,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAClC,IAA2B,EAC3B,WAA8B,EAC9B,GAAe,EACf,MAAyB,EACzB,QAAmC;QAEnC,OAAO,IAAI,CAAC,cAAc,CAAC,OAAO,CAChC;YACE,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,EAAE;YACnB,QAAQ,EAAE,IAAI,CAAC,QAAS;YACxB,QAAQ,EAAE,MAAe;YACzB,MAAM,EAAE;gBACN,OAAO,EAAE;oBACP,IAAI,EAAE,WAAW;oBACjB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;oBAC7B,WAAW;iBACZ;aACF;SACF,EACD,GAAG,EACH,MAAM,EACN,QAAQ,CACT,CAAC;IACJ,CAAC;CACF,CAAA;AAxJY,8CAAiB;AAGN;IAArB,IAAA,mBAAU,GAAE;8BAAyB,qBAAc;yDAAC;AAUrD;IAHC,IAAA,cAAK,EAAC;QACL,MAAM,EAAE,uBAAuB;KAChC,CAAC;;+CAC0B;4BAbjB,iBAAiB;IAN7B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,MAAM,EAAE;YACN,WAAW,EAAE,yFAAyF;SACvG;KACF,CAAC;qCAOsC,2CAAoC;QAC/B,6BAAsB;GAPtD,iBAAiB,CAwJ7B"}
+{"version":3,"file":"delegate-tool-calls.tool.js","sourceRoot":"","sources":["../../src/tools/delegate-tool-calls.tool.ts"],"names":[],"mappings":";;;;;;;;;;AACA,2CAAoD;AACpD,6BAAwB;AACxB,8CAQ2B;AAG3B,MAAM,uBAAuB,GAAG,OAAC,CAAC,MAAM,CAAC;IACvC,OAAO,EAAE,OAAC,CAAC,MAAM,CAAC;QAChB,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACzB,OAAO,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC;KAC1B,CAAC;IACF,QAAQ,EAAE,OAAC;SACR,MAAM,CAAgB,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,GAAG,KAAK,UAAU,IAAI,IAAA,iCAAwB,EAAC,GAAa,CAAC,KAAK,UAAU,CAAC;SACnH,QAAQ,EAAE;IACb,mBAAmB,EAAE,OAAC,CAAC,OAAO,EAAE,CAAC,QAAQ,EAAE;IAC3C,QAAQ,EAAE,OAAC;SACR,MAAM,CAAC;QACN,UAAU,EAAE,OAAC,CAAC,MAAM,EAAE;KACvB,CAAC;SACD,QAAQ,EAAE;CACd,CAAC,CAAC;AAWI,IAAM,iBAAiB,yBAAvB,MAAM,iBAAkB,SAAQ,iBAAQ;IAC5B,MAAM,GAAG,IAAI,eAAM,CAAC,mBAAiB,CAAC,IAAI,CAAC,CAAC;IAE7D,KAAK,CAAC,IAAI,CAAC,IAA2B;QACpC,MAAM,aAAa,GAAG,IAAI,CAAC,OAAO,CAAC,OAAmC,CAAC;QACvE,MAAM,aAAa,GAAG,aAAa,CAAC,MAAM,CAAC,CAAC,CAAC,EAA+B,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC;QAEtG,IAAI,aAAa,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;YAC/B,OAAO;gBACL,IAAI,EAAE;oBACJ,YAAY,EAAE,IAAI;oBAClB,WAAW,EAAE,EAAE;oBACf,OAAO,EAAE,IAAI,CAAC,OAA6C;oBAC3D,YAAY,EAAE,CAAC;iBAChB;aACF,CAAC;QACJ,CAAC;QAGD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,GAAG,CAC/B,aAAa,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE;YAC1B,MAAM,YAAY,GAAG,IAAI,CAAC,QAAQ;gBAChC,CAAC,CAAC;oBACE,UAAU,EAAE,IAAI,CAAC,QAAQ,CAAC,UAAU;oBACpC,QAAQ,EAAE,EAAE,SAAS,EAAE,KAAK,CAAC,EAAE,EAAE,QAAQ,EAAE,KAAK,CAAC,IAAI,EAAE;iBACxD;gBACH,CAAC,CAAC,SAAS,CAAC;YACd,OAAO,IAAI,CAAC,WAAW,CAAC,KAAK,EAAE,YAAY,CAAC,CAAC,CAAC,EAAE,QAAQ,EAAE,YAAY,EAAE,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC;QACxF,CAAC,CAAC,CACH,CAAC;QAGF,IAAI,YAAY,GAAG,CAAC,CAAC;QACrB,MAAM,WAAW,GAA8B,EAAE,CAAC;QAElD,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,aAAa,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;YAC9C,MAAM,KAAK,GAAG,aAAa,CAAC,CAAC,CAAC,CAAC;YAC/B,MAAM,MAAM,GAAG,OAAO,CAAC,CAAC,CAAC,CAAC;YAE1B,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;gBACnB,YAAY,EAAE,CAAC;YACjB,CAAC;iBAAM,CAAC;gBACN,WAAW,CAAC,IAAI,CAAC;oBACf,IAAI,EAAE,aAAa;oBACnB,WAAW,EAAE,KAAK,CAAC,EAAE;oBACrB,OAAO,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;oBAChE,QAAQ,EAAE,CAAC,CAAC,MAAM,CAAC,KAAK;iBACzB,CAAC,CAAC;YACL,CAAC;QACH,CAAC;QAGD,IAAI,IAAI,CAAC,QAAQ,IAAI,CAAC,IAAI,CAAC,mBAAmB,EAAE,CAAC;YAC/C,MAAM,IAAI,CAAC,sBAAsB,CAAC,IAAI,CAAC,QAAQ,EAAE,IAAI,EAAE,WAAW,CAAC,CAAC;QACtE,CAAC;QAED,OAAO;YACL,IAAI,EAAE;gBACJ,YAAY,EAAE,YAAY,KAAK,CAAC;gBAChC,WAAW;gBACX,OAAO,EAAE,IAAI,CAAC,OAA6C;gBAC3D,YAAY;aACb;SACF,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,WAAW,CAAC,KAA6B,EAAE,OAAyB;QAChF,IAAI,CAAC;YACH,MAAM,IAAI,GAAG,IAAA,qBAAY,EAAW,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;YACjE,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,IAAI,KAAK,CAAC,kBAAkB,KAAK,CAAC,IAAI,aAAa,CAAC,CAAC;YAC7D,CAAC;YAED,OAAO,MAAM,IAAI,CAAC,IAAI,CAAC,KAAK,CAAC,KAAgC,EAAE,OAAO,CAAC,CAAC;QAC1E,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,KAAK,CAAC,IAAI,aAAa,YAAY,EAAE,CAAC,CAAC;YAClE,OAAO;gBACL,IAAI,EAAE,YAAY;gBAClB,KAAK,EAAE,YAAY;aACpB,CAAC;QACJ,CAAC;IACH,CAAC;IAEO,KAAK,CAAC,sBAAsB,CAClC,QAAuB,EACvB,IAA2B,EAC3B,WAAsC;QAEtC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACxB,QAAQ,EACR;YACE,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,IAAI,CAAC,OAAO,CAAC,OAAO;YAC7B,WAAW;SACZ,EACD,EAAE,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,EAAE,EAAE,QAAQ,EAAE,MAAM,EAAE,CAC1C,CAAC;IACJ,CAAC;CACF,CAAA;AAnGY,8CAAiB;4BAAjB,iBAAiB;IAP7B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,QAAQ,EAAE;YACR,WAAW,EAAE,yFAAyF;SACvG;QACD,MAAM,EAAE,uBAAuB;KAChC,CAAC;GACW,iBAAiB,CAmG7B"}

package/dist/tools/update-tool-result.tool.d.ts

@@ -1,6 +1,5 @@
 import { z } from 'zod';
-import { RunContext, ToolInterface, ToolResult, WorkflowInterface, WorkflowMetadataInterface } from '@loopstack/common';
-import { StateMachineToolCallProcessorService } from '@loopstack/core';
+import { BaseTool, DocumentClass, ToolResult } from '@loopstack/common';
 declare const UpdateToolResultSchema: z.ZodObject<{
     delegateResult: z.ZodObject<{
         allCompleted: z.ZodBoolean;
@@ -9,16 +8,12 @@ declare const UpdateToolResultSchema: z.ZodObject<{
         pendingCount: z.ZodNumber;
     }, z.core.$strip>;
     completedTool: z.ZodAny;
-    document: z.ZodOptional<z.ZodString>;
+    document: z.ZodOptional<z.ZodCustom<DocumentClass, DocumentClass>>;
 }, z.core.$strip>;
 type UpdateToolResultArgs = z.infer<typeof UpdateToolResultSchema>;
-export declare class UpdateToolResult implements ToolInterface<UpdateToolResultArgs> {
-    private readonly toolCallProcessor;
+export declare class UpdateToolResult extends BaseTool {
     private readonly logger;
-    private createDocument;
-    constructor(toolCallProcessor: StateMachineToolCallProcessorService);
-    args: UpdateToolResultArgs;
-    execute(args: UpdateToolResultArgs, ctx: RunContext, parent: WorkflowInterface, metadata: WorkflowMetadataInterface): Promise<ToolResult>;
+    call(args: UpdateToolResultArgs): Promise<ToolResult>;
 }
 export {};
 //# sourceMappingURL=update-tool-result.tool.d.ts.map

package/dist/tools/update-tool-result.tool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"update-tool-result.tool.d.ts","sourceRoot":"","sources":["../../src/tools/update-tool-result.tool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAGL,UAAU,EAEV,aAAa,EACb,UAAU,EAEV,iBAAiB,EACjB,yBAAyB,EAC1B,MAAM,mBAAmB,CAAC;AAC3B,OAAO,EAAkB,oCAAoC,EAAE,MAAM,iBAAiB,CAAC;AAEvF,QAAA,MAAM,sBAAsB;;;;;;;;;iBAS1B,CAAC;AAEH,KAAK,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAEnE,qBAMa,gBAAiB,YAAW,aAAa,CAAC,oBAAoB,CAAC;IAK9D,OAAO,CAAC,QAAQ,CAAC,iBAAiB;IAJ9C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAqC;IAE9C,OAAO,CAAC,cAAc,CAAiB;gBAExB,iBAAiB,EAAE,oCAAoC;IAKpF,IAAI,EAAE,oBAAoB,CAAC;IAErB,OAAO,CACX,IAAI,EAAE,oBAAoB,EAC1B,GAAG,EAAE,UAAU,EACf,MAAM,EAAE,iBAAiB,EACzB,QAAQ,EAAE,yBAAyB,GAClC,OAAO,CAAC,UAAU,CAAC;CAuFvB"}
+{"version":3,"file":"update-tool-result.tool.d.ts","sourceRoot":"","sources":["../../src/tools/update-tool-result.tool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,QAAQ,EAAE,aAAa,EAAQ,UAAU,EAA0C,MAAM,mBAAmB,CAAC;AAEtH,QAAA,MAAM,sBAAsB;;;;;;;;;iBAW1B,CAAC;AAEH,KAAK,oBAAoB,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,sBAAsB,CAAC,CAAC;AAEnE,qBAOa,gBAAiB,SAAQ,QAAQ;IAC5C,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAqC;IAEtD,IAAI,CAAC,IAAI,EAAE,oBAAoB,GAAG,OAAO,CAAC,UAAU,CAAC;CAqE5D"}

package/dist/tools/update-tool-result.tool.js

@@ -5,16 +5,12 @@ 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);
-};
 var UpdateToolResult_1;
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.UpdateToolResult = void 0;
 const common_1 = require("@nestjs/common");
 const zod_1 = require("zod");
 const common_2 = require("@loopstack/common");
-const core_1 = require("@loopstack/core");
 const UpdateToolResultSchema = zod_1.z.object({
     delegateResult: zod_1.z.object({
         allCompleted: zod_1.z.boolean(),
@@ -23,17 +19,13 @@ const UpdateToolResultSchema = zod_1.z.object({
         pendingCount: zod_1.z.number(),
     }),
     completedTool: zod_1.z.any(),
-    document: zod_1.z.string().optional(),
+    document: zod_1.z
+        .custom((val) => typeof val === 'function' && (0, common_2.getBlockTypeFromMetadata)(val) === 'document')
+        .optional(),
 });
-let UpdateToolResult = UpdateToolResult_1 = class UpdateToolResult {
-    toolCallProcessor;
+let UpdateToolResult = UpdateToolResult_1 = class UpdateToolResult extends common_2.BaseTool {
     logger = new common_1.Logger(UpdateToolResult_1.name);
-    createDocument;
-    constructor(toolCallProcessor) {
-        this.toolCallProcessor = toolCallProcessor;
-    }
-    args;
-    async execute(args, ctx, parent, metadata) {
+    async call(args) {
         const { delegateResult } = args;
         const completedToolRecord = args.completedTool;
         const subscriberMetadata = completedToolRecord._subscriberMetadata;
@@ -42,15 +34,13 @@ let UpdateToolResult = UpdateToolResult_1 = class UpdateToolResult {
                 'Ensure the event subscriber was registered with metadata by delegateToolCalls.');
         }
         const { toolUseId, toolName } = subscriberMetadata;
-        const tool = this.toolCallProcessor.getTool(parent, toolName);
+        const tool = (0, common_2.getBlockTool)(this.ctx.parent, toolName);
+        if (!tool) {
+            throw new Error(`Tool with name ${toolName} not found.`);
+        }
         let toolResult;
         try {
-            if (tool.complete) {
-                toolResult = await tool.complete(completedToolRecord, ctx, parent, metadata);
-            }
-            else {
-                toolResult = { data: completedToolRecord.result ?? completedToolRecord };
-            }
+            toolResult = await tool.complete(completedToolRecord);
         }
         catch (error) {
             const errorMessage = error instanceof Error ? error.message : String(error);
@@ -69,26 +59,12 @@ let UpdateToolResult = UpdateToolResult_1 = class UpdateToolResult {
         const updatedResults = [...delegateResult.toolResults, completedEntry];
         const pendingCount = delegateResult.pendingCount - 1;
         const allCompleted = pendingCount === 0;
-        const allEffects = [];
-        if (toolResult.effects) {
-            allEffects.push(...toolResult.effects);
-        }
         if (args.document) {
-            const docResult = await this.createDocument.execute({
-                id: delegateResult.message.id,
-                document: args.document,
-                validate: 'skip',
-                update: {
-                    content: {
-                        role: 'assistant',
-                        content: delegateResult.message.content,
-                        toolResults: updatedResults,
-                    },
-                },
-            }, ctx, parent, metadata);
-            if (docResult.effects) {
-                allEffects.push(...docResult.effects);
-            }
+            await this.repository.save(args.document, {
+                role: 'assistant',
+                content: delegateResult.message.content,
+                toolResults: updatedResults,
+            }, { id: delegateResult.message.id, validate: 'skip' });
         }
         return {
             data: {
@@ -97,28 +73,17 @@ let UpdateToolResult = UpdateToolResult_1 = class UpdateToolResult {
                 allCompleted,
                 pendingCount,
             },
-            effects: allEffects.length > 0 ? allEffects : undefined,
         };
     }
 };
 exports.UpdateToolResult = UpdateToolResult;
-__decorate([
-    (0, common_2.InjectTool)(),
-    __metadata("design:type", core_1.CreateDocument)
-], UpdateToolResult.prototype, "createDocument", void 0);
-__decorate([
-    (0, common_2.Input)({
-        schema: UpdateToolResultSchema,
-    }),
-    __metadata("design:type", Object)
-], UpdateToolResult.prototype, "args", void 0);
 exports.UpdateToolResult = UpdateToolResult = UpdateToolResult_1 = __decorate([
     (0, common_1.Injectable)(),
     (0, common_2.Tool)({
-        config: {
+        uiConfig: {
             description: 'Handle async tool completion callback. Updates the response document and tracks completion state.',
         },
-    }),
-    __metadata("design:paramtypes", [core_1.StateMachineToolCallProcessorService])
+        schema: UpdateToolResultSchema,
+    })
 ], UpdateToolResult);
 //# sourceMappingURL=update-tool-result.tool.js.map

package/dist/tools/update-tool-result.tool.js.map

@@ -1 +1 @@
-{"version":3,"file":"update-tool-result.tool.js","sourceRoot":"","sources":["../../src/tools/update-tool-result.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;;AAAA,2CAAoD;AACpD,6BAAwB;AACxB,8CAU2B;AAC3B,0CAAuF;AAEvF,MAAM,sBAAsB,GAAG,OAAC,CAAC,MAAM,CAAC;IACtC,cAAc,EAAE,OAAC,CAAC,MAAM,CAAC;QACvB,YAAY,EAAE,OAAC,CAAC,OAAO,EAAE;QACzB,WAAW,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC;QAC7B,OAAO,EAAE,OAAC,CAAC,GAAG,EAAE;QAChB,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE;KACzB,CAAC;IACF,aAAa,EAAE,OAAC,CAAC,GAAG,EAAE;IACtB,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;CAChC,CAAC,CAAC;AAUI,IAAM,gBAAgB,wBAAtB,MAAM,gBAAgB;IAKE;IAJZ,MAAM,GAAG,IAAI,eAAM,CAAC,kBAAgB,CAAC,IAAI,CAAC,CAAC;IAEtC,cAAc,CAAiB;IAErD,YAA6B,iBAAuD;QAAvD,sBAAiB,GAAjB,iBAAiB,CAAsC;IAAG,CAAC;IAKxF,IAAI,CAAuB;IAE3B,KAAK,CAAC,OAAO,CACX,IAA0B,EAC1B,GAAe,EACf,MAAyB,EACzB,QAAmC;QAEnC,MAAM,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC;QAChC,MAAM,mBAAmB,GAAG,IAAI,CAAC,aAAwC,CAAC;QAG1E,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,mBAElC,CAAC;QACd,IAAI,CAAC,kBAAkB,EAAE,SAAS,IAAI,CAAC,kBAAkB,EAAE,QAAQ,EAAE,CAAC;YACpE,MAAM,IAAI,KAAK,CACb,+EAA+E;gBAC7E,gFAAgF,CACnF,CAAC;QACJ,CAAC;QAED,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,kBAAkB,CAAC;QAGnD,MAAM,IAAI,GAAG,IAAI,CAAC,iBAAiB,CAAC,OAAO,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC9D,IAAI,UAAsB,CAAC;QAC3B,IAAI,CAAC;YACH,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;gBAClB,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,mBAAmB,EAAE,GAAG,EAAE,MAAM,EAAE,QAAQ,CAAC,CAAC;YAC/E,CAAC;iBAAM,CAAC;gBACN,UAAU,GAAG,EAAE,IAAI,EAAE,mBAAmB,CAAC,MAAM,IAAI,mBAAmB,EAAE,CAAC;YAC3E,CAAC;QACH,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,QAAQ,wBAAwB,YAAY,EAAE,CAAC,CAAC;YAC3E,UAAU,GAAG;gBACX,IAAI,EAAE,YAAY;gBAClB,KAAK,EAAE,YAAY;aACpB,CAAC;QACJ,CAAC;QAGD,MAAM,cAAc,GAAG;YACrB,IAAI,EAAE,aAAsB;YAC5B,WAAW,EAAE,SAAS;YACtB,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;YACxE,QAAQ,EAAE,CAAC,CAAC,UAAU,CAAC,KAAK;SAC7B,CAAC;QACF,MAAM,cAAc,GAAG,CAAC,GAAI,cAAc,CAAC,WAAyC,EAAE,cAAc,CAAC,CAAC;QAGtG,MAAM,YAAY,GAAG,cAAc,CAAC,YAAY,GAAG,CAAC,CAAC;QACrD,MAAM,YAAY,GAAG,YAAY,KAAK,CAAC,CAAC;QAGxC,MAAM,UAAU,GAAsB,EAAE,CAAC;QACzC,IAAI,UAAU,CAAC,OAAO,EAAE,CAAC;YACvB,UAAU,CAAC,IAAI,CAAC,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;QACzC,CAAC;QAED,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,SAAS,GAAG,MAAM,IAAI,CAAC,cAAc,CAAC,OAAO,CACjD;gBACE,EAAE,EAAG,cAAc,CAAC,OAAmC,CAAC,EAAY;gBACpE,QAAQ,EAAE,IAAI,CAAC,QAAQ;gBACvB,QAAQ,EAAE,MAAe;gBACzB,MAAM,EAAE;oBACN,OAAO,EAAE;wBACP,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAG,cAAc,CAAC,OAAmC,CAAC,OAAO;wBACpE,WAAW,EAAE,cAAc;qBAC5B;iBACF;aACF,EACD,GAAG,EACH,MAAM,EACN,QAAQ,CACT,CAAC;YACF,IAAI,SAAS,CAAC,OAAO,EAAE,CAAC;gBACtB,UAAU,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,OAAO,CAAC,CAAC;YACxC,CAAC;QACH,CAAC;QAED,OAAO;YACL,IAAI,EAAE;gBACJ,GAAG,cAAc;gBACjB,WAAW,EAAE,cAAc;gBAC3B,YAAY;gBACZ,YAAY;aACb;YACD,OAAO,EAAE,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,SAAS;SACxD,CAAC;IACJ,CAAC;CACF,CAAA;AAxGY,4CAAgB;AAGL;IAArB,IAAA,mBAAU,GAAE;8BAAyB,qBAAc;wDAAC;AAOrD;IAHC,IAAA,cAAK,EAAC;QACL,MAAM,EAAE,sBAAsB;KAC/B,CAAC;;8CACyB;2BAVhB,gBAAgB;IAN5B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,MAAM,EAAE;YACN,WAAW,EAAE,mGAAmG;SACjH;KACF,CAAC;qCAMgD,2CAAoC;GALzE,gBAAgB,CAwG5B"}
+{"version":3,"file":"update-tool-result.tool.js","sourceRoot":"","sources":["../../src/tools/update-tool-result.tool.ts"],"names":[],"mappings":";;;;;;;;;;AAAA,2CAAoD;AACpD,6BAAwB;AACxB,8CAAsH;AAEtH,MAAM,sBAAsB,GAAG,OAAC,CAAC,MAAM,CAAC;IACtC,cAAc,EAAE,OAAC,CAAC,MAAM,CAAC;QACvB,YAAY,EAAE,OAAC,CAAC,OAAO,EAAE;QACzB,WAAW,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC;QAC7B,OAAO,EAAE,OAAC,CAAC,GAAG,EAAE;QAChB,YAAY,EAAE,OAAC,CAAC,MAAM,EAAE;KACzB,CAAC;IACF,aAAa,EAAE,OAAC,CAAC,GAAG,EAAE;IACtB,QAAQ,EAAE,OAAC;SACR,MAAM,CAAgB,CAAC,GAAG,EAAE,EAAE,CAAC,OAAO,GAAG,KAAK,UAAU,IAAI,IAAA,iCAAwB,EAAC,GAAa,CAAC,KAAK,UAAU,CAAC;SACnH,QAAQ,EAAE;CACd,CAAC,CAAC;AAWI,IAAM,gBAAgB,wBAAtB,MAAM,gBAAiB,SAAQ,iBAAQ;IAC3B,MAAM,GAAG,IAAI,eAAM,CAAC,kBAAgB,CAAC,IAAI,CAAC,CAAC;IAE5D,KAAK,CAAC,IAAI,CAAC,IAA0B;QACnC,MAAM,EAAE,cAAc,EAAE,GAAG,IAAI,CAAC;QAChC,MAAM,mBAAmB,GAAG,IAAI,CAAC,aAAwC,CAAC;QAG1E,MAAM,kBAAkB,GAAG,mBAAmB,CAAC,mBAElC,CAAC;QACd,IAAI,CAAC,kBAAkB,EAAE,SAAS,IAAI,CAAC,kBAAkB,EAAE,QAAQ,EAAE,CAAC;YACpE,MAAM,IAAI,KAAK,CACb,+EAA+E;gBAC7E,gFAAgF,CACnF,CAAC;QACJ,CAAC;QAED,MAAM,EAAE,SAAS,EAAE,QAAQ,EAAE,GAAG,kBAAkB,CAAC;QAGnD,MAAM,IAAI,GAAG,IAAA,qBAAY,EAAW,IAAI,CAAC,GAAG,CAAC,MAAM,EAAE,QAAQ,CAAC,CAAC;QAC/D,IAAI,CAAC,IAAI,EAAE,CAAC;YACV,MAAM,IAAI,KAAK,CAAC,kBAAkB,QAAQ,aAAa,CAAC,CAAC;QAC3D,CAAC;QACD,IAAI,UAAsB,CAAC;QAC3B,IAAI,CAAC;YACH,UAAU,GAAG,MAAM,IAAI,CAAC,QAAQ,CAAC,mBAAmB,CAAC,CAAC;QACxD,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,YAAY,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;YAC5E,IAAI,CAAC,MAAM,CAAC,KAAK,CAAC,SAAS,QAAQ,wBAAwB,YAAY,EAAE,CAAC,CAAC;YAC3E,UAAU,GAAG;gBACX,IAAI,EAAE,YAAY;gBAClB,KAAK,EAAE,YAAY;aACpB,CAAC;QACJ,CAAC;QAGD,MAAM,cAAc,GAAG;YACrB,IAAI,EAAE,aAAsB;YAC5B,WAAW,EAAE,SAAS;YACtB,OAAO,EAAE,UAAU,CAAC,IAAI,CAAC,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,UAAU,CAAC,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE;YACxE,QAAQ,EAAE,CAAC,CAAC,UAAU,CAAC,KAAK;SAC7B,CAAC;QACF,MAAM,cAAc,GAAG,CAAC,GAAI,cAAc,CAAC,WAAyC,EAAE,cAAc,CAAC,CAAC;QAGtG,MAAM,YAAY,GAAG,cAAc,CAAC,YAAY,GAAG,CAAC,CAAC;QACrD,MAAM,YAAY,GAAG,YAAY,KAAK,CAAC,CAAC;QAGxC,IAAI,IAAI,CAAC,QAAQ,EAAE,CAAC;YAClB,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACxB,IAAI,CAAC,QAAQ,EACb;gBACE,IAAI,EAAE,WAAW;gBACjB,OAAO,EAAG,cAAc,CAAC,OAAmC,CAAC,OAAO;gBACpE,WAAW,EAAE,cAAc;aAC5B,EACD,EAAE,EAAE,EAAG,cAAc,CAAC,OAAmC,CAAC,EAAY,EAAE,QAAQ,EAAE,MAAM,EAAE,CAC3F,CAAC;QACJ,CAAC;QAED,OAAO;YACL,IAAI,EAAE;gBACJ,GAAG,cAAc;gBACjB,WAAW,EAAE,cAAc;gBAC3B,YAAY;gBACZ,YAAY;aACb;SACF,CAAC;IACJ,CAAC;CACF,CAAA;AAxEY,4CAAgB;2BAAhB,gBAAgB;IAP5B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,QAAQ,EAAE;YACR,WAAW,EAAE,mGAAmG;SACjH;QACD,MAAM,EAAE,sBAAsB;KAC/B,CAAC;GACW,gBAAgB,CAwE5B"}

package/dist/types/claude.types.d.ts

@@ -1,4 +1,5 @@
 import Anthropic from '@anthropic-ai/sdk';
+import type { ToolCallsMap } from '@loopstack/common';
 export interface ClaudeModelConfig {
     model?: string;
     envApiKey?: string;
@@ -15,5 +16,23 @@ export interface ClaudeToolDefinition {
     description: string;
     input_schema: Anthropic.Tool['input_schema'];
 }
+export interface ClaudeGenerateTextResult extends Anthropic.Message {
+    toolCalls?: ToolCallsMap;
+}
+export interface DelegateToolCallsResult {
+    allCompleted: boolean;
+    toolResults: DelegateToolResultEntry[];
+    message: {
+        id?: string;
+        content: Anthropic.ContentBlock[];
+    };
+    pendingCount: number;
+}
+export interface DelegateToolResultEntry {
+    type: 'tool_result';
+    tool_use_id: string;
+    content?: string;
+    is_error?: boolean;
+}
 export { default as Anthropic } from '@anthropic-ai/sdk';
 //# sourceMappingURL=claude.types.d.ts.map

package/dist/types/claude.types.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"claude.types.d.ts","sourceRoot":"","sources":["../../src/types/claude.types.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAK1C,MAAM,WAAW,iBAAiB;IAEhC,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAKD,MAAM,WAAW,qBAAqB;IACpC,QAAQ,EAAE,SAAS,CAAC,YAAY,EAAE,CAAC;IACnC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,SAAS,CAAC,IAAI,EAAE,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,SAAS,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAAC;CAC3D;AAKD,MAAM,WAAW,oBAAoB;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;CAC9C;AAGD,OAAO,EAAE,OAAO,IAAI,SAAS,EAAE,MAAM,mBAAmB,CAAC"}
+{"version":3,"file":"claude.types.d.ts","sourceRoot":"","sources":["../../src/types/claude.types.ts"],"names":[],"mappings":"AAAA,OAAO,SAAS,MAAM,mBAAmB,CAAC;AAC1C,OAAO,KAAK,EAAE,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAKtD,MAAM,WAAW,iBAAiB;IAEhC,KAAK,CAAC,EAAE,MAAM,CAAC;IAEf,SAAS,CAAC,EAAE,MAAM,CAAC;CACpB;AAKD,MAAM,WAAW,qBAAqB;IACpC,QAAQ,EAAE,SAAS,CAAC,YAAY,EAAE,CAAC;IACnC,MAAM,CAAC,EAAE,MAAM,CAAC;IAChB,KAAK,CAAC,EAAE,SAAS,CAAC,IAAI,EAAE,CAAC;IACzB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,UAAU,CAAC,EAAE,SAAS,CAAC,mBAAmB,CAAC,aAAa,CAAC,CAAC;CAC3D;AAKD,MAAM,WAAW,oBAAoB;IACnC,IAAI,EAAE,MAAM,CAAC;IACb,WAAW,EAAE,MAAM,CAAC;IACpB,YAAY,EAAE,SAAS,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;CAC9C;AAMD,MAAM,WAAW,wBAAyB,SAAQ,SAAS,CAAC,OAAO;IACjE,SAAS,CAAC,EAAE,YAAY,CAAC;CAC1B;AAKD,MAAM,WAAW,uBAAuB;IACtC,YAAY,EAAE,OAAO,CAAC;IACtB,WAAW,EAAE,uBAAuB,EAAE,CAAC;IACvC,OAAO,EAAE;QAAE,EAAE,CAAC,EAAE,MAAM,CAAC;QAAC,OAAO,EAAE,SAAS,CAAC,YAAY,EAAE,CAAA;KAAE,CAAC;IAC5D,YAAY,EAAE,MAAM,CAAC;CACtB;AAED,MAAM,WAAW,uBAAuB;IACtC,IAAI,EAAE,aAAa,CAAC;IACpB,WAAW,EAAE,MAAM,CAAC;IACpB,OAAO,CAAC,EAAE,MAAM,CAAC;IACjB,QAAQ,CAAC,EAAE,OAAO,CAAC;CACpB;AAGD,OAAO,EAAE,OAAO,IAAI,SAAS,EAAE,MAAM,mBAAmB,CAAC"}

package/dist/types/claude.types.js.map

@@ -1 +1 @@
-{"version":3,"file":"claude.types.js","sourceRoot":"","sources":["../../src/types/claude.types.ts"],"names":[],"mappings":";;;;;;AAiCA,yCAAyD;AAAhD,iHAAA,OAAO,OAAa"}
+{"version":3,"file":"claude.types.js","sourceRoot":"","sources":["../../src/types/claude.types.ts"],"names":[],"mappings":";;;;;;AA2DA,yCAAyD;AAAhD,iHAAA,OAAO,OAAa"}

package/package.json

@@ -10,7 +10,7 @@
     "loopstack",
     "tool"
   ],
-  "version": "0.21.1",
+  "version": "0.22.1",
   "license": "Apache-2.0",
   "author": {
     "name": "Jakob Klippel",
@@ -31,12 +31,12 @@
     "watch": "nest build --watch"
   },
   "dependencies": {
-    "@anthropic-ai/sdk": "^0.39.0",
-    "@loopstack/common": "^0.24.0",
-    "@loopstack/core": "^0.24.0",
-    "@nestjs/common": "^11.1.14",
-    "@nestjs/config": "^4.0.3",
-    "@nestjs/core": "^11.1.14",
+    "@anthropic-ai/sdk": "^0.88.0",
+    "@loopstack/common": "^0.25.1",
+    "@loopstack/core": "^0.25.1",
+    "@nestjs/common": "^11.1.19",
+    "@nestjs/config": "^4.0.4",
+    "@nestjs/core": "^11.1.19",
     "zod": "^4.3.6"
   },
   "files": [
@@ -52,7 +52,12 @@
     "rootDir": "src",
     "testRegex": ".*\\.spec\\.ts$",
     "transform": {
-      "^.+\\.ts$": "ts-jest"
+      "^.+\\.ts$": [
+        "ts-jest",
+        {
+          "tsconfig": "tsconfig.spec.json"
+        }
+      ]
     },
     "collectCoverageFrom": [
       "**/*.(t|j)s"

package/src/claude.module.ts

@@ -1,6 +1,5 @@
 import { Module } from '@nestjs/common';
 import { LoopCoreModule } from '@loopstack/core';
-import { ClaudeMessageDocument } from './documents';
 import { ClaudeClientService, ClaudeMessagesHelperService, ClaudeToolsHelperService } from './services';
 import {
   ClaudeGenerateDocument,
@@ -24,17 +23,7 @@ import {
     ClaudeGenerateDocument,
     DelegateToolCalls,
     UpdateToolResult,
-
-    // documents
-    ClaudeMessageDocument,
-  ],
-  exports: [
-    ClaudeGenerateText,
-    ClaudeGenerateObject,
-    ClaudeGenerateDocument,
-    DelegateToolCalls,
-    UpdateToolResult,
-    ClaudeMessageDocument,
   ],
+  exports: [ClaudeGenerateText, ClaudeGenerateObject, ClaudeGenerateDocument, DelegateToolCalls, UpdateToolResult],
 })
 export class ClaudeModule {}

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

@@ -1,6 +1,5 @@
-import { Injectable } from '@nestjs/common';
 import { z } from 'zod';
-import { Document, Input } from '@loopstack/common';
+import { Document } from '@loopstack/common';
 
 export const ClaudeMessageDocumentSchema = z.object({
   role: z.enum(['user', 'assistant']),
@@ -10,17 +9,12 @@ export const ClaudeMessageDocumentSchema = z.object({
 
 export type ClaudeMessageDocumentContentType = z.infer<typeof ClaudeMessageDocumentSchema>;
 
-@Injectable()
 @Document({
-  config: {
-    type: 'document',
-    description: 'Claude Message Document. Stores messages in Anthropic native format.',
-  },
-  configFile: __dirname + '/claude-message-document.yaml',
+  schema: ClaudeMessageDocumentSchema,
+  uiConfig: __dirname + '/claude-message-document.yaml',
 })
 export class ClaudeMessageDocument {
-  @Input({
-    schema: ClaudeMessageDocumentSchema,
-  })
-  content: z.infer<typeof ClaudeMessageDocumentSchema>;
+  role: 'user' | 'assistant';
+  content: string | any[];
+  toolResults?: any[];
 }

package/src/documents/claude-message-document.yaml

@@ -1,5 +1,5 @@
 tags:
   - message
 ui:
-  form:
-    widget: claude-message
+  widgets:
+    - widget: claude-message

package/src/index.ts

@@ -1,3 +1,4 @@
 export * from './claude.module';
 export * from './tools';
 export * from './documents';
+export * from './types';

package/src/services/claude-tools-helper.service.ts

@@ -3,7 +3,7 @@ import { Injectable } from '@nestjs/common';
 import { toJSONSchema } from 'zod';
 import { ToolInterface, WorkflowInterface, getBlockArgsSchema, getBlockConfig } from '@loopstack/common';
 import { getBlockTool } from '@loopstack/common';
-import { WorkflowType } from '@loopstack/contracts/dist/types';
+import { WorkflowType } from '@loopstack/contracts/types';
 
 @Injectable()
 export class ClaudeToolsHelperService {

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

@@ -1,15 +1,4 @@
-import {
-  InjectTool,
-  Input,
-  RunContext,
-  Tool,
-  ToolInterface,
-  ToolResult,
-  WorkflowInterface,
-  WorkflowMetadataInterface,
-  getBlockTool,
-} from '@loopstack/common';
-import { CreateDocument } from '@loopstack/core';
+import { BaseTool, InjectTool, Tool, ToolResult } from '@loopstack/common';
 import {
   ClaudeGenerateObject,
   ClaudeGenerateObjectArgsType,
@@ -17,49 +6,23 @@ import {
 } from './claude-generate-object.tool';
 
 @Tool({
-  config: {
+  uiConfig: {
     description: 'Generates a structured object using the Anthropic Claude API and creates it as a document',
   },
+  schema: ClaudeGenerateObjectSchema,
 })
-export class ClaudeGenerateDocument implements ToolInterface<ClaudeGenerateObjectArgsType> {
+export class ClaudeGenerateDocument extends BaseTool {
   @InjectTool() private claudeGenerateObject!: ClaudeGenerateObject;
-  @InjectTool() private createDocument!: CreateDocument;
-
-  @Input({
-    schema: ClaudeGenerateObjectSchema,
-  })
-  args: ClaudeGenerateObjectArgsType;
-
-  private getRequiredTool(name: string): ToolInterface {
-    const tool = getBlockTool<ToolInterface>(this, name);
-    if (tool === undefined) {
-      throw new Error(`Tool "${name}" is not available`);
-    }
-    return tool;
-  }
 
-  async execute(
-    args: ClaudeGenerateObjectArgsType,
-    ctx: RunContext,
-    parent: WorkflowInterface,
-    metadata: WorkflowMetadataInterface,
-  ): Promise<ToolResult> {
-    const generateResult = await this.getRequiredTool('claudeGenerateObject').execute(args, ctx, parent, metadata);
-    const documentResult = await this.getRequiredTool('createDocument').execute(
-      {
-        id: args.response.id,
-        document: args.response.document,
-        update: {
-          content: generateResult.data as unknown,
-        },
-      },
-      ctx,
-      parent,
-      metadata,
-    );
+  async call(args: ClaudeGenerateObjectArgsType): Promise<ToolResult> {
+    const generateResult = await this.claudeGenerateObject.call(args);
+    const entity = await this.repository.save(args.response.document, generateResult.data as Record<string, unknown>, {
+      id: args.response.id,
+      validate: 'skip',
+    });
 
     return {
-      ...documentResult,
+      data: entity,
       metadata: generateResult.metadata,
     };
   }