@loopstack/ai-module 0.15.0

package/dist/services/index.d.ts

@@ -0,0 +1,4 @@
+export * from './ai-messages-helper.service';
+export * from './ai-provider-model-helper.service';
+export * from './ai-messages-helper.service';
+export * from './ai-tools-helper.service';

package/dist/services/index.js

@@ -0,0 +1,21 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./ai-messages-helper.service"), exports);
+__exportStar(require("./ai-provider-model-helper.service"), exports);
+__exportStar(require("./ai-messages-helper.service"), exports);
+__exportStar(require("./ai-tools-helper.service"), exports);
+//# sourceMappingURL=index.js.map

package/dist/services/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/services/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,+DAA6C;AAC7C,qEAAmD;AACnD,+DAA6C;AAC7C,4DAA0C"}

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

@@ -0,0 +1,10 @@
+import { ToolResult } from '@loopstack/common';
+import { Block, ToolBase } from '@loopstack/core';
+import { AiGenerateObject, AiGenerateObjectArgsType } from './ai-generate-object.tool';
+import { WorkflowExecution } from '@loopstack/core/dist/workflow-processor/interfaces/workflow-execution.interface';
+import { CreateDocument } from '@loopstack/core-ui-module';
+export declare class AiGenerateDocument extends ToolBase<AiGenerateObjectArgsType> {
+    aiGenerateObject: AiGenerateObject;
+    createDocument: CreateDocument;
+    execute(args: AiGenerateObjectArgsType, ctx: WorkflowExecution, parent: Block): Promise<ToolResult>;
+}

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

@@ -0,0 +1,50 @@
+"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 __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AiGenerateDocument = void 0;
+const common_1 = require("@nestjs/common");
+const common_2 = require("@loopstack/common");
+const core_1 = require("@loopstack/core");
+const ai_generate_object_tool_1 = require("./ai-generate-object.tool");
+const core_ui_module_1 = require("@loopstack/core-ui-module");
+let AiGenerateDocument = class AiGenerateDocument extends core_1.ToolBase {
+    aiGenerateObject;
+    createDocument;
+    async execute(args, ctx, parent) {
+        const result = await this.getTool('aiGenerateObject').execute(args, ctx, parent);
+        return this.getTool('createDocument').execute({
+            id: args.response.id,
+            document: args.response.document,
+            update: {
+                content: result.data,
+            },
+        }, ctx, parent);
+    }
+};
+exports.AiGenerateDocument = AiGenerateDocument;
+__decorate([
+    (0, common_2.Tool)(),
+    __metadata("design:type", ai_generate_object_tool_1.AiGenerateObject)
+], AiGenerateDocument.prototype, "aiGenerateObject", void 0);
+__decorate([
+    (0, common_2.Tool)(),
+    __metadata("design:type", core_ui_module_1.CreateDocument)
+], AiGenerateDocument.prototype, "createDocument", void 0);
+exports.AiGenerateDocument = AiGenerateDocument = __decorate([
+    (0, common_1.Injectable)(),
+    (0, common_2.BlockConfig)({
+        config: {
+            description: 'Generates a structured object using a LLM and creates it as document',
+        },
+    }),
+    (0, common_2.WithArguments)(ai_generate_object_tool_1.AiGenerateObjectSchema)
+], AiGenerateDocument);
+//# sourceMappingURL=ai-generate-document.tool.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ai-generate-document.tool.js","sourceRoot":"","sources":["../../src/tools/ai-generate-document.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA4C;AAC5C,8CAK2B;AAC3B,0CAAkD;AAClD,uEAImC;AAEnC,8DAA2D;AAUpD,IAAM,kBAAkB,GAAxB,MAAM,kBAAmB,SAAQ,eAAkC;IAChE,gBAAgB,CAAmB;IACnC,cAAc,CAAiB;IAEvC,KAAK,CAAC,OAAO,CACX,IAA8B,EAC9B,GAAsB,EACtB,MAAa;QAEb,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,OAAO,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACjF,OAAO,IAAI,CAAC,OAAO,CAAC,gBAAgB,CAAC,CAAC,OAAO,CAC3C;YACE,EAAE,EAAE,IAAI,CAAC,QAAQ,CAAC,EAAE;YACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ;YAChC,MAAM,EAAE;gBACN,OAAO,EAAE,MAAM,CAAC,IAAI;aACrB;SACF,EACD,GAAG,EACH,MAAM,CACP,CAAC;IACJ,CAAC;CACF,CAAA;AAtBY,gDAAkB;AACrB;IAAP,IAAA,aAAI,GAAE;8BAAmB,0CAAgB;4DAAC;AACnC;IAAP,IAAA,aAAI,GAAE;8BAAiB,+BAAc;0DAAC;6BAF5B,kBAAkB;IAR9B,IAAA,mBAAU,GAAE;IACZ,IAAA,oBAAW,EAAC;QACX,MAAM,EAAE;YACN,WAAW,EACT,sEAAsE;SACzE;KACF,CAAC;IACD,IAAA,sBAAa,EAAC,gDAAsB,CAAC;GACzB,kBAAkB,CAsB9B"}

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

