@loopstack/ai-module 0.16.4 → 0.18.0-rc.1

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

@@ -1,70 +1,30 @@
 import { z } from 'zod';
 import { ToolResult } from '@loopstack/common';
 import { ToolBase, WorkflowBase } from '@loopstack/core';
+import { WorkflowExecution } from '@loopstack/core/dist/workflow-processor/interfaces/workflow-execution.interface';
 import { AiMessagesHelperService } from '../services';
 import { AiProviderModelHelperService } from '../services';
 import { AiToolsHelperService } from '../services';
-import { WorkflowExecution } from '@loopstack/core/dist/workflow-processor/interfaces/workflow-execution.interface';
 export declare const AiGenerateTextSchema: z.ZodObject<{
     llm: z.ZodOptional<z.ZodObject<{
         model: z.ZodOptional<z.ZodString>;
         provider: z.ZodOptional<z.ZodString>;
         envApiKey: z.ZodOptional<z.ZodString>;
         cacheResponse: z.ZodOptional<z.ZodBoolean>;
-    }, "strip", z.ZodTypeAny, {
-        model?: string | undefined;
-        provider?: string | undefined;
-        envApiKey?: string | undefined;
-        cacheResponse?: boolean | undefined;
-    }, {
-        model?: string | undefined;
-        provider?: string | undefined;
-        envApiKey?: string | undefined;
-        cacheResponse?: boolean | undefined;
-    }>>;
+    }, z.core.$strip>>;
     messages: z.ZodOptional<z.ZodArray<z.ZodObject<{
-        role: z.ZodEnum<["system", "assistant", "user", "tool"]>;
+        role: z.ZodEnum<{
+            system: "system";
+            user: "user";
+            assistant: "assistant";
+            tool: "tool";
+        }>;
         content: z.ZodAny;
-    }, "strip", z.ZodTypeAny, {
-        role: "system" | "assistant" | "user" | "tool";
-        content?: any;
-    }, {
-        role: "system" | "assistant" | "user" | "tool";
-        content?: any;
-    }>, "many">>;
+    }, z.core.$strip>>>;
     prompt: z.ZodOptional<z.ZodString>;
     messagesSearchTag: z.ZodOptional<z.ZodString>;
