@loopstack/claude-module 0.21.0 → 0.22.0

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

@@ -0,0 +1,117 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    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 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 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
+        .custom((val) => typeof val === 'function' && (0, common_2.getBlockTypeFromMetadata)(val) === 'document')
+        .optional(),
+    skipResponseMessage: zod_1.z.boolean().optional(),
+    callback: zod_1.z
+        .object({
+        transition: zod_1.z.string(),
+    })
+        .optional(),
+});
+let DelegateToolCalls = DelegateToolCalls_1 = class DelegateToolCalls extends common_2.BaseTool {
+    logger = new common_1.Logger(DelegateToolCalls_1.name);
+    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,
+                },
+            };
+        }
+        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 = [];
+        for (let i = 0; i < toolUseBlocks.length; i++) {
+            const block = toolUseBlocks[i];
+            const result = results[i];
+            if (result.pending) {
+                pendingCount++;
+            }
+            else {
+                toolResults.push({
+                    type: 'tool_result',
+                    tool_use_id: block.id,
+                    content: result.data ? JSON.stringify(result.data, null, 2) : '',
+                    is_error: !!result.error,
+                });
+            }
+        }
+        if (args.document && !args.skipResponseMessage) {
+            await this.createResponseDocument(args.document, args, toolResults);
+        }
+        return {
+            data: {
+                allCompleted: pendingCount === 0,
+                toolResults,
+                message: args.message,
+                pendingCount,
+            },
+        };
+    }
+    async executeTool(block, options) {
+        try {
+            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);
+            this.logger.error(`Tool "${block.name}" failed: ${errorMessage}`);
+            return {
+                data: errorMessage,
+                error: errorMessage,
+            };
+        }
+    }
+    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;
+exports.DelegateToolCalls = DelegateToolCalls = DelegateToolCalls_1 = __decorate([
+    (0, common_1.Injectable)(),
+    (0, common_2.Tool)({
+        uiConfig: {
+            description: 'Delegate tool calls from an LLM response. Handles both sync tools and async task tools.',
+        },
+        schema: DelegateToolCallsSchema,
+    })
+], DelegateToolCalls);
+//# sourceMappingURL=delegate-tool-calls.tool.js.map

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

@@ -0,0 +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,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/index.d.ts

@@ -1,5 +1,6 @@
 export * from './claude-generate-text.tool';
 export * from './claude-generate-object.tool';
 export * from './claude-generate-document.tool';
-export * from './claude-delegate-tool-call.tool';
+export * from './delegate-tool-calls.tool';
+export * from './update-tool-result.tool';
 //# sourceMappingURL=index.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,cAAc,6BAA6B,CAAC;AAC5C,cAAc,+BAA+B,CAAC;AAC9C,cAAc,iCAAiC,CAAC;AAChD,cAAc,kCAAkC,CAAC"}
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,cAAc,6BAA6B,CAAC;AAC5C,cAAc,+BAA+B,CAAC;AAC9C,cAAc,iCAAiC,CAAC;AAChD,cAAc,4BAA4B,CAAC;AAC3C,cAAc,2BAA2B,CAAC"}

package/dist/tools/index.js

@@ -17,5 +17,6 @@ Object.defineProperty(exports, "__esModule", { value: true });
 __exportStar(require("./claude-generate-text.tool"), exports);
 __exportStar(require("./claude-generate-object.tool"), exports);
 __exportStar(require("./claude-generate-document.tool"), exports);
-__exportStar(require("./claude-delegate-tool-call.tool"), exports);
+__exportStar(require("./delegate-tool-calls.tool"), exports);
+__exportStar(require("./update-tool-result.tool"), exports);
 //# sourceMappingURL=index.js.map

package/dist/tools/index.js.map