@@ -0,0 +1,89 @@
+import { z } from 'zod';
+import { ToolResult } from '@loopstack/common';
+import { ToolBase, WorkflowBase } from '@loopstack/core';
+import { AiMessagesHelperService } from '../services';
+import { AiProviderModelHelperService } from '../services';
+import { WorkflowExecution } from '@loopstack/core/dist/workflow-processor/interfaces/workflow-execution.interface';
+export declare const AiGenerateObjectSchema: 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;
+        provider?: string;
+        envApiKey?: string;
+        cacheResponse?: boolean;
+    }, {
+        model?: string;
+        provider?: string;
+        envApiKey?: string;
+        cacheResponse?: boolean;
+    }>>;
+    messages: z.ZodOptional<z.ZodArray<z.ZodObject<{
+        role: z.ZodEnum<["system", "assistant", "user", "tool"]>;
+        content: z.ZodAny;
+    }, "strip", z.ZodTypeAny, {
+        role?: "system" | "assistant" | "user" | "tool";
+        content?: any;
+    }, {
+        role?: "system" | "assistant" | "user" | "tool";
+        content?: any;
+    }>, "many">>;
+    prompt: z.ZodOptional<z.ZodString>;
+    messagesSearchTag: z.ZodOptional<z.ZodString>;
+} & {
+    response: z.ZodObject<{
+        id: z.ZodOptional<z.ZodString>;
+        document: z.ZodString;
+    }, "strip", z.ZodTypeAny, {
+        id?: string;
+        document?: string;
+    }, {
+        id?: string;
+        document?: string;
+    }>;
+}, "strict", z.ZodTypeAny, {
+    llm?: {
+        model?: string;
+        provider?: string;
+        envApiKey?: string;
+        cacheResponse?: boolean;
+    };
+    messages?: {
+        role?: "system" | "assistant" | "user" | "tool";
+        content?: any;
+    }[];
+    prompt?: string;
+    messagesSearchTag?: string;
+    response?: {
+        id?: string;
+        document?: string;
+    };
+}, {
+    llm?: {
+        model?: string;
+        provider?: string;
+        envApiKey?: string;
+        cacheResponse?: boolean;
+    };
+    messages?: {
+        role?: "system" | "assistant" | "user" | "tool";
+        content?: any;
+    }[];
+    prompt?: string;
+    messagesSearchTag?: string;
+    response?: {
+        id?: string;
+        document?: string;
+    };
+}>;
+export type AiGenerateObjectArgsType = z.infer<typeof AiGenerateObjectSchema>;
+export declare class AiGenerateObject extends ToolBase<AiGenerateObjectArgsType> {
+    private readonly aiMessagesHelperService;
+    private readonly aiProviderModelHelperService;
+    constructor(aiMessagesHelperService: AiMessagesHelperService, aiProviderModelHelperService: AiProviderModelHelperService);
+    execute(args: AiGenerateObjectArgsType, ctx: WorkflowExecution, parent: WorkflowBase): Promise<ToolResult>;
+    private handleGenerateObject;
+}

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