-} & {
-    tools: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
-}, "strict", z.ZodTypeAny, {
-    llm?: {
-        model?: string | undefined;
-        provider?: string | undefined;
-        envApiKey?: string | undefined;
-        cacheResponse?: boolean | undefined;
-    } | undefined;
-    messages?: {
-        role: "system" | "assistant" | "user" | "tool";
-        content?: any;
-    }[] | undefined;
-    prompt?: string | undefined;
-    messagesSearchTag?: string | undefined;
-    tools?: string[] | undefined;
-}, {
-    llm?: {
-        model?: string | undefined;
-        provider?: string | undefined;
-        envApiKey?: string | undefined;
-        cacheResponse?: boolean | undefined;
-    } | undefined;
-    messages?: {
-        role: "system" | "assistant" | "user" | "tool";
-        content?: any;
-    }[] | undefined;
-    prompt?: string | undefined;
-    messagesSearchTag?: string | undefined;
-    tools?: string[] | undefined;
-}>;
+    tools: z.ZodOptional<z.ZodArray<z.ZodString>>;
+}, z.core.$strict>;
 type AiGenerateTextArgsType = z.infer<typeof AiGenerateTextSchema>;
 export declare class AiGenerateText extends ToolBase<AiGenerateTextArgsType> {
     private readonly aiMessagesHelperService;

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

@@ -10,14 +10,15 @@ var __metadata = (this && this.__metadata) || function (k, v) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AiGenerateText = exports.AiGenerateTextSchema = void 0;
+const common_1 = require("@nestjs/common");
+const ai_1 = require("ai");
 const zod_1 = require("zod");
-const common_1 = require("@loopstack/common");
+const common_2 = require("@loopstack/common");
 const core_1 = require("@loopstack/core");
+const ai_generate_tool_base_schema_1 = require("../schemas/ai-generate-tool-base.schema");
 const services_1 = require("../services");
 const services_2 = require("../services");
 const services_3 = require("../services");
-const ai_1 = require("ai");
-const ai_generate_tool_base_schema_1 = require("../schemas/ai-generate-tool-base.schema");
 exports.AiGenerateTextSchema = ai_generate_tool_base_schema_1.AiGenerateToolBaseSchema.extend({
     tools: zod_1.z.array(zod_1.z.string()).optional(),
 }).strict();
@@ -33,19 +34,22 @@ let AiGenerateText = class AiGenerateText extends core_1.ToolBase {
     }
     async execute(args, ctx, parent) {
         const model = this.aiProviderModelHelperService.getProviderModel(args.llm);
-        const options = {};
-        options.tools = args.tools
-            ? this.aiToolsHelperService.getTools(args.tools, parent)
-            : undefined;
+        const options = {
+            messages: [],
+        };
+        options.tools = args.tools ? this.aiToolsHelperService.getTools(args.tools, parent) : undefined;
         if (args.prompt) {
-            options.prompt = args.prompt;
+            options.messages.push({
+                role: 'user',
+                content: args.prompt,
+            });
         }
         else {
             const messages = this.aiMessagesHelperService.getMessages(ctx.state.getMetadata('documents'), {
                 messages: args.messages,
                 messagesSearchTag: args.messagesSearchTag,
             });
-            options.messages = (0, ai_1.convertToModelMessages)(messages, {
+            options.messages = await (0, ai_1.convertToModelMessages)(messages, {
                 tools: options.tools,
             });
         }
@@ -72,7 +76,7 @@ let AiGenerateText = class AiGenerateText extends core_1.ToolBase {
                         resolve(data.responseMessage);
                     },
                 });
-                (async () => {
+                void (async () => {
                     try {
                         const reader = stream.getReader();
                         while (true) {
@@ -96,12 +100,13 @@ let AiGenerateText = class AiGenerateText extends core_1.ToolBase {
 };
 exports.AiGenerateText = AiGenerateText;
 exports.AiGenerateText = AiGenerateText = __decorate([
-    (0, common_1.BlockConfig)({
+    (0, common_1.Injectable)(),
+    (0, common_2.BlockConfig)({
         config: {
             description: 'Generates text using a LLM',
         },
     }),
-    (0, common_1.WithArguments)(exports.AiGenerateTextSchema),
+    (0, common_2.WithArguments)(exports.AiGenerateTextSchema),
     __metadata("design:paramtypes", [services_1.AiMessagesHelperService,
         services_3.AiToolsHelperService,
         services_2.AiProviderModelHelperService])

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":";;;;;;;;;;;;AAAA,6BAAwB;AACxB,8CAA2E;AAC3E,0CAAyD;AACzD,0CAAsD;AACtD,0CAA2D;AAC3D,0CAAmD;AACnD,2BAKY;AAEZ,0FAAmF;AAGtE,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,cAAe,SAAQ,eAAgC;IAE/C;IACA;IACA;IAHnB,YACmB,uBAAgD,EAChD,oBAA0C,EAC1C,4BAA0D;QAE3E,KAAK,EAAE,CAAC;QAJS,4BAAuB,GAAvB,uBAAuB,CAAyB;QAChD,yBAAoB,GAApB,oBAAoB,CAAsB;QAC1C,iCAA4B,GAA5B,4BAA4B,CAA8B;IAG7E,CAAC;IAED,KAAK,CAAC,OAAO,CACX,IAA4B,EAC5B,GAAsB,EACtB,MAAoB;QAEpB,MAAM,KAAK,GAAG,IAAI,CAAC,4BAA4B,CAAC,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE3E,MAAM,OAAO,GAIT,EAAE,CAAC;QAEP,OAAO,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK;YACxB,CAAC,CAAC,IAAI,CAAC,oBAAoB,CAAC,QAAQ,CAAC,IAAI,CAAC,KAAK,EAAE,MAAM,CAAC;YACxD,CAAC,CAAC,SAAS,CAAC;QAEd,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,MAAM,GAAG,IAAI,CAAC,MAAM,CAAC;QAC/B,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,WAAW,CACvD,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,WAAW,CAAC,EAClC;gBACE,QAAQ,EAAE,IAAI,CAAC,QAA0B;gBACzC,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;aAC1C,CACF,CAAC;YACF,OAAO,CAAC,QAAQ,GAAG,IAAA,2BAAsB,EAAC,QAAQ,EAAE;gBAClD,KAAK,EAAE,OAAO,CAAC,KAAK;aACrB,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,KAAU,EACV,OAAY;QAEZ,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,MAAM,MAAM,GAAG,IAAA,eAAU,EAAC;gBACxB,KAAK;gBACL,GAAG,OAAO;aACX,CAAC,CAAC;YAEH,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,CAAC,KAAK,IAAI,EAAE;oBACV,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;AA5FY,wCAAc;yBAAd,cAAc;IAN1B,IAAA,oBAAW,EAAC;QACX,MAAM,EAAE;YACN,WAAW,EAAE,4BAA4B;SAC1C;KACF,CAAC;IACD,IAAA,sBAAa,EAAC,4BAAoB,CAAC;qCAGU,kCAAuB;QAC1B,+BAAoB;QACZ,uCAA4B;GAJlE,cAAc,CA4F1B"}
+{"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,8CAA2E;AAC3E,0CAAyD;AAEzD,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,cAAe,SAAQ,eAAgC;IAE/C;IACA;IACA;IAHnB,YACmB,uBAAgD,EAChD,oBAA0C,EAC1C,4BAA0D;QAE3E,KAAK,EAAE,CAAC;QAJS,4BAAuB,GAAvB,uBAAuB,CAAyB;QAChD,yBAAoB,GAApB,oBAAoB,CAAsB;QAC1C,iCAA4B,GAA5B,4BAA4B,CAA8B;IAG7E,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,IAA4B,EAAE,GAAsB,EAAE,MAAoB;QACtF,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;AA7FY,wCAAc;yBAAd,cAAc;IAP1B,IAAA,mBAAU,GAAE;IACZ,IAAA,oBAAW,EAAC;QACX,MAAM,EAAE;YACN,WAAW,EAAE,4BAA4B;SAC1C;KACF,CAAC;IACD,IAAA,sBAAa,EAAC,4BAAoB,CAAC;qCAGU,kCAAuB;QAC1B,+BAAoB;QACZ,uCAA4B;GAJlE,cAAc,CA6F1B"}

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

@@ -1,5 +1,5 @@
-import { ToolResult } from '@loopstack/common';
 import { z } from 'zod';
+import { ToolResult } from '@loopstack/common';
 import { ToolBase, WorkflowBase } from '@loopstack/core';
 import { WorkflowExecution } from '@loopstack/core/dist/workflow-processor/interfaces/workflow-execution.interface';
 declare const DelegateToolCallsToolSchema: z.ZodObject<{
@@ -9,49 +9,9 @@ declare const DelegateToolCallsToolSchema: z.ZodObject<{
             type: z.ZodString;
             input: z.ZodOptional<z.ZodAny>;
             toolCallId: z.ZodOptional<z.ZodString>;
-        }, "strip", z.ZodTypeAny, {
-            type: string;
-            input?: any;
-            toolCallId?: string | undefined;
-        }, {
-            type: string;
-            input?: any;
-            toolCallId?: string | undefined;
-        }>, "many">;
-    }, "strip", z.ZodTypeAny, {
-        id: string;
-        parts: {
-            type: string;
-            input?: any;
-            toolCallId?: string | undefined;
-        }[];
-    }, {
-        id: string;
-        parts: {
-            type: string;
-            input?: any;
-            toolCallId?: string | undefined;
-        }[];
-    }>;
-}, "strip", z.ZodTypeAny, {
-    message: {
-        id: string;
-        parts: {
-            type: string;
-            input?: any;
-            toolCallId?: string | undefined;
-        }[];
-    };
-}, {
-    message: {
-        id: string;
-        parts: {
-            type: string;
-            input?: any;
-            toolCallId?: string | undefined;
-        }[];
-    };
-}>;
+        }, z.core.$strip>>;
+    }, z.core.$strip>;
+}, z.core.$strip>;
 type DelegateToolCallsToolArgs = z.infer<typeof DelegateToolCallsToolSchema>;
 export declare class DelegateToolCall extends ToolBase<DelegateToolCallsToolArgs> {
     execute(args: DelegateToolCallsToolArgs, ctx: WorkflowExecution, parent: WorkflowBase): Promise<ToolResult>;

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

@@ -7,8 +7,8 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DelegateToolCall = void 0;
-const common_1 = require("@loopstack/common");
 const zod_1 = require("zod");
+const common_1 = require("@loopstack/common");
 const core_1 = require("@loopstack/core");
 const DelegateToolCallsToolSchema = zod_1.z.object({
     message: zod_1.z.object({

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,8CAA2E;AAC3E,6BAAwB;AACxB,0CAAyD;AAIzD,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,gBAAiB,SAAQ,eAAmC;IACvE,KAAK,CAAC,OAAO,CACX,IAA+B,EAC/B,GAAsB,EACtB,MAAoB;QAEpB,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,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YACtC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,IAAI,KAAK,CAAC,QAAQ,QAAQ,aAAa,CAAC,CAAC;YACjD,CAAC;YACD,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;YAE3D,WAAW,CAAC,IAAI,CAAC;gBACf,IAAI,EAAE,IAAI,CAAC,IAAW;gBACtB,UAAU,EAAE,IAAI,CAAC,UAAU;gBAC3B,MAAM,EAAE;oBACN,IAAI,EAAE,MAAM,CAAC,IAAI,IAAI,MAAM;oBAC3B,KAAK,EAAE,MAAM,CAAC,IAAI;iBACnB;gBACD,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,KAAK,EAAE,kBAAkB;aACL,CAAC,CAAC;QAC1B,CAAC;QAED,MAAM,aAAa,GAAG;YACpB,EAAE,EAAE,IAAI,CAAC,OAAO,CAAC,EAAE;YACnB,IAAI,EAAE,WAAW;YACjB,KAAK,EAAE,WAAW;SACC,CAAC;QAEtB,OAAO;YACL,IAAI,EAAE,aAAa;SACpB,CAAC;IACJ,CAAC;CACF,CAAA;AAhDY,4CAAgB;2BAAhB,gBAAgB;IAN5B,IAAA,oBAAW,EAAC;QACX,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":";;;;;;;;;AACA,6BAAwB;AACxB,8CAA2E;AAC3E,0CAAyD;AAGzD,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,gBAAiB,SAAQ,eAAmC;IACvE,KAAK,CAAC,OAAO,CAAC,IAA+B,EAAE,GAAsB,EAAE,MAAoB;QACzF,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,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YACtC,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;AA5CY,4CAAgB;2BAAhB,gBAAgB;IAN5B,IAAA,oBAAW,EAAC;QACX,MAAM,EAAE;YACN,WAAW,EAAE,uBAAuB;SACrC;KACF,CAAC;IACD,IAAA,sBAAa,EAAC,2BAA2B,CAAC;GAC9B,gBAAgB,CA4C5B"}

package/package.json

@@ -2,84 +2,52 @@
   "name": "@loopstack/ai-module",
   "displayName": "Loopstack Ai Module",
   "description": "A collection of useful tools for performing AI actions using model provider APIs such as OpenAI and Anthropic",
-  "version": "0.16.4",
-  "author": {
-    "name": "Jakob Klippel",
-    "url": "https://www.linkedin.com/in/jakob-klippel/"
-  },
   "keywords": [
-    "loopstack",
-    "tool",
     "ai",
+    "anthropic",
     "llm",
+    "loopstack",
     "openai",
-    "anthropic"
+    "tool"
   ],
+  "version": "0.18.0-rc.1",
   "license": "Apache-2.0",
+  "author": {
+    "name": "Jakob Klippel",
+    "url": "https://www.linkedin.com/in/jakob-klippel/"
+  },
   "main": "dist/index.js",
   "types": "dist/index.d.ts",
+  "exports": {
+    ".": "./dist/index.js",
+    "./src/*": "./src/*"
+  },
   "scripts": {
     "build": "nest build",
-    "watch": "nest build --watch",
     "compile": "tsc --noEmit",
-    "format": "prettier --write \"src/**/*.ts\" \"test/**/*.ts\"",
-    "lint": "eslint \"{src,apps,libs,test}/**/*.ts\" --fix",
-    "test": "jest",
-    "test:watch": "jest --watch",
-    "test:cov": "jest --coverage",
-    "test:debug": "node --inspect-brk -r tsconfig-paths/register -r ts-node/register node_modules/.bin/jest --runInBand",
-    "test:e2e": "jest --config ./test/jest-e2e.json"
-  },
-  "peerDependencies": {
-    "@nestjs/common": "^11.0.0",
-    "@nestjs/config": "^4.0.1",
-    "@nestjs/core": "^11.0.0",
-    "@nestjs/platform-express": "^11.0.0",
-    "rxjs": "^7.8.0"
+    "format": "prettier --write .",
+    "lint": "eslint .",
+    "test": "jest --passWithNoTests",
+    "watch": "nest build --watch"
   },
   "dependencies": {
-    "@ai-sdk/anthropic": "^2.0.1",
-    "@ai-sdk/openai": "^2.0.4",
-    "@ai-sdk/provider-utils": "^3.0.8",
-    "@loopstack/common": "^0.17.0",
-    "@loopstack/core": "^0.17.0",
-    "@loopstack/core-ui-module": "^0.17.0",
-    "ai": "^5.0.6",
-    "lodash": "^4.17.21",
-    "zod": "^3.24.1"
-  },
-  "devDependencies": {
-    "@eslint/eslintrc": "^3.3.0",
-    "@eslint/js": "^9.22.0",
-    "@nestjs/cli": "^11.0.5",
-    "@nestjs/common": "^11.0.12",
-    "@nestjs/config": "^4.0.1",
-    "@nestjs/core": "^11.0.12",
-    "@nestjs/platform-express": "^11.0.12",
-    "@nestjs/schematics": "^11.0.2",
-    "@nestjs/testing": "^11.0.12",
-    "@swc/cli": "^0.6.0",
-    "@swc/core": "^1.11.11",
-    "@types/express": "^5.0.1",
-    "@types/jest": "^29.5.14",
-    "@types/node": "^22.13.10",
-    "@types/supertest": "^6.0.2",
-    "eslint": "^9.22.0",
-    "eslint-config-prettier": "^10.1.1",
-    "eslint-plugin-prettier": "^5.2.3",
-    "globals": "^16.0.0",
-    "jest": "^29.7.0",
-    "prettier": "^3.5.3",
-    "rxjs": "^7.8.2",
-    "source-map-support": "^0.5.21",
-    "supertest": "^7.0.0",
-    "ts-jest": "^29.2.6",
-    "ts-loader": "^9.5.2",
-    "ts-node": "^10.9.2",
-    "tsconfig-paths": "^4.2.0",
-    "typescript": "^5.8.3",
-    "typescript-eslint": "^8.38.0"
+    "@ai-sdk/anthropic": "^3.0.23",
+    "@ai-sdk/openai": "^3.0.19",
+    "@ai-sdk/provider-utils": "^4.0.9",
+    "@loopstack/common": "^0.18.0-rc.2",
+    "@loopstack/core": "^0.18.0-rc.2",
+    "@loopstack/core-ui-module": "^0.18.0-rc.1",
+    "@nestjs/common": "^11.1.12",
+    "@nestjs/config": "^4.0.2",
+    "@nestjs/core": "^11.1.12",
+    "ai": "^6.0.48",
+    "lodash": "^4.17.23",
+    "zod": "^4.3.5"
   },
+  "files": [
+    "dist",
+    "src"
+  ],
   "jest": {
     "moduleFileExtensions": [
       "js",
@@ -89,12 +57,13 @@
     "rootDir": "src",
     "testRegex": ".*\\.spec\\.ts$",
     "transform": {
-      "^.+\\.(t|j)s$": "ts-jest"
+      "^.+\\.ts$": "ts-jest"
     },
     "collectCoverageFrom": [
       "**/*.(t|j)s"
     ],
     "coverageDirectory": "../coverage",
-    "testEnvironment": "node"
+    "testEnvironment": "node",
+    "maxWorkers": 1
   }
 }

package/src/ai.module.ts

@@ -0,0 +1,38 @@
+import { Module } from '@nestjs/common';
+import { DiscoveryModule } from '@nestjs/core';
+import { LoopCoreModule } from '@loopstack/core';
+import { CoreUiModule } from '@loopstack/core-ui-module';
+import { AiMessageDocument } from './documents';
+import { AnthropicProviderService } from './providers/anthropic.provider';
+import { OpenAiProviderService } from './providers/openai.provider';
+import { AiMessagesHelperService } from './services';
+import { AiProviderModelHelperService } from './services';
+import { AiToolsHelperService } from './services';
+import { AiProviderRegistryService } from './services/ai-provider-registry.service';
+import { AiGenerateDocument, AiGenerateObject, AiGenerateText, DelegateToolCall } from './tools';
+
+@Module({
+  imports: [LoopCoreModule, CoreUiModule, DiscoveryModule],
+  providers: [
+    // services
+    AiMessagesHelperService,
+    AiProviderModelHelperService,
+    AiToolsHelperService,
+    AiProviderRegistryService,
+
+    // ai providers
+    OpenAiProviderService,
+    AnthropicProviderService,
+
+    // tools
+    AiGenerateDocument,
+    AiGenerateObject,
+    AiGenerateText,
+    DelegateToolCall,
+
+    // documents
+    AiMessageDocument,
+  ],
+  exports: [AiGenerateDocument, AiGenerateObject, AiGenerateText, DelegateToolCall, AiMessageDocument],
+})
+export class AiModule {}

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

@@ -0,0 +1,22 @@
+import { Injectable } from '@nestjs/common';
+import { z } from 'zod';
+import { BlockConfig, WithArguments } from '@loopstack/common';
+import { DocumentBase } from '@loopstack/core';
+
+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()),
+});
+
+@Injectable()
+@BlockConfig({
+  config: {
+    type: 'document',
+    description: 'Ai Message Document.',
+  },
+  configFile: __dirname + '/ai-message-document.yaml',
+})
+@WithArguments(AiMessageDocumentSchema)
+export class AiMessageDocument extends DocumentBase {}

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

@@ -0,0 +1,5 @@
+tags:
+  - message
+ui:
+  form:
+    widget: ai-message

package/src/documents/index.ts

@@ -0,0 +1 @@
+export * from './ai-message-document';

package/src/index.ts

@@ -0,0 +1,3 @@
+export * from './ai.module';
+export * from './tools';
+export * from './documents';

package/src/providers/anthropic.provider.ts

@@ -0,0 +1,18 @@
+import { createAnthropic } from '@ai-sdk/anthropic';
+import { AiProvider, AiProviderInterface, AiProviderOptions } from '@loopstack/common';
+
+@AiProvider({
+  name: 'anthropic',
+})
+export class AnthropicProviderService implements AiProviderInterface {
+  createClient(options: AiProviderOptions) {
+    return createAnthropic({
+      apiKey: options.apiKey,
+      baseURL: options.baseURL,
+    });
+  }
+
+  getModel(client: ReturnType<typeof createAnthropic>, model: string) {
+    return client(model);
+  }
+}

package/src/providers/openai.provider.ts

@@ -0,0 +1,18 @@
+import { createOpenAI } from '@ai-sdk/openai';
+import { AiProvider, AiProviderInterface, AiProviderOptions } from '@loopstack/common';
+
+@AiProvider({
+  name: 'openai',
+})
+export class OpenAiProviderService implements AiProviderInterface {
+  createClient(options: AiProviderOptions) {
+    return createOpenAI({
+      apiKey: options.apiKey,
+      baseURL: options.baseURL,
+    });
+  }
+
+  getModel(client: ReturnType<typeof createOpenAI>, model: string) {
+    return client(model);
+  }
+}

package/src/schemas/ai-generate-tool-base.schema.ts

@@ -0,0 +1,22 @@
+import { z } from 'zod';
+
+export const AiGenerateToolBaseSchema = z.object({
+  llm: z
+    .object({
+      model: z.string().optional(),
+      provider: z.string().optional(),
+      envApiKey: z.string().optional(),
+      cacheResponse: z.boolean().optional(),
+    })
+    .optional(),
+  messages: z
+    .array(
+      z.object({
+        role: z.enum(['system', 'assistant', 'user', 'tool']),
+        content: z.any(),
+      }),
+    )
+    .optional(),
+  prompt: z.string().optional(),
+  messagesSearchTag: z.string().optional(),
+});

package/src/services/ai-messages-helper.service.ts

@@ -0,0 +1,28 @@
+import { Injectable } from '@nestjs/common';
+import { UIDataTypes, UIMessage, UITools } from 'ai';
+import { DocumentEntity } from '@loopstack/common';
+
+@Injectable()
+export class AiMessagesHelperService {
+  private searchMessages(
+    documents: DocumentEntity[],
+    tag: string,
+  ): Omit<UIMessage<unknown, UIDataTypes, UITools>, 'id'>[] {
+    return documents
+      .filter((document) => document.tags?.includes(tag))
+      .sort((a, b) => a.createdAt.valueOf() - b.createdAt.valueOf())
+      .map((document) => document.content as Omit<UIMessage<unknown, UIDataTypes, UITools>, 'id'>);
+  }
+
+  getMessages(
+    documents: DocumentEntity[],
+    args: { messages?: Omit<UIMessage<unknown, UIDataTypes, UITools>, 'id'>[]; messagesSearchTag?: string },
+  ): Omit<UIMessage<unknown, UIDataTypes, UITools>, 'id'>[] {
+    let messages: Omit<UIMessage<unknown, UIDataTypes, UITools>, 'id'>[] | undefined = args.messages;
+    if (!messages?.length) {
+      messages = this.searchMessages(documents, args.messagesSearchTag ?? 'message');
+    }
+
+    return messages;
+  }
+}

package/src/services/ai-provider-model-helper.service.ts

@@ -0,0 +1,61 @@
+import { Injectable } from '@nestjs/common';
+import { LanguageModel } from 'ai';
+import { AiProviderRegistryService } from './ai-provider-registry.service';
+
+export interface AiProviderModelConfig {
+  model?: string;
+  provider?: string;
+  envApiKey?: string;
+  cacheResponse?: boolean;
+}
+
+@Injectable()
+export class AiProviderModelHelperService {
+  constructor(private readonly aiProviderRegistry: AiProviderRegistryService) {}
+
+  private getApiKey(envApiKey: string | undefined, providerName: string): string {
+    const defaultKey = `${providerName.toUpperCase()}_API_KEY`;
+    const apiKey = envApiKey ? process.env[envApiKey] : process.env[defaultKey];
+
+    if (!apiKey) {
+      throw new Error(`No API key found! Please make sure to provide "${envApiKey ?? defaultKey}" in your .env file.`);
+    }
+
+    return apiKey;
+  }
+
+  private getModel(modelName: string | undefined): string {
+    const model = modelName ?? process.env['DEFAULT_MODEL'];
+
+    if (!model) {
+      throw new Error(
+        `No Model defined. Please provide DEFAULT_MODEL or set the model parameter in the completion service call.`,
+      );
+    }
+
+    return model;
+  }
+
+  private getProvider(name: string | undefined): string {
+    const provider = name ?? process.env['DEFAULT_PROVIDER'];
+
+    if (!provider) {
+      throw new Error(
+        `No Provider defined. Please provide DEFAULT_PROVIDER or set the provider parameter in the completion service call.`,
+      );
+    }
+
+    return provider;
+  }
+
+  getProviderModel(config?: AiProviderModelConfig): LanguageModel {
+    const modelName = this.getModel(config?.model);
+    const providerName = this.getProvider(config?.provider);
+    const apiKey = this.getApiKey(config?.envApiKey, providerName);
+
+    return this.aiProviderRegistry.createModel(providerName, {
+      apiKey: apiKey,
+      model: modelName,
+    });
+  }
+}