@loopstack/ai-module 0.18.0 → 0.19.0-rc.0

package/README.md

@@ -19,35 +19,28 @@ This module is essential for workflows that need natural language processing, co
 
 ## Installation
 
-### Prerequisites
+You can add this module using the `loopstack` cli or via `npm`.
 
-Create a new Loopstack project if you haven't already:
+### a) Add Sources via `loopstack add`
 
 ```bash
-npx create-loopstack-app my-project
-cd my-project
+loopstack add @loopstack/ai-module
 ```
 
-Start Environment
-
-```bash
-cd my-project
-docker compose up -d
-```
+This command copies the source files into your `src` directory.
 
-### Install the Module
+- It is a great way to explore the code to learn new concepts or add own customizations
+- It will set up the module for you, so you do not need to manually update your application
 
-#### As Node Dependency via Npm:
+### b) Install via `npm install` (recommended)
 
 ```bash
 npm install --save @loopstack/ai-module
 ```
 
-#### OR: Copy Sources via Loopstack CLI
+Use npm install if you want to use and maintain the module as node dependency.
 
-```bash
-loopstack add @loopstack/ai-module
-```
+- Use this, if you do not need to make changes to the code or want to review the source code.
 
 ### Configure API Keys
 
@@ -63,22 +56,13 @@ ANTHROPIC_API_KEY=sk-ant-...
 
 ## Setup
 
-### 1. Import the Module
+### 1. Manual setup (optional)
 
-Add `AiModule` to your `default.module.ts` (included in the skeleton app) or to your own module:
+> This step is automatically done for you when using the `loopstack add` command.
 