@@ -0,0 +1,85 @@
+"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 __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AiGenerateObject = exports.AiGenerateObjectSchema = void 0;
+const zod_1 = require("zod");
+const common_1 = require("@loopstack/common");
+const lodash_1 = require("lodash");
+const ai_1 = require("ai");
+const core_1 = require("@loopstack/core");
+const services_1 = require("../services");
+const services_2 = require("../services");
+const ai_generate_tool_base_schema_1 = require("../schemas/ai-generate-tool-base.schema");
+exports.AiGenerateObjectSchema = ai_generate_tool_base_schema_1.AiGenerateToolBaseSchema.extend({
+    response: zod_1.z.object({
+        id: zod_1.z.string().optional(),
+        document: zod_1.z.string(),
+    }),
+}).strict();
+let AiGenerateObject = class AiGenerateObject extends core_1.ToolBase {
+    aiMessagesHelperService;
+    aiProviderModelHelperService;
+    constructor(aiMessagesHelperService, aiProviderModelHelperService) {
+        super();
+        this.aiMessagesHelperService = aiMessagesHelperService;
+        this.aiProviderModelHelperService = aiProviderModelHelperService;
+    }
+    async execute(args, ctx, parent) {
+        const model = this.aiProviderModelHelperService.getProviderModel(args.llm);
+        const options = {};
+        if (args.prompt) {
+            options.prompt = args.prompt;
+        }
+        else {
+            const messages = this.aiMessagesHelperService.getMessages(ctx.state.getMetadata('documents'), (0, lodash_1.pick)(args, ['prompt', 'messages', 'messagesSearchTag']));
+            options.messages = (0, ai_1.convertToModelMessages)(messages);
+        }
+        const document = parent.getDocument(args.response.document);
+        if (!document) {
+            throw new Error(`Document with name "${args.response.document}" not found in tool execution context.`);
+        }
+        const responseSchema = document.argsSchema;
+        if (!responseSchema) {
+            throw new Error(`AI object generation source document must have a schema.`);
+        }
+        options.schema = responseSchema;
+        const response = await this.handleGenerateObject(model, options);
+        return {
+            data: response.object,
+        };
+    }
+    async handleGenerateObject(model, options) {
+        const startTime = performance.now();
+        try {
+            return (0, ai_1.generateObject)({
+                model,
+                ...options,
+            });
+        }
+        catch (error) {
+            const errorResponseTime = performance.now() - startTime;
+            console.error(`Request failed after ${errorResponseTime}ms:`, error);
+            throw error;
+        }
+    }
+};
+exports.AiGenerateObject = AiGenerateObject;
+exports.AiGenerateObject = AiGenerateObject = __decorate([
+    (0, common_1.BlockConfig)({
+        config: {
+            description: 'Generates a structured object using a LLM',
+        },
+    }),
+    (0, common_1.WithArguments)(exports.AiGenerateObjectSchema),
+    __metadata("design:paramtypes", [services_1.AiMessagesHelperService,
+        services_2.AiProviderModelHelperService])
+], AiGenerateObject);
+//# sourceMappingURL=ai-generate-object.tool.js.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ai-generate-object.tool.js","sourceRoot":"","sources":["../../src/tools/ai-generate-object.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,6BAAwB;AACxB,8CAA2E;AAC3E,mCAA8B;AAC9B,2BAIY;AACZ,0CAAyD;AACzD,0CAAsD;AACtD,0CAA2D;AAG3D,0FAAmF;AAEtE,QAAA,sBAAsB,GAAG,uDAAwB,CAAC,MAAM,CAAC;IACpE,QAAQ,EAAE,OAAC,CAAC,MAAM,CAAC;QACjB,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACzB,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;KACrB,CAAC;CACH,CAAC,CAAC,MAAM,EAAE,CAAC;AAUL,IAAM,gBAAgB,GAAtB,MAAM,gBAAiB,SAAQ,eAAkC;IAEnD;IACA;IAFnB,YACmB,uBAAgD,EAChD,4BAA0D;QAE3E,KAAK,EAAE,CAAC;QAHS,4BAAuB,GAAvB,uBAAuB,CAAyB;QAChD,iCAA4B,GAA5B,4BAA4B,CAA8B;IAG7E,CAAC;IAED,KAAK,CAAC,OAAO,CACX,IAA8B,EAC9B,GAAsB,EACtB,MAAoB;QAEpB,MAAM,KAAK,GAAG,IAAI,CAAC,4BAA4B,CAAC,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE3E,MAAM,OAAO,GAAQ,EAAE,CAAC;QAExB,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,IAAA,aAAI,EAAC,IAAI,EAAE,CAAC,QAAQ,EAAE,UAAU,EAAE,mBAAmB,CAAC,CAAC,CACxD,CAAC;YACF,OAAO,CAAC,QAAQ,GAAG,IAAA,2BAAsB,EAAC,QAAQ,CAAC,CAAC;QACtD,CAAC;QAED,MAAM,QAAQ,GAAU,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACnE,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,IAAI,CAAC,QAAQ,CAAC,QAAQ,wCAAwC,CAAC,CAAC;QACzG,CAAC;QACD,MAAM,cAAc,GAAG,QAAQ,CAAC,UAAU,CAAC;QAC3C,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CACb,0DAA0D,CAC3D,CAAC;QACJ,CAAC;QAED,OAAO,CAAC,MAAM,GAAG,cAAc,CAAC;QAEhC,MAAM,QAAQ,GAA8B,MAAM,IAAI,CAAC,oBAAoB,CACzE,KAAK,EACL,OAAO,CACR,CAAC;QAEF,OAAO;YACL,IAAI,EAAE,QAAQ,CAAC,MAAM;SACtB,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAChC,KAAU,EACV,OAAY;QAEZ,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,OAAO,IAAA,mBAAc,EAAC;gBACpB,KAAK;gBACL,GAAG,OAAO;aACX,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;AAlEY,4CAAgB;2BAAhB,gBAAgB;IAN5B,IAAA,oBAAW,EAAC;QACX,MAAM,EAAE;YACN,WAAW,EAAE,2CAA2C;SACzD;KACF,CAAC;IACD,IAAA,sBAAa,EAAC,8BAAsB,CAAC;qCAGQ,kCAAuB;QAClB,uCAA4B;GAHlE,gBAAgB,CAkE5B"}

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