@@ -1 +1 @@
-{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,8DAA4C;AAC5C,gEAA8C;AAC9C,kEAAgD;AAChD,mEAAiD"}
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,8DAA4C;AAC5C,gEAA8C;AAC9C,kEAAgD;AAChD,6DAA2C;AAC3C,4DAA0C"}

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

@@ -0,0 +1,19 @@
+import { z } from 'zod';
+import { BaseTool, DocumentClass, ToolResult } from '@loopstack/common';
+declare const UpdateToolResultSchema: z.ZodObject<{
+    delegateResult: z.ZodObject<{
+        allCompleted: z.ZodBoolean;
+        toolResults: z.ZodArray<z.ZodAny>;
+        message: z.ZodAny;
+        pendingCount: z.ZodNumber;
+    }, z.core.$strip>;
+    completedTool: z.ZodAny;
+    document: z.ZodOptional<z.ZodCustom<DocumentClass, DocumentClass>>;
+}, z.core.$strip>;
+type UpdateToolResultArgs = z.infer<typeof UpdateToolResultSchema>;
+export declare class UpdateToolResult extends BaseTool {
+    private readonly logger;
+    call(args: UpdateToolResultArgs): Promise<ToolResult>;
+}
+export {};
+//# sourceMappingURL=update-tool-result.tool.d.ts.map

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

@@ -0,0 +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,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

@@ -0,0 +1,89 @@
+"use strict";
+var __decorate = (this && this.__decorate) || function (decorators, target, key, desc) {
+    var c = arguments.length, r = c < 3 ? target : desc === null ? desc = Object.getOwnPropertyDescriptor(target, key) : desc, d;
+    if (typeof Reflect === "object" && typeof Reflect.decorate === "function") r = Reflect.decorate(decorators, target, key, desc);
+    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 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 UpdateToolResultSchema = zod_1.z.object({
+    delegateResult: zod_1.z.object({
+        allCompleted: zod_1.z.boolean(),
+        toolResults: zod_1.z.array(zod_1.z.any()),
+        message: zod_1.z.any(),
+        pendingCount: zod_1.z.number(),
+    }),
+    completedTool: zod_1.z.any(),
+    document: zod_1.z
+        .custom((val) => typeof val === 'function' && (0, common_2.getBlockTypeFromMetadata)(val) === 'document')
+        .optional(),
+});
+let UpdateToolResult = UpdateToolResult_1 = class UpdateToolResult extends common_2.BaseTool {
+    logger = new common_1.Logger(UpdateToolResult_1.name);
+    async call(args) {
+        const { delegateResult } = args;
+        const completedToolRecord = args.completedTool;
+        const subscriberMetadata = completedToolRecord._subscriberMetadata;
+        if (!subscriberMetadata?.toolUseId || !subscriberMetadata?.toolName) {
+            throw new Error('Callback payload is missing _subscriberMetadata with toolUseId and toolName. ' +
+                'Ensure the event subscriber was registered with metadata by delegateToolCalls.');
+        }
+        const { toolUseId, toolName } = subscriberMetadata;
+        const tool = (0, common_2.getBlockTool)(this.ctx.parent, toolName);
+        if (!tool) {
+            throw new Error(`Tool with name ${toolName} not found.`);
+        }
+        let toolResult;
+        try {
+            toolResult = await tool.complete(completedToolRecord);
+        }
+        catch (error) {
+            const errorMessage = error instanceof Error ? error.message : String(error);
+            this.logger.error(`Tool "${toolName}" complete() failed: ${errorMessage}`);
+            toolResult = {
+                data: errorMessage,
+                error: errorMessage,
+            };
+        }
+        const completedEntry = {
+            type: 'tool_result',
+            tool_use_id: toolUseId,
+            content: toolResult.data ? JSON.stringify(toolResult.data, null, 2) : '',
+            is_error: !!toolResult.error,
+        };
+        const updatedResults = [...delegateResult.toolResults, completedEntry];
+        const pendingCount = delegateResult.pendingCount - 1;
+        const allCompleted = pendingCount === 0;
+        if (args.document) {
+            await this.repository.save(args.document, {
+                role: 'assistant',
+                content: delegateResult.message.content,
+                toolResults: updatedResults,
+            }, { id: delegateResult.message.id, validate: 'skip' });
+        }
+        return {
+            data: {
+                ...delegateResult,
+                toolResults: updatedResults,
+                allCompleted,
+                pendingCount,
+            },
+        };
+    }
+};
+exports.UpdateToolResult = UpdateToolResult;
+exports.UpdateToolResult = UpdateToolResult = UpdateToolResult_1 = __decorate([
+    (0, common_1.Injectable)(),
+    (0, common_2.Tool)({
+        uiConfig: {
+            description: 'Handle async tool completion callback. Updates the response document and tracks completion state.',
+        },
+        schema: UpdateToolResultSchema,
+    })
+], UpdateToolResult);
+//# sourceMappingURL=update-tool-result.tool.js.map

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

@@ -0,0 +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,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.0",
+  "version": "0.22.0",
   "license": "Apache-2.0",
   "author": {
     "name": "Jakob Klippel",
@@ -32,8 +32,8 @@
   },
   "dependencies": {
     "@anthropic-ai/sdk": "^0.39.0",
-    "@loopstack/common": "^0.23.0",
-    "@loopstack/core-ui-module": "^0.20.6",
+    "@loopstack/common": "^0.25.0",
+    "@loopstack/core": "^0.25.0",
     "@nestjs/common": "^11.1.14",
     "@nestjs/config": "^4.0.3",
     "@nestjs/core": "^11.1.14",

package/src/claude.module.ts

@@ -1,11 +1,16 @@
 import { Module } from '@nestjs/common';
-import { CoreUiModule } from '@loopstack/core-ui-module';
-import { ClaudeMessageDocument } from './documents';
+import { LoopCoreModule } from '@loopstack/core';
 import { ClaudeClientService, ClaudeMessagesHelperService, ClaudeToolsHelperService } from './services';
-import { ClaudeDelegateToolCall, ClaudeGenerateDocument, ClaudeGenerateObject, ClaudeGenerateText } from './tools';
+import {
+  ClaudeGenerateDocument,
+  ClaudeGenerateObject,
+  ClaudeGenerateText,
+  DelegateToolCalls,
+  UpdateToolResult,
+} from './tools';
 
 @Module({
-  imports: [CoreUiModule],
+  imports: [LoopCoreModule],
   providers: [
     // services
     ClaudeClientService,
@@ -16,17 +21,9 @@ import { ClaudeDelegateToolCall, ClaudeGenerateDocument, ClaudeGenerateObject, C
     ClaudeGenerateText,
     ClaudeGenerateObject,
     ClaudeGenerateDocument,
-    ClaudeDelegateToolCall,
-
-    // documents
-    ClaudeMessageDocument,
-  ],
-  exports: [
-    ClaudeGenerateText,
-    ClaudeGenerateObject,
-    ClaudeGenerateDocument,
-    ClaudeDelegateToolCall,
-    ClaudeMessageDocument,
+    DelegateToolCalls,
+    UpdateToolResult,
   ],
+  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

@@ -22,15 +22,13 @@ export class ClaudeToolsHelperService {
         throw new Error(`Block ${tool.constructor.name} is missing @BlockConfig decorator`);
       }
 
-      if (inputSchema) {
-        const jsonSchema = toJSONSchema(inputSchema);
+      const jsonSchema = inputSchema ? toJSONSchema(inputSchema) : { type: 'object' };
 
-        toolDefinitions.push({
-          name: toolName,
-          description: config.description ?? '',
-          input_schema: jsonSchema as Anthropic.Tool['input_schema'],
-        });
-      }
+      toolDefinitions.push({
+        name: toolName,
+        description: config.description ?? '',
+        input_schema: jsonSchema as Anthropic.Tool['input_schema'],
+      });
     }
 
     return toolDefinitions.length ? toolDefinitions : undefined;

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-ui-module';
+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,
     };
   }

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

@@ -2,16 +2,12 @@ import Anthropic from '@anthropic-ai/sdk';
 import { Inject } from '@nestjs/common';
 import { toJSONSchema, z } from 'zod';
 import {
-  DocumentInterface,
-  Input,
-  RunContext,
+  BaseTool,
+  DocumentClass,
   Tool,
-  ToolInterface,
   ToolResult,
-  WorkflowInterface,
-  WorkflowMetadataInterface,
-  getBlockArgsSchema,
-  getBlockDocument,
+  getBlockTypeFromMetadata,
+  getDocumentSchema,
 } from '@loopstack/common';
 import { ClaudeGenerateToolBaseSchema } from '../schemas/claude-generate-tool-base.schema';
 import { ClaudeClientService } from '../services';
@@ -21,7 +17,9 @@ import { applyCacheBreakpoints } from '../utils/cache.utils';
 export const ClaudeGenerateObjectSchema = ClaudeGenerateToolBaseSchema.extend({
   response: z.object({
     id: z.string().optional(),
-    document: z.string(),
+    document: z.custom<DocumentClass>(
+      (val) => typeof val === 'function' && getBlockTypeFromMetadata(val as object) === 'document',
+    ),
   }),
 }).strict();
 
@@ -30,27 +28,18 @@ export type ClaudeGenerateObjectArgsType = z.infer<typeof ClaudeGenerateObjectSc
 const STRUCTURED_OUTPUT_TOOL_NAME = 'structured_output';
 
 @Tool({
-  config: {
+  uiConfig: {
     description: 'Generates a structured object using the Anthropic Claude API',
   },
+  schema: ClaudeGenerateObjectSchema,
 })
-export class ClaudeGenerateObject implements ToolInterface<ClaudeGenerateObjectArgsType> {
+export class ClaudeGenerateObject extends BaseTool {
   @Inject()
   private readonly claudeClientService: ClaudeClientService;
   @Inject()
   private readonly claudeMessagesHelperService: ClaudeMessagesHelperService;
 
-  @Input({
-    schema: ClaudeGenerateObjectSchema,
-  })
-  args: ClaudeGenerateObjectArgsType;
-
-  async execute(
-    args: ClaudeGenerateObjectArgsType,
-    ctx: RunContext,
-    parent: WorkflowInterface,
-    metadata: WorkflowMetadataInterface,
-  ): Promise<ToolResult> {
+  async call(args: ClaudeGenerateObjectArgsType): Promise<ToolResult> {
     const client = this.claudeClientService.getClient(args.claude);
     const model = this.claudeClientService.getModel(args.claude);
 
@@ -59,19 +48,14 @@ export class ClaudeGenerateObject implements ToolInterface<ClaudeGenerateObjectA
     if (args.prompt) {
       messages.push({ role: 'user', content: args.prompt });
     } else {
-      const resolved = this.claudeMessagesHelperService.getMessages(metadata.documents, {
+      const resolved = this.claudeMessagesHelperService.getMessages(this.ctx.runtime.documents, {
         messages: args.messages as Anthropic.MessageParam[],
         messagesSearchTag: args.messagesSearchTag,
       });
       messages.push(...resolved);
     }
 
-    const document = getBlockDocument<DocumentInterface>(parent, args.response.document);
-    if (!document) {
-      throw new Error(`Document with name "${args.response.document}" not found in tool execution context.`);
-    }
-
-    const responseSchema = getBlockArgsSchema(document);
+    const responseSchema = getDocumentSchema(args.response.document);
     if (!responseSchema) {
       throw new Error('Claude object generation source document must have a schema.');
     }