-```typescript
-import { Module } from '@nestjs/common';
-import { AiModule } from '@loopstack/ai-module';
-import { LoopCoreModule } from '@loopstack/core';
-import { DefaultWorkspace } from './default.workspace';
-
-@Module({
-  imports: [LoopCoreModule, AiModule],
-  providers: [DefaultWorkspace],
-})
-export class DefaultModule {}
-```
+- Add `AiModule` to the imports of `default.module.ts` or any other custom module.
+
+See here for more information about working with [Modules](https://loopstack.ai/docs/building-with-loopstack/creating-a-module) and [Workspaces](https://loopstack.ai/docs/building-with-loopstack/creating-workspaces)
 
 ### 2. Use in Your Workflow
 
@@ -93,7 +77,7 @@ import {
   AiMessageDocument,
   DelegateToolCall,
 } from '@loopstack/ai-module';
-import { BlockConfig, Document, Helper, Tool } from '@loopstack/common';
+import { BlockConfig, DefineHelper, InjectDocument, InjectTool } from '@loopstack/common';
 import { WorkflowBase } from '@loopstack/core';
 
 @Injectable()
@@ -102,13 +86,13 @@ import { WorkflowBase } from '@loopstack/core';
 })
 export class MyWorkflow extends WorkflowBase {
   // Tools
-  @Tool() aiGenerateText: AiGenerateText;
-  @Tool() aiGenerateObject: AiGenerateObject;
-  @Tool() aiGenerateDocument: AiGenerateDocument;
-  @Tool() delegateToolCall: DelegateToolCall;
+  @InjectTool() aiGenerateText: AiGenerateText;
+  @InjectTool() aiGenerateObject: AiGenerateObject;
+  @InjectTool() aiGenerateDocument: AiGenerateDocument;
+  @InjectTool() delegateToolCall: DelegateToolCall;
 
   // Documents
-  @Document() aiMessageDocument: AiMessageDocument;
+  @InjectDocument() aiMessageDocument: AiMessageDocument;
 }
 ```
 

package/dist/ai.module.d.ts

@@ -1,2 +1,3 @@
 export declare class AiModule {
 }
+//# sourceMappingURL=ai.module.d.ts.map

package/dist/ai.module.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai.module.d.ts","sourceRoot":"","sources":["../src/ai.module.ts"],"names":[],"mappings":"AAYA,qBAwBa,QAAQ;CAAG"}

package/dist/ai.module.js

@@ -9,7 +9,6 @@ Object.defineProperty(exports, "__esModule", { value: true });
 exports.AiModule = void 0;
 const common_1 = require("@nestjs/common");
 const core_1 = require("@nestjs/core");
-const core_2 = require("@loopstack/core");
 const core_ui_module_1 = require("@loopstack/core-ui-module");
 const documents_1 = require("./documents");
 const anthropic_provider_1 = require("./providers/anthropic.provider");
@@ -24,7 +23,7 @@ let AiModule = class AiModule {
 exports.AiModule = AiModule;
 exports.AiModule = AiModule = __decorate([
     (0, common_1.Module)({
-        imports: [core_2.LoopCoreModule, core_ui_module_1.CoreUiModule, core_1.DiscoveryModule],
+        imports: [core_ui_module_1.CoreUiModule, core_1.DiscoveryModule],
         providers: [
             services_1.AiMessagesHelperService,
             services_2.AiProviderModelHelperService,

package/dist/ai.module.js.map

@@ -1 +1 @@
-{"version":3,"file":"ai.module.js","sourceRoot":"","sources":["../src/ai.module.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAAwC;AACxC,uCAA+C;AAC/C,0CAAiD;AACjD,8DAAyD;AACzD,2CAAgD;AAChD,uEAA0E;AAC1E,iEAAoE;AACpE,yCAAqD;AACrD,yCAA0D;AAC1D,yCAAkD;AAClD,0FAAoF;AACpF,mCAAiG;AA0B1F,IAAM,QAAQ,GAAd,MAAM,QAAQ;CAAG,CAAA;AAAX,4BAAQ;mBAAR,QAAQ;IAxBpB,IAAA,eAAM,EAAC;QACN,OAAO,EAAE,CAAC,qBAAc,EAAE,6BAAY,EAAE,sBAAe,CAAC;QACxD,SAAS,EAAE;YAET,kCAAuB;YACvB,uCAA4B;YAC5B,+BAAoB;YACpB,wDAAyB;YAGzB,uCAAqB;YACrB,6CAAwB;YAGxB,0BAAkB;YAClB,wBAAgB;YAChB,sBAAc;YACd,wBAAgB;YAGhB,6BAAiB;SAClB;QACD,OAAO,EAAE,CAAC,0BAAkB,EAAE,wBAAgB,EAAE,sBAAc,EAAE,wBAAgB,EAAE,6BAAiB,CAAC;KACrG,CAAC;GACW,QAAQ,CAAG"}
+{"version":3,"file":"ai.module.js","sourceRoot":"","sources":["../src/ai.module.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAAwC;AACxC,uCAA+C;AAC/C,8DAAyD;AACzD,2CAAgD;AAChD,uEAA0E;AAC1E,iEAAoE;AACpE,yCAAqD;AACrD,yCAA0D;AAC1D,yCAAkD;AAClD,0FAAoF;AACpF,mCAAiG;AA0B1F,IAAM,QAAQ,GAAd,MAAM,QAAQ;CAAG,CAAA;AAAX,4BAAQ;mBAAR,QAAQ;IAxBpB,IAAA,eAAM,EAAC;QACN,OAAO,EAAE,CAAC,6BAAY,EAAE,sBAAe,CAAC;QACxC,SAAS,EAAE;YAET,kCAAuB;YACvB,uCAA4B;YAC5B,+BAAoB;YACpB,wDAAyB;YAGzB,uCAAqB;YACrB,6CAAwB;YAGxB,0BAAkB;YAClB,wBAAgB;YAChB,sBAAc;YACd,wBAAgB;YAGhB,6BAAiB;SAClB;QACD,OAAO,EAAE,CAAC,0BAAkB,EAAE,wBAAgB,EAAE,sBAAc,EAAE,wBAAgB,EAAE,6BAAiB,CAAC;KACrG,CAAC;GACW,QAAQ,CAAG"}

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

@@ -1,3 +1,3 @@
-import { DocumentBase } from '@loopstack/core';
-export declare class AiMessageDocument extends DocumentBase {
+export declare class AiMessageDocument {
 }
+//# sourceMappingURL=ai-message-document.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"ai-message-document.d.ts","sourceRoot":"","sources":["../../src/documents/ai-message-document.ts"],"names":[],"mappings":"AAWA,qBASa,iBAAiB;CAAG"}

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

@@ -10,19 +10,18 @@ exports.AiMessageDocument = void 0;
 const common_1 = require("@nestjs/common");
 const zod_1 = require("zod");
 const common_2 = require("@loopstack/common");
-const core_1 = require("@loopstack/core");
 const AiMessageDocumentSchema = zod_1.z.object({
     id: zod_1.z.string().optional(),
     role: zod_1.z.union([zod_1.z.literal('system'), zod_1.z.literal('user'), zod_1.z.literal('assistant')]),
     metadata: zod_1.z.any().optional(),
     parts: zod_1.z.array(zod_1.z.any()),
 });
-let AiMessageDocument = class AiMessageDocument extends core_1.DocumentBase {
+let AiMessageDocument = class AiMessageDocument {
 };
 exports.AiMessageDocument = AiMessageDocument;
 exports.AiMessageDocument = AiMessageDocument = __decorate([
     (0, common_1.Injectable)(),
-    (0, common_2.BlockConfig)({
+    (0, common_2.Document)({
         config: {
             type: 'document',
             description: 'Ai Message Document.',

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

@@ -1 +1 @@
-{"version":3,"file":"ai-message-document.js","sourceRoot":"","sources":["../../src/documents/ai-message-document.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA4C;AAC5C,6BAAwB;AACxB,8CAA+D;AAC/D,0CAA+C;AAE/C,MAAM,uBAAuB,GAAG,OAAC,CAAC,MAAM,CAAC;IACvC,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACzB,IAAI,EAAE,OAAC,CAAC,KAAK,CAAC,CAAC,OAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,OAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,OAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;IAC/E,QAAQ,EAAE,OAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;IAC5B,KAAK,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC;CACxB,CAAC,CAAC;AAWI,IAAM,iBAAiB,GAAvB,MAAM,iBAAkB,SAAQ,mBAAY;CAAG,CAAA;AAAzC,8CAAiB;4BAAjB,iBAAiB;IAT7B,IAAA,mBAAU,GAAE;IACZ,IAAA,oBAAW,EAAC;QACX,MAAM,EAAE;YACN,IAAI,EAAE,UAAU;YAChB,WAAW,EAAE,sBAAsB;SACpC;QACD,UAAU,EAAE,SAAS,GAAG,2BAA2B;KACpD,CAAC;IACD,IAAA,sBAAa,EAAC,uBAAuB,CAAC;GAC1B,iBAAiB,CAAwB"}
+{"version":3,"file":"ai-message-document.js","sourceRoot":"","sources":["../../src/documents/ai-message-document.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA4C;AAC5C,6BAAwB;AACxB,8CAA4D;AAE5D,MAAM,uBAAuB,GAAG,OAAC,CAAC,MAAM,CAAC;IACvC,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;IACzB,IAAI,EAAE,OAAC,CAAC,KAAK,CAAC,CAAC,OAAC,CAAC,OAAO,CAAC,QAAQ,CAAC,EAAE,OAAC,CAAC,OAAO,CAAC,MAAM,CAAC,EAAE,OAAC,CAAC,OAAO,CAAC,WAAW,CAAC,CAAC,CAAC;IAC/E,QAAQ,EAAE,OAAC,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;IAC5B,KAAK,EAAE,OAAC,CAAC,KAAK,CAAC,OAAC,CAAC,GAAG,EAAE,CAAC;CACxB,CAAC,CAAC;AAWI,IAAM,iBAAiB,GAAvB,MAAM,iBAAiB;CAAG,CAAA;AAApB,8CAAiB;4BAAjB,iBAAiB;IAT7B,IAAA,mBAAU,GAAE;IACZ,IAAA,iBAAQ,EAAC;QACR,MAAM,EAAE;YACN,IAAI,EAAE,UAAU;YAChB,WAAW,EAAE,sBAAsB;SACpC;QACD,UAAU,EAAE,SAAS,GAAG,2BAA2B;KACpD,CAAC;IACD,IAAA,sBAAa,EAAC,uBAAuB,CAAC;GAC1B,iBAAiB,CAAG"}

package/dist/documents/index.d.ts

@@ -1 +1,2 @@
 export * from './ai-message-document';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/documents/index.ts"],"names":[],"mappings":"AAAA,cAAc,uBAAuB,CAAC"}

package/dist/index.d.ts

@@ -1,3 +1,4 @@
 export * from './ai.module';
 export * from './tools';
 export * from './documents';
+//# sourceMappingURL=index.d.ts.map

package/dist/index.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../src/index.ts"],"names":[],"mappings":"AAAA,cAAc,aAAa,CAAC;AAC5B,cAAc,SAAS,CAAC;AACxB,cAAc,aAAa,CAAC"}

package/dist/providers/anthropic.provider.d.ts

@@ -4,3 +4,4 @@ export declare class AnthropicProviderService implements AiProviderInterface {
     createClient(options: AiProviderOptions): import("@ai-sdk/anthropic").AnthropicProvider;
     getModel(client: ReturnType<typeof createAnthropic>, model: string): import("@ai-sdk/provider").LanguageModelV3;
 }
+//# sourceMappingURL=anthropic.provider.d.ts.map

package/dist/providers/anthropic.provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"anthropic.provider.d.ts","sourceRoot":"","sources":["../../src/providers/anthropic.provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACpD,OAAO,EAAc,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAEvF,qBAGa,wBAAyB,YAAW,mBAAmB;IAClE,YAAY,CAAC,OAAO,EAAE,iBAAiB;IAOvC,QAAQ,CAAC,MAAM,EAAE,UAAU,CAAC,OAAO,eAAe,CAAC,EAAE,KAAK,EAAE,MAAM;CAGnE"}

package/dist/providers/openai.provider.d.ts

@@ -4,3 +4,4 @@ export declare class OpenAiProviderService implements AiProviderInterface {
     createClient(options: AiProviderOptions): import("@ai-sdk/openai").OpenAIProvider;
     getModel(client: ReturnType<typeof createOpenAI>, model: string): import("@ai-sdk/provider").LanguageModelV3;
 }
+//# sourceMappingURL=openai.provider.d.ts.map

package/dist/providers/openai.provider.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"openai.provider.d.ts","sourceRoot":"","sources":["../../src/providers/openai.provider.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAC9C,OAAO,EAAc,mBAAmB,EAAE,iBAAiB,EAAE,MAAM,mBAAmB,CAAC;AAEvF,qBAGa,qBAAsB,YAAW,mBAAmB;IAC/D,YAAY,CAAC,OAAO,EAAE,iBAAiB;IAOvC,QAAQ,CAAC,MAAM,EAAE,UAAU,CAAC,OAAO,YAAY,CAAC,EAAE,KAAK,EAAE,MAAM;CAGhE"}

package/dist/schemas/ai-generate-tool-base.schema.d.ts

@@ -18,3 +18,4 @@ export declare const AiGenerateToolBaseSchema: z.ZodObject<{
     prompt: z.ZodOptional<z.ZodString>;
     messagesSearchTag: z.ZodOptional<z.ZodString>;
 }, z.core.$strip>;
+//# sourceMappingURL=ai-generate-tool-base.schema.d.ts.map

package/dist/schemas/ai-generate-tool-base.schema.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-generate-tool-base.schema.d.ts","sourceRoot":"","sources":["../../src/schemas/ai-generate-tool-base.schema.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,eAAO,MAAM,wBAAwB;;;;;;;;;;;;;;;;;;iBAmBnC,CAAC"}

package/dist/services/ai-messages-helper.service.d.ts

@@ -7,3 +7,4 @@ export declare class AiMessagesHelperService {
         messagesSearchTag?: string;
     }): Omit<UIMessage<unknown, UIDataTypes, UITools>, 'id'>[];
 }
+//# sourceMappingURL=ai-messages-helper.service.d.ts.map

package/dist/services/ai-messages-helper.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-messages-helper.service.d.ts","sourceRoot":"","sources":["../../src/services/ai-messages-helper.service.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,WAAW,EAAE,SAAS,EAAE,OAAO,EAAE,MAAM,IAAI,CAAC;AACrD,OAAO,EAAE,cAAc,EAAE,MAAM,mBAAmB,CAAC;AAEnD,qBACa,uBAAuB;IAClC,OAAO,CAAC,cAAc;IAUtB,WAAW,CACT,SAAS,EAAE,cAAc,EAAE,EAC3B,IAAI,EAAE;QAAE,QAAQ,CAAC,EAAE,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,WAAW,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,EAAE,CAAC;QAAC,iBAAiB,CAAC,EAAE,MAAM,CAAA;KAAE,GACtG,IAAI,CAAC,SAAS,CAAC,OAAO,EAAE,WAAW,EAAE,OAAO,CAAC,EAAE,IAAI,CAAC,EAAE;CAQ1D"}

package/dist/services/ai-provider-model-helper.service.d.ts

@@ -14,3 +14,4 @@ export declare class AiProviderModelHelperService {
     private getProvider;
     getProviderModel(config?: AiProviderModelConfig): LanguageModel;
 }
+//# sourceMappingURL=ai-provider-model-helper.service.d.ts.map

package/dist/services/ai-provider-model-helper.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-provider-model-helper.service.d.ts","sourceRoot":"","sources":["../../src/services/ai-provider-model-helper.service.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACnC,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAE3E,MAAM,WAAW,qBAAqB;IACpC,KAAK,CAAC,EAAE,MAAM,CAAC;IACf,QAAQ,CAAC,EAAE,MAAM,CAAC;IAClB,SAAS,CAAC,EAAE,MAAM,CAAC;IACnB,aAAa,CAAC,EAAE,OAAO,CAAC;CACzB;AAED,qBACa,4BAA4B;IAC3B,OAAO,CAAC,QAAQ,CAAC,kBAAkB;gBAAlB,kBAAkB,EAAE,yBAAyB;IAE1E,OAAO,CAAC,SAAS;IAWjB,OAAO,CAAC,QAAQ;IAYhB,OAAO,CAAC,WAAW;IAYnB,gBAAgB,CAAC,MAAM,CAAC,EAAE,qBAAqB,GAAG,aAAa;CAUhE"}

package/dist/services/ai-provider-registry.service.d.ts

@@ -14,3 +14,4 @@ export declare class AiProviderRegistryService implements OnModuleInit {
     createModel(providerName: string, options: AiProviderOptions): LanguageModel;
     getAvailableProviders(): string[];
 }
+//# sourceMappingURL=ai-provider-registry.service.d.ts.map

package/dist/services/ai-provider-registry.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-provider-registry.service.d.ts","sourceRoot":"","sources":["../../src/services/ai-provider-registry.service.ts"],"names":[],"mappings":"AAAA,OAAO,EAAsB,YAAY,EAAE,MAAM,gBAAgB,CAAC;AAClE,OAAO,EAAE,gBAAgB,EAAE,SAAS,EAAE,MAAM,cAAc,CAAC;AAC3D,OAAO,EAAE,aAAa,EAAE,MAAM,IAAI,CAAC;AACnC,OAAO,EAIL,iBAAiB,EAClB,MAAM,mBAAmB,CAAC;AAE3B,qBACa,yBAA0B,YAAW,YAAY;IAK1D,OAAO,CAAC,QAAQ,CAAC,gBAAgB;IACjC,OAAO,CAAC,QAAQ,CAAC,SAAS;IAL5B,OAAO,CAAC,QAAQ,CAAC,MAAM,CAA8C;IACrE,OAAO,CAAC,QAAQ,CAAC,SAAS,CAA0C;gBAGjD,gBAAgB,EAAE,gBAAgB,EAClC,SAAS,EAAE,SAAS;IAGvC,YAAY;IAIZ,OAAO,CAAC,aAAa;IAoBrB,OAAO,CAAC,gBAAgB;IAOxB,WAAW,CAAC,YAAY,EAAE,MAAM,EAAE,OAAO,EAAE,iBAAiB,GAAG,aAAa;IAkB5E,qBAAqB,IAAI,MAAM,EAAE;CAGlC"}

package/dist/services/ai-tools-helper.service.d.ts

@@ -1,4 +1,5 @@
-import { WorkflowBase } from '@loopstack/core';
+import { WorkflowInterface } from '@loopstack/common';
 export declare class AiToolsHelperService {
-    getTools(tools: string[], parent: WorkflowBase): Record<string, any> | undefined;
+    getTools(tools: string[], parent: WorkflowInterface): Record<string, any> | undefined;
 }
+//# sourceMappingURL=ai-tools-helper.service.d.ts.map

package/dist/services/ai-tools-helper.service.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"ai-tools-helper.service.d.ts","sourceRoot":"","sources":["../../src/services/ai-tools-helper.service.ts"],"names":[],"mappings":"AACA,OAAO,EAAiB,iBAAiB,EAAoD,MAAM,mBAAmB,CAAC;AAGvH,qBACa,oBAAoB;IAC/B,QAAQ,CAAC,KAAK,EAAE,MAAM,EAAE,EAAE,MAAM,EAAE,iBAAiB,GAAG,MAAM,CAAC,MAAM,EAAE,GAAG,CAAC,GAAG,SAAS;CA2BtF"}

package/dist/services/ai-tools-helper.service.js

@@ -8,18 +8,23 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AiToolsHelperService = void 0;
 const common_1 = require("@nestjs/common");
+const common_2 = require("@loopstack/common");
 let AiToolsHelperService = class AiToolsHelperService {
     getTools(tools, parent) {
         const toolDefinitions = {};
         for (const toolName of tools) {
-            const tool = parent.getTool(toolName);
+            const tool = (0, common_2.getBlockTool)(parent, toolName);
             if (!tool) {
                 throw new Error(`Tool with name ${toolName} not available in Workflow context.`);
             }
-            const inputSchema = tool.argsSchema;
+            const inputSchema = (0, common_2.getBlockArgsSchema)(tool);
+            const config = (0, common_2.getBlockConfig)(tool);
+            if (!config) {
+                throw new Error(`Block ${tool.constructor.name} is missing @BlockConfig decorator`);
+            }
             if (inputSchema) {
                 toolDefinitions[toolName] = {
-                    description: tool.config.description,
+                    description: config.description,
                     inputSchema,
                 };
             }

package/dist/services/ai-tools-helper.service.js.map

@@ -1 +1 @@
-{"version":3,"file":"ai-tools-helper.service.js","sourceRoot":"","sources":["../../src/services/ai-tools-helper.service.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA4C;AAIrC,IAAM,oBAAoB,GAA1B,MAAM,oBAAoB;IAC/B,QAAQ,CAAC,KAAe,EAAE,MAAoB;QAE5C,MAAM,eAAe,GAAwB,EAAE,CAAC;QAEhD,KAAK,MAAM,QAAQ,IAAI,KAAK,EAAE,CAAC;YAC7B,MAAM,IAAI,GAAG,MAAM,CAAC,OAAO,CAAC,QAAQ,CAAC,CAAC;YACtC,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,IAAI,KAAK,CAAC,kBAAkB,QAAQ,qCAAqC,CAAC,CAAC;YACnF,CAAC;YAED,MAAM,WAAW,GAAG,IAAI,CAAC,UAAU,CAAC;YAEpC,IAAI,WAAW,EAAE,CAAC;gBAChB,eAAe,CAAC,QAAQ,CAAC,GAAG;oBAC1B,WAAW,EAAE,IAAI,CAAC,MAAM,CAAC,WAAW;oBACpC,WAAW;iBACD,CAAC;YACf,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,SAAS,CAAC;IAC3E,CAAC;CACF,CAAA;AAvBY,oDAAoB;+BAApB,oBAAoB;IADhC,IAAA,mBAAU,GAAE;GACA,oBAAoB,CAuBhC"}
+{"version":3,"file":"ai-tools-helper.service.js","sourceRoot":"","sources":["../../src/services/ai-tools-helper.service.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAA4C;AAC5C,8CAAuH;AAIhH,IAAM,oBAAoB,GAA1B,MAAM,oBAAoB;IAC/B,QAAQ,CAAC,KAAe,EAAE,MAAyB;QAEjD,MAAM,eAAe,GAAwB,EAAE,CAAC;QAEhD,KAAK,MAAM,QAAQ,IAAI,KAAK,EAAE,CAAC;YAC7B,MAAM,IAAI,GAAG,IAAA,qBAAY,EAAgB,MAAM,EAAE,QAAQ,CAAC,CAAC;YAC3D,IAAI,CAAC,IAAI,EAAE,CAAC;gBACV,MAAM,IAAI,KAAK,CAAC,kBAAkB,QAAQ,qCAAqC,CAAC,CAAC;YACnF,CAAC;YAED,MAAM,WAAW,GAAG,IAAA,2BAAkB,EAAC,IAAI,CAAC,CAAC;YAE7C,MAAM,MAAM,GAAG,IAAA,uBAAc,EAAe,IAAI,CAAC,CAAC;YAClD,IAAI,CAAC,MAAM,EAAE,CAAC;gBACZ,MAAM,IAAI,KAAK,CAAC,SAAS,IAAI,CAAC,WAAW,CAAC,IAAI,oCAAoC,CAAC,CAAC;YACtF,CAAC;YAED,IAAI,WAAW,EAAE,CAAC;gBAChB,eAAe,CAAC,QAAQ,CAAC,GAAG;oBAC1B,WAAW,EAAE,MAAM,CAAC,WAAW;oBAC/B,WAAW;iBACD,CAAC;YACf,CAAC;QACH,CAAC;QAED,OAAO,MAAM,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,eAAe,CAAC,CAAC,CAAC,SAAS,CAAC;IAC3E,CAAC;CACF,CAAA;AA5BY,oDAAoB;+BAApB,oBAAoB;IADhC,IAAA,mBAAU,GAAE;GACA,oBAAoB,CA4BhC"}

package/dist/services/index.d.ts

@@ -2,3 +2,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';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/services/index.ts"],"names":[],"mappings":"AAAA,cAAc,8BAA8B,CAAC;AAC7C,cAAc,oCAAoC,CAAC;AACnD,cAAc,8BAA8B,CAAC;AAC7C,cAAc,2BAA2B,CAAC"}

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

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

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

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

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

@@ -12,14 +12,13 @@ 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 core_ui_module_1 = require("@loopstack/core-ui-module");
 const ai_generate_object_tool_1 = require("./ai-generate-object.tool");
-let AiGenerateDocument = class AiGenerateDocument extends core_1.ToolBase {
+let AiGenerateDocument = class AiGenerateDocument {
     aiGenerateObject;
     createDocument;
     getRequiredTool(name) {
-        const tool = this.getTool(name);
+        const tool = (0, common_2.getBlockTool)(this, name);
         if (tool === undefined) {
             throw new Error(`Tool "${name}" is not available`);
         }
@@ -38,16 +37,16 @@ let AiGenerateDocument = class AiGenerateDocument extends core_1.ToolBase {
 };
 exports.AiGenerateDocument = AiGenerateDocument;
 __decorate([
-    (0, common_2.Tool)(),
+    (0, common_2.InjectTool)(),
     __metadata("design:type", ai_generate_object_tool_1.AiGenerateObject)
 ], AiGenerateDocument.prototype, "aiGenerateObject", void 0);
 __decorate([
-    (0, common_2.Tool)(),
+    (0, common_2.InjectTool)(),
     __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)({
+    (0, common_2.Tool)({
         config: {
             description: 'Generates a structured object using a LLM and creates it as document',
         },

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

@@ -1 +1 @@
-{"version":3,"file":"ai-generate-document.tool.js","sourceRoot":"","sources":["../../src/tools/ai-generate-document.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA4C;AAC5C,8CAAiF;AACjF,0CAAyD;AACzD,8DAA2D;AAE3D,uEAA+G;AASxG,IAAM,kBAAkB,GAAxB,MAAM,kBAAmB,SAAQ,eAAkC;IACxD,gBAAgB,CAAoB;IACpC,cAAc,CAAkB;IAExC,eAAe,CAAC,IAAY;QAClC,MAAM,IAAI,GAAG,IAAI,CAAC,OAAO,CAAC,IAAI,CAAC,CAAC;QAChC,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,SAAS,IAAI,oBAAoB,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,OAAO,CAAC,IAA8B,EAAE,GAAsB,EAAE,MAAoB;QACxF,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACzF,OAAO,IAAI,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC,OAAO,CACnD;YACE,EAAE,EAAE,IAAI,CAAC,QAAQ,CAAC,EAAE;YACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ;YAChC,MAAM,EAAE;gBACN,OAAO,EAAE,MAAM,CAAC,IAAe;aAChC;SACF,EACD,GAAG,EACH,MAAM,CACP,CAAC;IACJ,CAAC;CACF,CAAA;AA1BY,gDAAkB;AACb;IAAf,IAAA,aAAI,GAAE;8BAA4B,0CAAgB;4DAAC;AACpC;IAAf,IAAA,aAAI,GAAE;8BAA0B,+BAAc;0DAAC;6BAFrC,kBAAkB;IAP9B,IAAA,mBAAU,GAAE;IACZ,IAAA,oBAAW,EAAC;QACX,MAAM,EAAE;YACN,WAAW,EAAE,sEAAsE;SACpF;KACF,CAAC;IACD,IAAA,sBAAa,EAAC,gDAAsB,CAAC;GACzB,kBAAkB,CA0B9B"}
+{"version":3,"file":"ai-generate-document.tool.js","sourceRoot":"","sources":["../../src/tools/ai-generate-document.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,2CAA4C;AAC5C,8CAS2B;AAC3B,8DAA2D;AAC3D,uEAA+G;AASxG,IAAM,kBAAkB,GAAxB,MAAM,kBAAkB;IACP,gBAAgB,CAAoB;IACpC,cAAc,CAAkB;IAE9C,eAAe,CAAC,IAAY;QAClC,MAAM,IAAI,GAAG,IAAA,qBAAY,EAAgB,IAAI,EAAE,IAAI,CAAC,CAAC;QACrD,IAAI,IAAI,KAAK,SAAS,EAAE,CAAC;YACvB,MAAM,IAAI,KAAK,CAAC,SAAS,IAAI,oBAAoB,CAAC,CAAC;QACrD,CAAC;QACD,OAAO,IAAI,CAAC;IACd,CAAC;IAED,KAAK,CAAC,OAAO,CACX,IAA8B,EAC9B,GAA2B,EAC3B,MAAyB;QAEzB,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,kBAAkB,CAAC,CAAC,OAAO,CAAC,IAAI,EAAE,GAAG,EAAE,MAAM,CAAC,CAAC;QACzF,OAAO,IAAI,CAAC,eAAe,CAAC,gBAAgB,CAAC,CAAC,OAAO,CACnD;YACE,EAAE,EAAE,IAAI,CAAC,QAAQ,CAAC,EAAE;YACpB,QAAQ,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ;YAChC,MAAM,EAAE;gBACN,OAAO,EAAE,MAAM,CAAC,IAAe;aAChC;SACF,EACD,GAAG,EACH,MAAM,CACP,CAAC;IACJ,CAAC;CACF,CAAA;AA9BY,gDAAkB;AACP;IAArB,IAAA,mBAAU,GAAE;8BAA4B,0CAAgB;4DAAC;AACpC;IAArB,IAAA,mBAAU,GAAE;8BAA0B,+BAAc;0DAAC;6BAF3C,kBAAkB;IAP9B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,MAAM,EAAE;YACN,WAAW,EAAE,sEAAsE;SACpF;KACF,CAAC;IACD,IAAA,sBAAa,EAAC,gDAAsB,CAAC;GACzB,kBAAkB,CA8B9B"}

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

@@ -1,7 +1,5 @@
 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 { ToolInterface, ToolResult, WorkflowExecution, WorkflowInterface } from '@loopstack/common';
 import { AiMessagesHelperService } from '../services';
 import { AiProviderModelHelperService } from '../services';
 export declare const AiGenerateObjectSchema: z.ZodObject<{
@@ -28,10 +26,11 @@ export declare const AiGenerateObjectSchema: z.ZodObject<{
     }, z.core.$strip>;
 }, z.core.$strict>;
 export type AiGenerateObjectArgsType = z.infer<typeof AiGenerateObjectSchema>;
-export declare class AiGenerateObject extends ToolBase<AiGenerateObjectArgsType> {
+export declare class AiGenerateObject implements ToolInterface<AiGenerateObjectArgsType> {
     private readonly aiMessagesHelperService;
     private readonly aiProviderModelHelperService;
     constructor(aiMessagesHelperService: AiMessagesHelperService, aiProviderModelHelperService: AiProviderModelHelperService);
-    execute(args: AiGenerateObjectArgsType, ctx: WorkflowExecution, parent: WorkflowBase): Promise<ToolResult>;
+    execute(args: AiGenerateObjectArgsType, ctx: WorkflowExecution<any>, parent: WorkflowInterface): Promise<ToolResult>;
     private handleGenerateObject;
 }
+//# sourceMappingURL=ai-generate-object.tool.d.ts.map

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

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

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

@@ -10,10 +10,10 @@ var __metadata = (this && this.__metadata) || function (k, v) {
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AiGenerateObject = exports.AiGenerateObjectSchema = 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 core_1 = require("@loopstack/core");
+const common_2 = require("@loopstack/common");
 const ai_generate_tool_base_schema_1 = require("../schemas/ai-generate-tool-base.schema");
 const services_1 = require("../services");
 const services_2 = require("../services");
@@ -23,11 +23,10 @@ exports.AiGenerateObjectSchema = ai_generate_tool_base_schema_1.AiGenerateToolBa
         document: zod_1.z.string(),
     }),
 }).strict();
-let AiGenerateObject = class AiGenerateObject extends core_1.ToolBase {
+let AiGenerateObject = class AiGenerateObject {
     aiMessagesHelperService;
     aiProviderModelHelperService;
     constructor(aiMessagesHelperService, aiProviderModelHelperService) {
-        super();
         this.aiMessagesHelperService = aiMessagesHelperService;
         this.aiProviderModelHelperService = aiProviderModelHelperService;
     }
@@ -49,11 +48,11 @@ let AiGenerateObject = class AiGenerateObject extends core_1.ToolBase {
             });
             options.messages = await (0, ai_1.convertToModelMessages)(messages);
         }
-        const document = parent.getDocument(args.response.document);
+        const document = (0, common_2.getBlockDocument)(parent, args.response.document);
         if (!document) {
             throw new Error(`Document with name "${args.response.document}" not found in tool execution context.`);
         }
-        const responseSchema = document.argsSchema;
+        const responseSchema = (0, common_2.getBlockArgsSchema)(document);
         if (!responseSchema) {
             throw new Error(`AI object generation source document must have a schema.`);
         }
@@ -83,12 +82,13 @@ let AiGenerateObject = class AiGenerateObject extends core_1.ToolBase {
 };
 exports.AiGenerateObject = AiGenerateObject;
 exports.AiGenerateObject = AiGenerateObject = __decorate([
-    (0, common_1.BlockConfig)({
+    (0, common_1.Injectable)(),
+    (0, common_2.Tool)({
         config: {
             description: 'Generates a structured object using a LLM',
         },
     }),
-    (0, common_1.WithArguments)(exports.AiGenerateObjectSchema),
+    (0, common_2.WithArguments)(exports.AiGenerateObjectSchema),
     __metadata("design:paramtypes", [services_1.AiMessagesHelperService,
         services_2.AiProviderModelHelperService])
 ], AiGenerateObject);

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

@@ -1 +1 @@
-{"version":3,"file":"ai-generate-object.tool.js","sourceRoot":"","sources":["../../src/tools/ai-generate-object.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,2BAQY;AACZ,6BAAwB;AACxB,8CAA2E;AAC3E,0CAAyD;AAGzD,0FAAmF;AACnF,0CAAsD;AACtD,0CAA2D;AAE9C,QAAA,sBAAsB,GAAG,uDAAwB,CAAC,MAAM,CAAC;IACpE,QAAQ,EAAE,OAAC,CAAC,MAAM,CAAC;QACjB,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACzB,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;KACrB,CAAC;CACH,CAAC,CAAC,MAAM,EAAE,CAAC;AAUL,IAAM,gBAAgB,GAAtB,MAAM,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,CAAC,IAA8B,EAAE,GAAsB,EAAE,MAAoB;QACxF,MAAM,KAAK,GAAG,IAAI,CAAC,4BAA4B,CAAC,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE3E,MAAM,OAAO,GAGT;YACF,QAAQ,EAAE,EAAE;SACb,CAAC;QAEF,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACpB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,IAAI,CAAC,MAAM;aACL,CAAC,CAAC;QACrB,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;gBAC5F,QAAQ,EAAE,IAAI,CAAC,QAAkC;gBACjD,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;aAC1C,CAAC,CAAC;YAEH,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAA,2BAAsB,EAAC,QAAQ,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,QAAQ,GAAsB,MAAM,CAAC,WAAW,CAAC,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QAC/E,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,CAAC,0DAA0D,CAAC,CAAC;QAC9E,CAAC;QAED,OAAO,CAAC,MAAM,GAAG,cAAc,CAAC;QAEhC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAEjE,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAEtB,OAAO;YACL,IAAI,EAAE,QAAQ,CAAC,MAAM;SACtB,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAChC,KAAoB,EACpB,OAGC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAE5B,OAAO,IAAA,iBAAY,EAAC;gBAClB,KAAK;gBACL,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,WAAM,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACjF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACxD,OAAO,CAAC,KAAK,CAAC,wBAAwB,iBAAiB,KAAK,EAAE,KAAK,CAAC,CAAC;YACrE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAA;AA1EY,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,CA0E5B"}
+{"version":3,"file":"ai-generate-object.tool.js","sourceRoot":"","sources":["../../src/tools/ai-generate-object.tool.ts"],"names":[],"mappings":";;;;;;;;;;;;AACA,2CAA4C;AAC5C,2BAQY;AACZ,6BAAwB;AACxB,8CAU2B;AAC3B,0FAAmF;AACnF,0CAAsD;AACtD,0CAA2D;AAE9C,QAAA,sBAAsB,GAAG,uDAAwB,CAAC,MAAM,CAAC;IACpE,QAAQ,EAAE,OAAC,CAAC,MAAM,CAAC;QACjB,EAAE,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,EAAE;QACzB,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE;KACrB,CAAC;CACH,CAAC,CAAC,MAAM,EAAE,CAAC;AAWL,IAAM,gBAAgB,GAAtB,MAAM,gBAAgB;IAER;IACA;IAFnB,YACmB,uBAAgD,EAChD,4BAA0D;QAD1D,4BAAuB,GAAvB,uBAAuB,CAAyB;QAChD,iCAA4B,GAA5B,4BAA4B,CAA8B;IAC1E,CAAC;IAEJ,KAAK,CAAC,OAAO,CACX,IAA8B,EAC9B,GAA2B,EAC3B,MAAyB;QAEzB,MAAM,KAAK,GAAG,IAAI,CAAC,4BAA4B,CAAC,gBAAgB,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;QAE3E,MAAM,OAAO,GAGT;YACF,QAAQ,EAAE,EAAE;SACb,CAAC;QAEF,IAAI,IAAI,CAAC,MAAM,EAAE,CAAC;YAChB,OAAO,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACpB,IAAI,EAAE,MAAM;gBACZ,OAAO,EAAE,IAAI,CAAC,MAAM;aACL,CAAC,CAAC;QACrB,CAAC;aAAM,CAAC;YACN,MAAM,QAAQ,GAAG,IAAI,CAAC,uBAAuB,CAAC,WAAW,CAAC,GAAG,CAAC,KAAK,CAAC,WAAW,CAAC,WAAW,CAAC,EAAE;gBAC5F,QAAQ,EAAE,IAAI,CAAC,QAAkC;gBACjD,iBAAiB,EAAE,IAAI,CAAC,iBAAiB;aAC1C,CAAC,CAAC;YAEH,OAAO,CAAC,QAAQ,GAAG,MAAM,IAAA,2BAAsB,EAAC,QAAQ,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,QAAQ,GAAG,IAAA,yBAAgB,EAAoB,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;QACrF,IAAI,CAAC,QAAQ,EAAE,CAAC;YACd,MAAM,IAAI,KAAK,CAAC,uBAAuB,IAAI,CAAC,QAAQ,CAAC,QAAQ,wCAAwC,CAAC,CAAC;QACzG,CAAC;QACD,MAAM,cAAc,GAAG,IAAA,2BAAkB,EAAC,QAAQ,CAAC,CAAC;QACpD,IAAI,CAAC,cAAc,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,0DAA0D,CAAC,CAAC;QAC9E,CAAC;QAED,OAAO,CAAC,MAAM,GAAG,cAAc,CAAC;QAEhC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,KAAK,EAAE,OAAO,CAAC,CAAC;QAEjE,OAAO,CAAC,GAAG,CAAC,QAAQ,CAAC,CAAC;QAEtB,OAAO;YACL,IAAI,EAAE,QAAQ,CAAC,MAAM;SACtB,CAAC;IACJ,CAAC;IAEO,KAAK,CAAC,oBAAoB,CAChC,KAAoB,EACpB,OAGC;QAED,MAAM,SAAS,GAAG,WAAW,CAAC,GAAG,EAAE,CAAC;QACpC,IAAI,CAAC;YACH,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC;YAE5B,OAAO,IAAA,iBAAY,EAAC;gBAClB,KAAK;gBACL,QAAQ,EAAE,OAAO,CAAC,QAAQ;gBAC1B,GAAG,CAAC,OAAO,CAAC,MAAM,CAAC,CAAC,CAAC,EAAE,MAAM,EAAE,WAAM,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,OAAO,CAAC,MAAM,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,CAAC;aACjF,CAAC,CAAC;QACL,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,MAAM,iBAAiB,GAAG,WAAW,CAAC,GAAG,EAAE,GAAG,SAAS,CAAC;YACxD,OAAO,CAAC,KAAK,CAAC,wBAAwB,iBAAiB,KAAK,EAAE,KAAK,CAAC,CAAC;YACrE,MAAM,KAAK,CAAC;QACd,CAAC;IACH,CAAC;CACF,CAAA;AA5EY,4CAAgB;2BAAhB,gBAAgB;IAP5B,IAAA,mBAAU,GAAE;IACZ,IAAA,aAAI,EAAC;QACJ,MAAM,EAAE;YACN,WAAW,EAAE,2CAA2C;SACzD;KACF,CAAC;IACD,IAAA,sBAAa,EAAC,8BAAsB,CAAC;qCAGQ,kCAAuB;QAClB,uCAA4B;GAHlE,gBAAgB,CA4E5B"}

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

@@ -1,7 +1,5 @@
 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 { ToolInterface, ToolResult, WorkflowExecution, WorkflowInterface } from '@loopstack/common';
 import { AiMessagesHelperService } from '../services';
 import { AiProviderModelHelperService } from '../services';
 import { AiToolsHelperService } from '../services';
@@ -26,12 +24,13 @@ export declare const AiGenerateTextSchema: z.ZodObject<{
     tools: z.ZodOptional<z.ZodArray<z.ZodString>>;
 }, z.core.$strict>;
 type AiGenerateTextArgsType = z.infer<typeof AiGenerateTextSchema>;
-export declare class AiGenerateText extends ToolBase<AiGenerateTextArgsType> {
+export declare class AiGenerateText implements ToolInterface<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>;
+    execute(args: AiGenerateTextArgsType, ctx: WorkflowExecution<any>, parent: WorkflowInterface): Promise<ToolResult>;
     private handleGenerateText;
 }
 export {};
+//# sourceMappingURL=ai-generate-text.tool.d.ts.map

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

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

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

@@ -14,7 +14,6 @@ const common_1 = require("@nestjs/common");
 const ai_1 = require("ai");
 const zod_1 = require("zod");
 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");
@@ -22,12 +21,11 @@ const services_3 = require("../services");
 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 {
+let AiGenerateText = class AiGenerateText {
     aiMessagesHelperService;
     aiToolsHelperService;
     aiProviderModelHelperService;
     constructor(aiMessagesHelperService, aiToolsHelperService, aiProviderModelHelperService) {
-        super();
         this.aiMessagesHelperService = aiMessagesHelperService;
         this.aiToolsHelperService = aiToolsHelperService;
         this.aiProviderModelHelperService = aiProviderModelHelperService;
@@ -101,7 +99,7 @@ let AiGenerateText = class AiGenerateText extends core_1.ToolBase {
 exports.AiGenerateText = AiGenerateText;
 exports.AiGenerateText = AiGenerateText = __decorate([
     (0, common_1.Injectable)(),
-    (0, common_2.BlockConfig)({
+    (0, common_2.Tool)({
         config: {
             description: 'Generates text using a LLM',
         },

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

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

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

@@ -1,7 +1,5 @@
 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 { ToolInterface, ToolResult, WorkflowExecution, WorkflowInterface } from '@loopstack/common';
 declare const DelegateToolCallsToolSchema: z.ZodObject<{
     message: z.ZodObject<{
         id: z.ZodString;
@@ -13,7 +11,8 @@ declare const DelegateToolCallsToolSchema: z.ZodObject<{
     }, 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>;
+export declare class DelegateToolCall implements ToolInterface<DelegateToolCallsToolArgs> {
+    execute(args: DelegateToolCallsToolArgs, ctx: WorkflowExecution<any>, parent: WorkflowInterface): Promise<ToolResult>;
 }
 export {};
+//# sourceMappingURL=delegate-tool-call.tool.d.ts.map

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

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

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

@@ -7,9 +7,9 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
 };
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.DelegateToolCall = void 0;
+const common_1 = require("@nestjs/common");
 const zod_1 = require("zod");
-const common_1 = require("@loopstack/common");
-const core_1 = require("@loopstack/core");
+const common_2 = require("@loopstack/common");
 const DelegateToolCallsToolSchema = zod_1.z.object({
     message: zod_1.z.object({
         id: zod_1.z.string(),
@@ -20,7 +20,7 @@ const DelegateToolCallsToolSchema = zod_1.z.object({
         })),
     }),
 });
-let DelegateToolCall = class DelegateToolCall extends core_1.ToolBase {
+let DelegateToolCall = class DelegateToolCall {
     async execute(args, ctx, parent) {
         const parts = args.message.parts;
         const resultParts = [];
@@ -32,7 +32,7 @@ let DelegateToolCall = class DelegateToolCall extends core_1.ToolBase {
                 throw new Error(`No toolCallId provided`);
             }
             const toolName = part.type.replace(/^tool-/, '');
-            const tool = parent.getTool(toolName);
+            const tool = (0, common_2.getBlockTool)(parent, toolName);
             if (!tool) {
                 throw new Error(`Tool ${toolName} not found.`);
             }
@@ -60,11 +60,12 @@ let DelegateToolCall = class DelegateToolCall extends core_1.ToolBase {
 };
 exports.DelegateToolCall = DelegateToolCall;
 exports.DelegateToolCall = DelegateToolCall = __decorate([
-    (0, common_1.BlockConfig)({
+    (0, common_1.Injectable)(),
+    (0, common_2.Tool)({
         config: {
             description: 'Delegate a tool call.',
         },
     }),
-    (0, common_1.WithArguments)(DelegateToolCallsToolSchema)
+    (0, common_2.WithArguments)(DelegateToolCallsToolSchema)
 ], DelegateToolCall);
 //# sourceMappingURL=delegate-tool-call.tool.js.map

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

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

package/dist/tools/index.d.ts

@@ -2,3 +2,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';
+//# sourceMappingURL=index.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"index.d.ts","sourceRoot":"","sources":["../../src/tools/index.ts"],"names":[],"mappings":"AAAA,cAAc,6BAA6B,CAAC;AAC5C,cAAc,yBAAyB,CAAC;AACxC,cAAc,2BAA2B,CAAC;AAC1C,cAAc,2BAA2B,CAAC"}

package/package.json

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

package/src/ai.module.ts

@@ -1,6 +1,5 @@
 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';
@@ -12,7 +11,7 @@ import { AiProviderRegistryService } from './services/ai-provider-registry.servi
 import { AiGenerateDocument, AiGenerateObject, AiGenerateText, DelegateToolCall } from './tools';
 
 @Module({
-  imports: [LoopCoreModule, CoreUiModule, DiscoveryModule],
+  imports: [CoreUiModule, DiscoveryModule],
   providers: [
     // services
     AiMessagesHelperService,

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

@@ -1,7 +1,6 @@
 import { Injectable } from '@nestjs/common';
 import { z } from 'zod';
-import { BlockConfig, WithArguments } from '@loopstack/common';
-import { DocumentBase } from '@loopstack/core';
+import { Document, WithArguments } from '@loopstack/common';
 
 const AiMessageDocumentSchema = z.object({
   id: z.string().optional(),
@@ -11,7 +10,7 @@ const AiMessageDocumentSchema = z.object({
 });
 
 @Injectable()
-@BlockConfig({
+@Document({
   config: {
     type: 'document',
     description: 'Ai Message Document.',
@@ -19,4 +18,4 @@ const AiMessageDocumentSchema = z.object({
   configFile: __dirname + '/ai-message-document.yaml',
 })
 @WithArguments(AiMessageDocumentSchema)
-export class AiMessageDocument extends DocumentBase {}
+export class AiMessageDocument {}

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

@@ -1,23 +1,29 @@
 import { Injectable } from '@nestjs/common';
-import { WorkflowBase } from '@loopstack/core';
+import { ToolInterface, WorkflowInterface, getBlockArgsSchema, getBlockConfig, getBlockTool } from '@loopstack/common';
+import { WorkflowType } from '@loopstack/contracts/dist/types';
 
 @Injectable()
 export class AiToolsHelperService {
-  getTools(tools: string[], parent: WorkflowBase): Record<string, any> | undefined {
+  getTools(tools: string[], parent: WorkflowInterface): Record<string, any> | undefined {
     // using any instead of ToolSet bc ai sdk types have some nesting issue
     const toolDefinitions: Record<string, any> = {};
 
     for (const toolName of tools) {
-      const tool = parent.getTool(toolName);
+      const tool = getBlockTool<ToolInterface>(parent, toolName);
       if (!tool) {
         throw new Error(`Tool with name ${toolName} not available in Workflow context.`);
       }
 
-      const inputSchema = tool.argsSchema;
+      const inputSchema = getBlockArgsSchema(tool);
+
+      const config = getBlockConfig<WorkflowType>(tool);
+      if (!config) {
+        throw new Error(`Block ${tool.constructor.name} is missing @BlockConfig decorator`);
+      }
 
       if (inputSchema) {
         toolDefinitions[toolName] = {
-          description: tool.config.description,
+          description: config.description,
           inputSchema,
         } as unknown; // using unknown bc ai sdk types have some nesting issue
       }

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

@@ -1,30 +1,41 @@
 import { Injectable } from '@nestjs/common';
-import { BlockConfig, Tool, ToolResult, WithArguments } from '@loopstack/common';
-import { ToolBase, WorkflowBase } from '@loopstack/core';
+import {
+  InjectTool,
+  Tool,
+  ToolInterface,
+  ToolResult,
+  WithArguments,
+  WorkflowExecution,
+  WorkflowInterface,
+  getBlockTool,
+} from '@loopstack/common';
 import { CreateDocument } from '@loopstack/core-ui-module';
-import { WorkflowExecution } from '@loopstack/core/dist/workflow-processor/interfaces/workflow-execution.interface';
 import { AiGenerateObject, AiGenerateObjectArgsType, AiGenerateObjectSchema } from './ai-generate-object.tool';
 
 @Injectable()
-@BlockConfig({
+@Tool({
   config: {
     description: 'Generates a structured object using a LLM and creates it as document',
   },
 })
 @WithArguments(AiGenerateObjectSchema)
-export class AiGenerateDocument extends ToolBase<AiGenerateObjectArgsType> {
-  @Tool() private aiGenerateObject!: AiGenerateObject;
-  @Tool() private createDocument!: CreateDocument;
+export class AiGenerateDocument implements ToolInterface<AiGenerateObjectArgsType> {
+  @InjectTool() private aiGenerateObject!: AiGenerateObject;
+  @InjectTool() private createDocument!: CreateDocument;
 
-  private getRequiredTool(name: string): ToolBase<any> {
-    const tool = this.getTool(name);
+  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: AiGenerateObjectArgsType, ctx: WorkflowExecution, parent: WorkflowBase): Promise<ToolResult> {
+  async execute(
+    args: AiGenerateObjectArgsType,
+    ctx: WorkflowExecution<any>,
+    parent: WorkflowInterface,
+  ): Promise<ToolResult> {
     const result = await this.getRequiredTool('aiGenerateObject').execute(args, ctx, parent);
     return this.getRequiredTool('createDocument').execute(
       {

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

@@ -1,4 +1,5 @@
 import { ModelMessage } from '@ai-sdk/provider-utils';
+import { Injectable } from '@nestjs/common';
 import {
   GenerateTextResult,
   LanguageModel,
@@ -9,10 +10,17 @@ import {
   generateText,
 } from 'ai';
 import { z } from 'zod';
-import { BlockConfig, ToolResult, WithArguments } from '@loopstack/common';
-import { ToolBase, WorkflowBase } from '@loopstack/core';
-import { Block } from '@loopstack/core/dist/workflow-processor/abstract/block.abstract';
-import { WorkflowExecution } from '@loopstack/core/dist/workflow-processor/interfaces/workflow-execution.interface';
+import {
+  DocumentInterface,
+  Tool,
+  ToolInterface,
+  ToolResult,
+  WithArguments,
+  WorkflowExecution,
+  WorkflowInterface,
+  getBlockArgsSchema,
+  getBlockDocument,
+} from '@loopstack/common';
 import { AiGenerateToolBaseSchema } from '../schemas/ai-generate-tool-base.schema';
 import { AiMessagesHelperService } from '../services';
 import { AiProviderModelHelperService } from '../services';
@@ -26,21 +34,24 @@ export const AiGenerateObjectSchema = AiGenerateToolBaseSchema.extend({
 
 export type AiGenerateObjectArgsType = z.infer<typeof AiGenerateObjectSchema>;
 
-@BlockConfig({
+@Injectable()
+@Tool({
   config: {
     description: 'Generates a structured object using a LLM',
   },
 })
 @WithArguments(AiGenerateObjectSchema)
-export class AiGenerateObject extends ToolBase<AiGenerateObjectArgsType> {
+export class AiGenerateObject implements ToolInterface<AiGenerateObjectArgsType> {
   constructor(
     private readonly aiMessagesHelperService: AiMessagesHelperService,
     private readonly aiProviderModelHelperService: AiProviderModelHelperService,
-  ) {
-    super();
-  }
+  ) {}
 
-  async execute(args: AiGenerateObjectArgsType, ctx: WorkflowExecution, parent: WorkflowBase): Promise<ToolResult> {
+  async execute(
+    args: AiGenerateObjectArgsType,
+    ctx: WorkflowExecution<any>,
+    parent: WorkflowInterface,
+  ): Promise<ToolResult> {
     const model = this.aiProviderModelHelperService.getProviderModel(args.llm);
 
     const options: {
@@ -64,11 +75,11 @@ export class AiGenerateObject extends ToolBase<AiGenerateObjectArgsType> {
       options.messages = await convertToModelMessages(messages);
     }
 
-    const document: Block | undefined = parent.getDocument(args.response.document);
+    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 = document.argsSchema;
+    const responseSchema = getBlockArgsSchema(document);
     if (!responseSchema) {
       throw new Error(`AI object generation source document must have a schema.`);
     }

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

@@ -2,9 +2,14 @@ import { ModelMessage } from '@ai-sdk/provider-utils';
 import { Injectable } from '@nestjs/common';
 import { LanguageModel, ToolSet, UIMessage, convertToModelMessages, createUIMessageStream, streamText } from 'ai';
 import { z } from 'zod';
-import { BlockConfig, ToolResult, WithArguments } from '@loopstack/common';
-import { ToolBase, WorkflowBase } from '@loopstack/core';
-import { WorkflowExecution } from '@loopstack/core/dist/workflow-processor/interfaces/workflow-execution.interface';
+import {
+  Tool,
+  ToolInterface,
+  ToolResult,
+  WithArguments,
+  WorkflowExecution,
+  WorkflowInterface,
+} from '@loopstack/common';
 import { AiGenerateToolBaseSchema } from '../schemas/ai-generate-tool-base.schema';
 import { AiMessagesHelperService } from '../services';
 import { AiProviderModelHelperService } from '../services';
@@ -17,22 +22,24 @@ export const AiGenerateTextSchema = AiGenerateToolBaseSchema.extend({
 type AiGenerateTextArgsType = z.infer<typeof AiGenerateTextSchema>;
 
 @Injectable()
-@BlockConfig({
+@Tool({
   config: {
     description: 'Generates text using a LLM',
   },
 })
 @WithArguments(AiGenerateTextSchema)
-export class AiGenerateText extends ToolBase<AiGenerateTextArgsType> {
+export class AiGenerateText implements ToolInterface<AiGenerateTextArgsType> {
   constructor(
     private readonly aiMessagesHelperService: AiMessagesHelperService,
     private readonly aiToolsHelperService: AiToolsHelperService,
     private readonly aiProviderModelHelperService: AiProviderModelHelperService,
-  ) {
-    super();
-  }
+  ) {}
 
-  async execute(args: AiGenerateTextArgsType, ctx: WorkflowExecution, parent: WorkflowBase): Promise<ToolResult> {
+  async execute(
+    args: AiGenerateTextArgsType,
+    ctx: WorkflowExecution<any>,
+    parent: WorkflowInterface,
+  ): Promise<ToolResult> {
     const model = this.aiProviderModelHelperService.getProviderModel(args.llm);
 
     const options: {

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

@@ -1,8 +1,15 @@
+import { Injectable } from '@nestjs/common';
 import { ToolUIPart, UIMessage } from 'ai';
 import { z } from 'zod';
-import { BlockConfig, ToolResult, WithArguments } from '@loopstack/common';
-import { ToolBase, WorkflowBase } from '@loopstack/core';
-import { WorkflowExecution } from '@loopstack/core/dist/workflow-processor/interfaces/workflow-execution.interface';
+import {
+  Tool,
+  ToolInterface,
+  ToolResult,
+  WithArguments,
+  WorkflowExecution,
+  WorkflowInterface,
+  getBlockTool,
+} from '@loopstack/common';
 
 const DelegateToolCallsToolSchema = z.object({
   message: z.object({
@@ -19,14 +26,19 @@ const DelegateToolCallsToolSchema = z.object({
 
 type DelegateToolCallsToolArgs = z.infer<typeof DelegateToolCallsToolSchema>;
 
-@BlockConfig({
+@Injectable()
+@Tool({
   config: {
     description: 'Delegate a tool call.',
   },
 })
 @WithArguments(DelegateToolCallsToolSchema)
-export class DelegateToolCall extends ToolBase<DelegateToolCallsToolArgs> {
-  async execute(args: DelegateToolCallsToolArgs, ctx: WorkflowExecution, parent: WorkflowBase): Promise<ToolResult> {
+export class DelegateToolCall implements ToolInterface<DelegateToolCallsToolArgs> {
+  async execute(
+    args: DelegateToolCallsToolArgs,
+    ctx: WorkflowExecution<any>,
+    parent: WorkflowInterface,
+  ): Promise<ToolResult> {
     const parts = args.message.parts;
     const resultParts: ToolUIPart[] = [];
 
@@ -41,7 +53,7 @@ export class DelegateToolCall extends ToolBase<DelegateToolCallsToolArgs> {
 
       const toolName = part.type.replace(/^tool-/, '');
 
-      const tool = parent.getTool(toolName);
+      const tool = getBlockTool<ToolInterface>(parent, toolName);
       if (!tool) {
         throw new Error(`Tool ${toolName} not found.`);
       }