@@ -0,0 +1,77 @@
+import { z } from 'zod';
+import { ToolResult } from '@loopstack/common';
+import { ToolBase, WorkflowBase } from '@loopstack/core';
+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;
+        provider?: string;
+        envApiKey?: string;
+        cacheResponse?: boolean;
+    }, {
+        model?: string;
+        provider?: string;
+        envApiKey?: string;
+        cacheResponse?: boolean;
+    }>>;
+    messages: z.ZodOptional<z.ZodArray<z.ZodObject<{
+        role: z.ZodEnum<["system", "assistant", "user", "tool"]>;
+        content: z.ZodAny;
+    }, "strip", z.ZodTypeAny, {
+        role?: "system" | "assistant" | "user" | "tool";
+        content?: any;
+    }, {
+        role?: "system" | "assistant" | "user" | "tool";
+        content?: any;
+    }>, "many">>;
+    prompt: z.ZodOptional<z.ZodString>;
+    messagesSearchTag: z.ZodOptional<z.ZodString>;
+} & {
+    tools: z.ZodOptional<z.ZodArray<z.ZodString, "many">>;
+}, "strict", z.ZodTypeAny, {
+    llm?: {
+        model?: string;
+        provider?: string;
+        envApiKey?: string;
+        cacheResponse?: boolean;
+    };
+    messages?: {
+        role?: "system" | "assistant" | "user" | "tool";
+        content?: any;
+    }[];
+    prompt?: string;
+    messagesSearchTag?: string;
+    tools?: string[];
+}, {
+    llm?: {
+        model?: string;
+        provider?: string;
+        envApiKey?: string;
+        cacheResponse?: boolean;
+    };
+    messages?: {
+        role?: "system" | "assistant" | "user" | "tool";
+        content?: any;
+    }[];
+    prompt?: string;
+    messagesSearchTag?: string;
+    tools?: string[];
+}>;
+type AiGenerateTextArgsType = z.infer<typeof AiGenerateTextSchema>;
+export declare class AiGenerateText extends ToolBase<AiGenerateTextArgsType> {
+    private readonly aiMessagesHelperService;
+    private readonly aiToolsHelperService;
+    private readonly aiProviderModelHelperService;
+    constructor(aiMessagesHelperService: AiMessagesHelperService, aiToolsHelperService: AiToolsHelperService, aiProviderModelHelperService: AiProviderModelHelperService);
+    execute(args: AiGenerateTextArgsType, ctx: WorkflowExecution, parent: WorkflowBase): Promise<ToolResult>;
+    private handleGenerateText;
+}
+export {};

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

@@ -0,0 +1,112 @@
+"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 __metadata = (this && this.__metadata) || function (k, v) {
+    if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.AiGenerateText = exports.AiGenerateTextSchema = void 0;
+const zod_1 = require("zod");
+const common_1 = require("@loopstack/common");
+const lodash_1 = require("lodash");
+const core_1 = require("@loopstack/core");
+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();
+let AiGenerateText = class AiGenerateText extends core_1.ToolBase {
+    aiMessagesHelperService;
+    aiToolsHelperService;
+    aiProviderModelHelperService;
+    constructor(aiMessagesHelperService, aiToolsHelperService, aiProviderModelHelperService) {
+        super();
+        this.aiMessagesHelperService = aiMessagesHelperService;
+        this.aiToolsHelperService = aiToolsHelperService;
+        this.aiProviderModelHelperService = aiProviderModelHelperService;
+    }
+    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;
+        if (args.prompt) {
+            options.prompt = args.prompt;
+        }
+        else {
+            const messages = this.aiMessagesHelperService.getMessages(ctx.state.getMetadata('documents'), (0, lodash_1.pick)(args, ['prompt', 'messages', 'messagesSearchTag']));
+            options.messages = (0, ai_1.convertToModelMessages)(messages, {
+                tools: options.tools,
+            });
+        }
+        const uiMessage = await this.handleGenerateText(model, options);
+        return {
+            data: uiMessage,
+        };
+    }
+    async handleGenerateText(model, options) {
+        const startTime = performance.now();
+        try {
+            const result = (0, ai_1.streamText)({
+                model,
+                ...options,
+            });
+            return new Promise((resolve, reject) => {
+                const stream = (0, ai_1.createUIMessageStream)({
+                    execute({ writer }) {
+                        writer.merge(result.toUIMessageStream({
+                            sendReasoning: true,
+                        }));
+                    },
+                    onFinish: async (data) => {
+                        try {
+                            resolve(data.responseMessage);
+                        }
+                        catch (error) {
+                            reject(error);
+                        }
+                    },
+                });
+                (async () => {
+                    try {
+                        const reader = stream.getReader();
+                        while (true) {
+                            const { done } = await reader.read();
+                            if (done)
+                                break;
+                        }
+                    }
+                    catch (error) {
+                        reject(error);
+                    }
+                })();
+            });
+        }
+        catch (error) {
+            const errorResponseTime = performance.now() - startTime;
+            console.error(`Request failed after ${errorResponseTime}ms:`, error);
+            throw error;
+        }
+    }
+};
+exports.AiGenerateText = AiGenerateText;
+exports.AiGenerateText = AiGenerateText = __decorate([
+    (0, common_1.BlockConfig)({
+        config: {
+            description: 'Generates text using a LLM',
+        },
+    }),
+    (0, common_1.WithArguments)(exports.AiGenerateTextSchema),
+    __metadata("design:paramtypes", [services_1.AiMessagesHelperService,
+        services_3.AiToolsHelperService,
+        services_2.AiProviderModelHelperService])
+], AiGenerateText);
+//# sourceMappingURL=ai-generate-text.tool.js.map

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

@@ -0,0 +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,mCAA8B;AAC9B,0CAAyD;AACzD,0CAAsD;AACtD,0CAA2D;AAC3D,0CAAmD;AACnD,2BAKY;AAEZ,0FAAmF;AAEtE,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,GAAQ,EAAE,CAAC;QAExB,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,IAAA,aAAI,EAAC,IAAI,EAAE,CAAC,QAAQ,EAAE,UAAU,EAAE,mBAAmB,CAAC,CAAC,CACxD,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,KAAK,EAAE,IAAI,EAAE,EAAE;wBACvB,IAAI,CAAC;4BACH,OAAO,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;wBAChC,CAAC;wBAAC,OAAO,KAAK,EAAE,CAAC;4BACf,MAAM,CAAC,KAAK,CAAC,CAAC;wBAChB,CAAC;oBACH,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,CAAC,CAAC;oBAChB,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;AAzFY,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,CAyF1B"}

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

@@ -0,0 +1,16 @@
+import { ToolResult } from '@loopstack/common';
+import { z } from 'zod';
+import { ToolBase, WorkflowBase } from '@loopstack/core';
+import { WorkflowExecution } from '@loopstack/core/dist/workflow-processor/interfaces/workflow-execution.interface';
+declare const DelegateToolCallsToolSchema: z.ZodObject<{
+    message: z.ZodAny;
+}, "strip", z.ZodTypeAny, {
+    message?: any;
+}, {
+    message?: any;
+}>;
+type DelegateToolCallsToolArgs = z.infer<typeof DelegateToolCallsToolSchema>;
+export declare class DelegateToolCall extends ToolBase<DelegateToolCallsToolArgs> {
+    execute(args: DelegateToolCallsToolArgs, ctx: WorkflowExecution, parent: WorkflowBase): Promise<ToolResult>;
+}
+export {};

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

@@ -0,0 +1,57 @@
+"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;
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+exports.DelegateToolCall = void 0;
+const common_1 = require("@loopstack/common");
+const zod_1 = require("zod");
+const core_1 = require("@loopstack/core");
+const DelegateToolCallsToolSchema = zod_1.z.object({
+    message: zod_1.z.any(),
+});
+let DelegateToolCall = class DelegateToolCall extends core_1.ToolBase {
+    async execute(args, ctx, parent) {
+        const parts = args.message.parts;
+        const resultParts = [];
+        for (const part of parts) {
+            if (!part.type.startsWith('tool-')) {
+                continue;
+            }
+            const toolName = part.type.replace(/^tool-/, '');
+            const tool = parent.getTool(toolName);
+            const result = await tool.execute(part.input, ctx, parent);
+            resultParts.push({
+                type: part.type,
+                toolCallId: part.toolCallId,
+                output: {
+                    type: result.type || 'text',
+                    value: result.data,
+                },
+                input: part.input,
+                state: 'output-available',
+            });
+        }
+        const resultMessage = {
+            id: args.message.id,
+            role: 'assistant',
+            parts: resultParts,
+        };
+        return {
+            data: resultMessage,
+        };
+    }
+};
+exports.DelegateToolCall = DelegateToolCall;
+exports.DelegateToolCall = DelegateToolCall = __decorate([
+    (0, common_1.BlockConfig)({
+        config: {
+            description: 'Delegate a tool call.',
+        },
+    }),
+    (0, common_1.WithArguments)(DelegateToolCallsToolSchema)
+], DelegateToolCall);
+//# sourceMappingURL=delegate-tool-call.tool.js.map

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

@@ -0,0 +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,GAAG,EAAE;CACjB,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,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,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,IAAI;gBACf,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;AAzCY,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,CAyC5B"}

package/dist/tools/index.d.ts

@@ -0,0 +1,4 @@
+export * from './ai-generate-document.tool';
+export * from './ai-generate-text.tool';
+export * from './ai-generate-object.tool';
+export * from './delegate-tool-call.tool';

package/dist/tools/index.js

@@ -0,0 +1,21 @@
+"use strict";
+var __createBinding = (this && this.__createBinding) || (Object.create ? (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    var desc = Object.getOwnPropertyDescriptor(m, k);
+    if (!desc || ("get" in desc ? !m.__esModule : desc.writable || desc.configurable)) {
+      desc = { enumerable: true, get: function() { return m[k]; } };
+    }
+    Object.defineProperty(o, k2, desc);
+}) : (function(o, m, k, k2) {
+    if (k2 === undefined) k2 = k;
+    o[k2] = m[k];
+}));
+var __exportStar = (this && this.__exportStar) || function(m, exports) {
+    for (var p in m) if (p !== "default" && !Object.prototype.hasOwnProperty.call(exports, p)) __createBinding(exports, m, p);
+};
+Object.defineProperty(exports, "__esModule", { value: true });
+__exportStar(require("./ai-generate-document.tool"), exports);
+__exportStar(require("./ai-generate-text.tool"), exports);
+__exportStar(require("./ai-generate-object.tool"), exports);
+__exportStar(require("./delegate-tool-call.tool"), exports);
+//# sourceMappingURL=index.js.map

package/dist/tools/index.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.js","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;;AAAA,8DAA4C;AAC5C,0DAAwC;AACxC,4DAA0C;AAC1C,4DAA0C"}