@loopstack/agent-example-workflow 0.2.1 → 0.3.0

package/README.md

@@ -0,0 +1,36 @@
+# @loopstack/agent-example-workflow
+
+Demonstrates how to launch `AgentWorkflow` from [`@loopstack/agent`](../../features/agent-module) as a sub-workflow with custom tools (`weather_lookup`, `calculator`) and render progress/results in Studio.
+
+## By using this example you'll get...
+
+- A parent workflow that queues `AgentWorkflow` via `WORKFLOW_ORCHESTRATOR`
+- A `LinkDocument` that embeds the running child workflow
+- A callback transition that stores the final agent response as a `MessageDocument`
+
+## Installation
+
+```sh
+npm install @loopstack/agent-example-workflow
+```
+
+`@loopstack/agent` requires an LLM provider configuration. Use `AgentExampleModule.forFeature({ llm: ... })` when you want to override provider config per module.
+
+## How It Works
+
+1. `start` queues `AgentWorkflow` with a system prompt and selected tools.
+2. A `LinkDocument` is saved to show embedded progress in Studio.
+3. When the child workflow finishes, `agentComplete` receives callback payload data.
+4. The workflow marks the link as successful and saves the final assistant message.
+
+## Public API
+
+- `AgentExampleModule`
+- `AgentExampleWorkflow`
+- `CalculatorTool`
+- `WeatherLookupTool`
+
+## Dependencies
+
+- `@loopstack/common`
+- `@loopstack/agent`

package/dist/agent-example.module.d.ts

@@ -1,3 +1,8 @@
+import { DynamicModule } from '@nestjs/common';
+import type { LlmModuleConfig } from '@loopstack/llm-provider-module';
 export declare class AgentExampleModule {
+    static forFeature(config?: {
+        llm?: LlmModuleConfig;
+    }): DynamicModule;
 }
 //# sourceMappingURL=agent-example.module.d.ts.map

package/dist/agent-example.module.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent-example.module.d.ts","sourceRoot":"","sources":["../src/agent-example.module.ts"],"names":[],"mappings":"AAMA,qBAKa,kBAAkB;CAAG"}
+{"version":3,"file":"agent-example.module.d.ts","sourceRoot":"","sources":["../src/agent-example.module.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,aAAa,EAAU,MAAM,gBAAgB,CAAC;AAEvD,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,gCAAgC,CAAC;AAUtE,qBAKa,kBAAkB;IAC7B,MAAM,CAAC,UAAU,CAAC,MAAM,CAAC,EAAE;QAAE,GAAG,CAAC,EAAE,eAAe,CAAA;KAAE,GAAG,aAAa;CAQrE"}

package/dist/agent-example.module.js

@@ -12,14 +12,28 @@ const agent_1 = require("@loopstack/agent");
 const agent_example_workflow_1 = require("./agent-example.workflow");
 const calculator_tool_1 = require("./tools/calculator.tool");
 const weather_lookup_tool_1 = require("./tools/weather-lookup.tool");
+const PROVIDERS = [agent_example_workflow_1.AgentExampleWorkflow, calculator_tool_1.CalculatorTool, weather_lookup_tool_1.WeatherLookupTool];
+let AgentExampleRootModule = class AgentExampleRootModule {
+};
+AgentExampleRootModule = __decorate([
+    (0, common_1.Module)({})
+], AgentExampleRootModule);
 let AgentExampleModule = class AgentExampleModule {
+    static forFeature(config) {
+        return {
+            module: AgentExampleRootModule,
+            imports: [agent_1.AgentModule.forFeature(config)],
+            providers: PROVIDERS,
+            exports: PROVIDERS,
+        };
+    }
 };
 exports.AgentExampleModule = AgentExampleModule;
 exports.AgentExampleModule = AgentExampleModule = __decorate([
     (0, common_1.Module)({
         imports: [agent_1.AgentModule],
-        providers: [agent_example_workflow_1.AgentExampleWorkflow, calculator_tool_1.CalculatorTool, weather_lookup_tool_1.WeatherLookupTool],
-        exports: [agent_example_workflow_1.AgentExampleWorkflow, calculator_tool_1.CalculatorTool, weather_lookup_tool_1.WeatherLookupTool],
+        providers: PROVIDERS,
+        exports: PROVIDERS,
     })
 ], AgentExampleModule);
 //# sourceMappingURL=agent-example.module.js.map

package/dist/agent-example.module.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-example.module.js","sourceRoot":"","sources":["../src/agent-example.module.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAAwC;AACxC,4CAA+C;AAC/C,qEAAgE;AAChE,6DAAyD;AACzD,qEAAgE;AAOzD,IAAM,kBAAkB,GAAxB,MAAM,kBAAkB;CAAG,CAAA;AAArB,gDAAkB;6BAAlB,kBAAkB;IAL9B,IAAA,eAAM,EAAC;QACN,OAAO,EAAE,CAAC,mBAAW,CAAC;QACtB,SAAS,EAAE,CAAC,6CAAoB,EAAE,gCAAc,EAAE,uCAAiB,CAAC;QACpE,OAAO,EAAE,CAAC,6CAAoB,EAAE,gCAAc,EAAE,uCAAiB,CAAC;KACnE,CAAC;GACW,kBAAkB,CAAG"}
+{"version":3,"file":"agent-example.module.js","sourceRoot":"","sources":["../src/agent-example.module.ts"],"names":[],"mappings":";;;;;;;;;AAAA,2CAAuD;AACvD,4CAA+C;AAE/C,qEAAgE;AAChE,6DAAyD;AACzD,qEAAgE;AAEhE,MAAM,SAAS,GAAG,CAAC,6CAAoB,EAAE,gCAAc,EAAE,uCAAiB,CAAC,CAAC;AAG5E,IAAM,sBAAsB,GAA5B,MAAM,sBAAsB;CAAG,CAAA;AAAzB,sBAAsB;IAD3B,IAAA,eAAM,EAAC,EAAE,CAAC;GACL,sBAAsB,CAAG;AAOxB,IAAM,kBAAkB,GAAxB,MAAM,kBAAkB;IAC7B,MAAM,CAAC,UAAU,CAAC,MAAkC;QAClD,OAAO;YACL,MAAM,EAAE,sBAAsB;YAC9B,OAAO,EAAE,CAAC,mBAAW,CAAC,UAAU,CAAC,MAAM,CAAC,CAAC;YACzC,SAAS,EAAE,SAAS;YACpB,OAAO,EAAE,SAAS;SACnB,CAAC;IACJ,CAAC;CACF,CAAA;AATY,gDAAkB;6BAAlB,kBAAkB;IAL9B,IAAA,eAAM,EAAC;QACN,OAAO,EAAE,CAAC,mBAAW,CAAC;QACtB,SAAS,EAAE,SAAS;QACpB,OAAO,EAAE,SAAS;KACnB,CAAC;GACW,kBAAkB,CAS9B"}

package/dist/agent-example.workflow.d.ts

@@ -1,5 +1,7 @@
 import { z } from 'zod';
+import { AgentWorkflow } from '@loopstack/agent';
 import { BaseWorkflow } from '@loopstack/common';
+import type { TemplateRenderFn } from '@loopstack/common';
 declare const AgentCallbackSchema: z.ZodObject<{
     workflowId: z.ZodString;
     status: z.ZodString;
@@ -9,9 +11,11 @@ declare const AgentCallbackSchema: z.ZodObject<{
 }, z.core.$strip>;
 type AgentCallback = z.infer<typeof AgentCallbackSchema>;
 export declare class AgentExampleWorkflow extends BaseWorkflow {
-    private agent;
-    start(): Promise<void>;
-    agentComplete(payload: AgentCallback): Promise<void>;
+    private readonly agentWorkflow;
+    private readonly render;
+    constructor(agentWorkflow: AgentWorkflow, render: TemplateRenderFn);
+    start(state: Record<string, unknown>): Promise<Record<string, unknown>>;
+    agentComplete(state: Record<string, unknown>, payload: AgentCallback): Promise<unknown>;
 }
 export {};
 //# sourceMappingURL=agent-example.workflow.d.ts.map

package/dist/agent-example.workflow.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"agent-example.workflow.d.ts","sourceRoot":"","sources":["../src/agent-example.workflow.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,OAAO,EACL,YAAY,EASb,MAAM,mBAAmB,CAAC;AAE3B,QAAA,MAAM,mBAAmB;;;;;;iBAEvB,CAAC;AAEH,KAAK,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAEzD,qBAGa,oBAAqB,SAAQ,YAAY;IAClC,OAAO,CAAC,KAAK,CAAgB;IAGzC,KAAK;IAsBL,aAAa,CAAC,OAAO,EAAE,aAAa;CAY3C"}
+{"version":3,"file":"agent-example.workflow.d.ts","sourceRoot":"","sources":["../src/agent-example.workflow.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACjD,OAAO,EACL,YAAY,EAQb,MAAM,mBAAmB,CAAC;AAC3B,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAE1D,QAAA,MAAM,mBAAmB;;;;;;iBAEvB,CAAC;AAEH,KAAK,aAAa,GAAG,CAAC,CAAC,KAAK,CAAC,OAAO,mBAAmB,CAAC,CAAC;AAEzD,qBAIa,oBAAqB,SAAQ,YAAY;IAElD,OAAO,CAAC,QAAQ,CAAC,aAAa;IACH,OAAO,CAAC,QAAQ,CAAC,MAAM;gBADjC,aAAa,EAAE,aAAa,EACD,MAAM,EAAE,gBAAgB;IAMhE,KAAK,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,GAAG,OAAO,CAAC,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAwBvE,aAAa,CAAC,KAAK,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,OAAO,EAAE,aAAa,GAAG,OAAO,CAAC,OAAO,CAAC;CAa9F"}

package/dist/agent-example.workflow.js

@@ -8,56 +8,68 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
 var __metadata = (this && this.__metadata) || function (k, v) {
     if (typeof Reflect === "object" && typeof Reflect.metadata === "function") return Reflect.metadata(k, v);
 };
+var __param = (this && this.__param) || function (paramIndex, decorator) {
+    return function (target, key) { decorator(target, key, paramIndex); }
+};
 Object.defineProperty(exports, "__esModule", { value: true });
 exports.AgentExampleWorkflow = void 0;
+const common_1 = require("@nestjs/common");
 const zod_1 = require("zod");
 const agent_1 = require("@loopstack/agent");
-const common_1 = require("@loopstack/common");
-const AgentCallbackSchema = common_1.CallbackSchema.extend({
+const common_2 = require("@loopstack/common");
+const AgentCallbackSchema = common_2.CallbackSchema.extend({
     data: zod_1.z.object({ response: zod_1.z.string() }),
 });
-let AgentExampleWorkflow = class AgentExampleWorkflow extends common_1.BaseWorkflow {
-    agent;
-    async start() {
-        const result = await this.agent.run({
+let AgentExampleWorkflow = class AgentExampleWorkflow extends common_2.BaseWorkflow {
+    agentWorkflow;
+    render;
+    constructor(agentWorkflow, render) {
+        super();
+        this.agentWorkflow = agentWorkflow;
+        this.render = render;
+    }
+    async start(state) {
+        const result = await this.agentWorkflow.run({
             system: this.render(__dirname + '/templates/system.md'),
-            tools: ['weatherLookup', 'calculator'],
+            tools: ['weather_lookup', 'calculator'],
             userMessage: "What's the weather in Tokyo? Also, what is 42 * 17?",
-        }, { alias: 'agent', callback: { transition: 'agentComplete' } });
-        await this.repository.save(common_1.LinkDocument, { label: 'Agent working...', workflowId: result.workflowId, embed: true, expanded: true }, { id: `link_${result.workflowId}` });
+        }, { callback: { transition: 'agentComplete' } });
+        await this.documentStore.save(common_2.LinkDocument, { label: 'Agent working...', workflowId: result.workflowId, embed: true, expanded: true }, { id: `link_${result.workflowId}` });
+        return state;
     }
-    async agentComplete(payload) {
-        await this.repository.save(common_1.LinkDocument, { label: 'Agent complete', status: 'success', workflowId: payload.workflowId }, { id: `link_${payload.workflowId}` });
-        await this.repository.save(common_1.MessageDocument, {
+    async agentComplete(state, payload) {
+        await this.documentStore.save(common_2.LinkDocument, { label: 'Agent complete', status: 'success', workflowId: payload.workflowId }, { id: `link_${payload.workflowId}` });
+        await this.documentStore.save(common_2.MessageDocument, {
             role: 'assistant',
             content: payload.data.response,
         });
+        return {};
     }
 };
 exports.AgentExampleWorkflow = AgentExampleWorkflow;
 __decorate([
-    (0, common_1.InjectWorkflow)(),
-    __metadata("design:type", agent_1.AgentWorkflow)
-], AgentExampleWorkflow.prototype, "agent", void 0);
-__decorate([
-    (0, common_1.Initial)({ to: 'running' }),
+    (0, common_2.Transition)({ to: 'running' }),
     __metadata("design:type", Function),
-    __metadata("design:paramtypes", []),
+    __metadata("design:paramtypes", [Object]),
     __metadata("design:returntype", Promise)
 ], AgentExampleWorkflow.prototype, "start", null);
 __decorate([
-    (0, common_1.Final)({
+    (0, common_2.Transition)({
         from: 'running',
+        to: 'end',
         wait: true,
         schema: AgentCallbackSchema,
     }),
     __metadata("design:type", Function),
-    __metadata("design:paramtypes", [Object]),
+    __metadata("design:paramtypes", [Object, Object]),
     __metadata("design:returntype", Promise)
 ], AgentExampleWorkflow.prototype, "agentComplete", null);
 exports.AgentExampleWorkflow = AgentExampleWorkflow = __decorate([
-    (0, common_1.Workflow)({
-        uiConfig: __dirname + '/agent-example.ui.yaml',
-    })
+    (0, common_2.Workflow)({
+        title: 'Agent Example',
+        description: 'Launches a generic agent sub-workflow with weather and calculator tools.',
+    }),
+    __param(1, (0, common_1.Inject)(common_2.TEMPLATE_RENDERER)),
+    __metadata("design:paramtypes", [agent_1.AgentWorkflow, Function])
 ], AgentExampleWorkflow);
 //# sourceMappingURL=agent-example.workflow.js.map

package/dist/agent-example.workflow.js.map

@@ -1 +1 @@
-{"version":3,"file":"agent-example.workflow.js","sourceRoot":"","sources":["../src/agent-example.workflow.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,6BAAwB;AACxB,4CAAiD;AACjD,8CAU2B;AAE3B,MAAM,mBAAmB,GAAG,uBAAc,CAAC,MAAM,CAAC;IAChD,IAAI,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,EAAE,CAAC;CACzC,CAAC,CAAC;AAOI,IAAM,oBAAoB,GAA1B,MAAM,oBAAqB,SAAQ,qBAAY;IAC1B,KAAK,CAAgB;IAGzC,AAAN,KAAK,CAAC,KAAK;QACT,MAAM,MAAM,GAAgB,MAAM,IAAI,CAAC,KAAK,CAAC,GAAG,CAC9C;YACE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,sBAAsB,CAAC;YACvD,KAAK,EAAE,CAAC,eAAe,EAAE,YAAY,CAAC;YACtC,WAAW,EAAE,qDAAqD;SACnE,EACD,EAAE,KAAK,EAAE,OAAO,EAAE,QAAQ,EAAE,EAAE,UAAU,EAAE,eAAe,EAAE,EAAE,CAC9D,CAAC;QAEF,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACxB,qBAAY,EACZ,EAAE,KAAK,EAAE,kBAAkB,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,EACzF,EAAE,EAAE,EAAE,QAAQ,MAAM,CAAC,UAAU,EAAE,EAAE,CACpC,CAAC;IACJ,CAAC;IAOK,AAAN,KAAK,CAAC,aAAa,CAAC,OAAsB;QACxC,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CACxB,qBAAY,EACZ,EAAE,KAAK,EAAE,gBAAgB,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,EAC9E,EAAE,EAAE,EAAE,QAAQ,OAAO,CAAC,UAAU,EAAE,EAAE,CACrC,CAAC;QAEF,MAAM,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,wBAAe,EAAE;YAC1C,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,QAAQ;SAC/B,CAAC,CAAC;IACL,CAAC;CACF,CAAA;AAtCY,oDAAoB;AACL;IAAzB,IAAA,uBAAc,GAAE;8BAAgB,qBAAa;mDAAC;AAGzC;IADL,IAAA,gBAAO,EAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC;;;;iDAgB1B;AAOK;IALL,IAAA,cAAK,EAAC;QACL,IAAI,EAAE,SAAS;QACf,IAAI,EAAE,IAAI;QACV,MAAM,EAAE,mBAAmB;KAC5B,CAAC;;;;yDAYD;+BArCU,oBAAoB;IAHhC,IAAA,iBAAQ,EAAC;QACR,QAAQ,EAAE,SAAS,GAAG,wBAAwB;KAC/C,CAAC;GACW,oBAAoB,CAsChC"}
+{"version":3,"file":"agent-example.workflow.js","sourceRoot":"","sources":["../src/agent-example.workflow.ts"],"names":[],"mappings":";;;;;;;;;;;;;;;AAAA,2CAAwC;AACxC,6BAAwB;AACxB,4CAAiD;AACjD,8CAS2B;AAG3B,MAAM,mBAAmB,GAAG,uBAAc,CAAC,MAAM,CAAC;IAChD,IAAI,EAAE,OAAC,CAAC,MAAM,CAAC,EAAE,QAAQ,EAAE,OAAC,CAAC,MAAM,EAAE,EAAE,CAAC;CACzC,CAAC,CAAC;AAQI,IAAM,oBAAoB,GAA1B,MAAM,oBAAqB,SAAQ,qBAAY;IAEjC;IAC2B;IAF9C,YACmB,aAA4B,EACD,MAAwB;QAEpE,KAAK,EAAE,CAAC;QAHS,kBAAa,GAAb,aAAa,CAAe;QACD,WAAM,GAAN,MAAM,CAAkB;IAGtE,CAAC;IAGK,AAAN,KAAK,CAAC,KAAK,CAAC,KAA8B;QACxC,MAAM,MAAM,GAAgB,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CACtD;YACE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,SAAS,GAAG,sBAAsB,CAAC;YACvD,KAAK,EAAE,CAAC,gBAAgB,EAAE,YAAY,CAAC;YACvC,WAAW,EAAE,qDAAqD;SACnE,EACD,EAAE,QAAQ,EAAE,EAAE,UAAU,EAAE,eAAe,EAAE,EAAE,CAC9C,CAAC;QAEF,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAC3B,qBAAY,EACZ,EAAE,KAAK,EAAE,kBAAkB,EAAE,UAAU,EAAE,MAAM,CAAC,UAAU,EAAE,KAAK,EAAE,IAAI,EAAE,QAAQ,EAAE,IAAI,EAAE,EACzF,EAAE,EAAE,EAAE,QAAQ,MAAM,CAAC,UAAU,EAAE,EAAE,CACpC,CAAC;QACF,OAAO,KAAK,CAAC;IACf,CAAC;IAQK,AAAN,KAAK,CAAC,aAAa,CAAC,KAA8B,EAAE,OAAsB;QACxE,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAC3B,qBAAY,EACZ,EAAE,KAAK,EAAE,gBAAgB,EAAE,MAAM,EAAE,SAAS,EAAE,UAAU,EAAE,OAAO,CAAC,UAAU,EAAE,EAC9E,EAAE,EAAE,EAAE,QAAQ,OAAO,CAAC,UAAU,EAAE,EAAE,CACrC,CAAC;QAEF,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,wBAAe,EAAE;YAC7C,IAAI,EAAE,WAAW;YACjB,OAAO,EAAE,OAAO,CAAC,IAAI,CAAC,QAAQ;SAC/B,CAAC,CAAC;QACH,OAAO,EAAE,CAAC;IACZ,CAAC;CACF,CAAA;AA9CY,oDAAoB;AASzB;IADL,IAAA,mBAAU,EAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC;;;;iDAiB7B;AAQK;IANL,IAAA,mBAAU,EAAC;QACV,IAAI,EAAE,SAAS;QACf,EAAE,EAAE,KAAK;QACT,IAAI,EAAE,IAAI;QACV,MAAM,EAAE,mBAAmB;KAC5B,CAAC;;;;yDAaD;+BA7CU,oBAAoB;IAJhC,IAAA,iBAAQ,EAAC;QACR,KAAK,EAAE,eAAe;QACtB,WAAW,EAAE,0EAA0E;KACxF,CAAC;IAIG,WAAA,IAAA,eAAM,EAAC,0BAAiB,CAAC,CAAA;qCADM,qBAAa;GAFpC,oBAAoB,CA8ChC"}

package/dist/tools/calculator.tool.d.ts

@@ -1,9 +1,15 @@
 import { BaseTool, ToolResult } from '@loopstack/common';
-export declare class CalculatorTool extends BaseTool {
-    call(args: {
+import type { LoopstackContext } from '@loopstack/common';
+export type CalculatorResult = string;
+export declare class CalculatorTool extends BaseTool<{
+    operation: string;
+    a: number;
+    b: number;
+}, object, CalculatorResult> {
+    protected handle(args: {
         operation: string;
         a: number;
         b: number;
-    }): Promise<ToolResult>;
+    }, _ctx: LoopstackContext): Promise<ToolResult<CalculatorResult>>;
 }
 //# sourceMappingURL=calculator.tool.d.ts.map

package/dist/tools/calculator.tool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"calculator.tool.d.ts","sourceRoot":"","sources":["../../src/tools/calculator.tool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAQ,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE/D,qBAUa,cAAe,SAAQ,QAAQ;IAC1C,IAAI,CAAC,IAAI,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,UAAU,CAAC;CA2B7E"}
+{"version":3,"file":"calculator.tool.d.ts","sourceRoot":"","sources":["../../src/tools/calculator.tool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAQ,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAE1D,MAAM,MAAM,gBAAgB,GAAG,MAAM,CAAC;AAEtC,qBASa,cAAe,SAAQ,QAAQ,CAAC;IAAE,SAAS,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAC;IAAC,CAAC,EAAE,MAAM,CAAA;CAAE,EAAE,MAAM,EAAE,gBAAgB,CAAC;cACjG,MAAM,CACpB,IAAI,EAAE;QAAE,SAAS,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAC;QAAC,CAAC,EAAE,MAAM,CAAA;KAAE,EACjD,IAAI,EAAE,gBAAgB,GACrB,OAAO,CAAC,UAAU,CAAC,gBAAgB,CAAC,CAAC;CA2BzC"}

package/dist/tools/calculator.tool.js

@@ -10,7 +10,7 @@ exports.CalculatorTool = void 0;
 const zod_1 = require("zod");
 const common_1 = require("@loopstack/common");
 let CalculatorTool = class CalculatorTool extends common_1.BaseTool {
-    call(args) {
+    async handle(args, _ctx) {
         let result;
         switch (args.operation) {
             case 'add':
@@ -24,24 +24,23 @@ let CalculatorTool = class CalculatorTool extends common_1.BaseTool {
                 break;
             case 'divide':
                 if (args.b === 0) {
-                    return Promise.resolve({ data: 'Error: Division by zero', error: 'Cannot divide by zero' });
+                    return { data: 'Error: Division by zero', error: 'Cannot divide by zero' };
                 }
                 result = args.a / args.b;
                 break;
             default:
-                return Promise.resolve({ data: `Unknown operation: ${args.operation}`, error: 'Unsupported operation' });
+                return { data: `Unknown operation: ${args.operation}`, error: 'Unsupported operation' };
         }
-        return Promise.resolve({
+        return {
             data: `${args.a} ${args.operation} ${args.b} = ${result}`,
-        });
+        };
     }
 };
 exports.CalculatorTool = CalculatorTool;
 exports.CalculatorTool = CalculatorTool = __decorate([
     (0, common_1.Tool)({
-        uiConfig: {
-            description: 'Perform a basic arithmetic calculation. Supports add, subtract, multiply, divide.',
-        },
+        name: 'calculator',
+        description: 'Perform a basic arithmetic calculation. Supports add, subtract, multiply, divide.',
         schema: zod_1.z.object({
             operation: zod_1.z.enum(['add', 'subtract', 'multiply', 'divide']).describe('The arithmetic operation.'),
             a: zod_1.z.number().describe('First operand.'),

package/dist/tools/calculator.tool.js.map

@@ -1 +1 @@
-{"version":3,"file":"calculator.tool.js","sourceRoot":"","sources":["../../src/tools/calculator.tool.ts"],"names":[],"mappings":";;;;;;;;;AAAA,6BAAwB;AACxB,8CAA+D;AAYxD,IAAM,cAAc,GAApB,MAAM,cAAe,SAAQ,iBAAQ;IAC1C,IAAI,CAAC,IAAiD;QACpD,IAAI,MAAc,CAAC;QAEnB,QAAQ,IAAI,CAAC,SAAS,EAAE,CAAC;YACvB,KAAK,KAAK;gBACR,MAAM,GAAG,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;gBACzB,MAAM;YACR,KAAK,UAAU;gBACb,MAAM,GAAG,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;gBACzB,MAAM;YACR,KAAK,UAAU;gBACb,MAAM,GAAG,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;gBACzB,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,IAAI,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;oBACjB,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,yBAAyB,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC,CAAC;gBAC9F,CAAC;gBACD,MAAM,GAAG,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;gBACzB,MAAM;YACR;gBACE,OAAO,OAAO,CAAC,OAAO,CAAC,EAAE,IAAI,EAAE,sBAAsB,IAAI,CAAC,SAAS,EAAE,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC,CAAC;QAC7G,CAAC;QAED,OAAO,OAAO,CAAC,OAAO,CAAC;YACrB,IAAI,EAAE,GAAG,IAAI,CAAC,CAAC,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,MAAM,MAAM,EAAE;SAC1D,CAAC,CAAC;IACL,CAAC;CACF,CAAA;AA5BY,wCAAc;yBAAd,cAAc;IAV1B,IAAA,aAAI,EAAC;QACJ,QAAQ,EAAE;YACR,WAAW,EAAE,mFAAmF;SACjG;QACD,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC;YACf,SAAS,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,2BAA2B,CAAC;YAClG,CAAC,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC;YACxC,CAAC,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;SAC1C,CAAC;KACH,CAAC;GACW,cAAc,CA4B1B"}
+{"version":3,"file":"calculator.tool.js","sourceRoot":"","sources":["../../src/tools/calculator.tool.ts"],"names":[],"mappings":";;;;;;;;;AAAA,6BAAwB;AACxB,8CAA+D;AAcxD,IAAM,cAAc,GAApB,MAAM,cAAe,SAAQ,iBAA+E;IACvG,KAAK,CAAC,MAAM,CACpB,IAAiD,EACjD,IAAsB;QAEtB,IAAI,MAAc,CAAC;QAEnB,QAAQ,IAAI,CAAC,SAAS,EAAE,CAAC;YACvB,KAAK,KAAK;gBACR,MAAM,GAAG,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;gBACzB,MAAM;YACR,KAAK,UAAU;gBACb,MAAM,GAAG,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;gBACzB,MAAM;YACR,KAAK,UAAU;gBACb,MAAM,GAAG,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;gBACzB,MAAM;YACR,KAAK,QAAQ;gBACX,IAAI,IAAI,CAAC,CAAC,KAAK,CAAC,EAAE,CAAC;oBACjB,OAAO,EAAE,IAAI,EAAE,yBAAyB,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC;gBAC7E,CAAC;gBACD,MAAM,GAAG,IAAI,CAAC,CAAC,GAAG,IAAI,CAAC,CAAC,CAAC;gBACzB,MAAM;YACR;gBACE,OAAO,EAAE,IAAI,EAAE,sBAAsB,IAAI,CAAC,SAAS,EAAE,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAAC;QAC5F,CAAC;QAED,OAAO;YACL,IAAI,EAAE,GAAG,IAAI,CAAC,CAAC,IAAI,IAAI,CAAC,SAAS,IAAI,IAAI,CAAC,CAAC,MAAM,MAAM,EAAE;SAC1D,CAAC;IACJ,CAAC;CACF,CAAA;AA/BY,wCAAc;yBAAd,cAAc;IAT1B,IAAA,aAAI,EAAC;QACJ,IAAI,EAAE,YAAY;QAClB,WAAW,EAAE,mFAAmF;QAChG,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC;YACf,SAAS,EAAE,OAAC,CAAC,IAAI,CAAC,CAAC,KAAK,EAAE,UAAU,EAAE,UAAU,EAAE,QAAQ,CAAC,CAAC,CAAC,QAAQ,CAAC,2BAA2B,CAAC;YAClG,CAAC,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,gBAAgB,CAAC;YACxC,CAAC,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,iBAAiB,CAAC;SAC1C,CAAC;KACH,CAAC;GACW,cAAc,CA+B1B"}

package/dist/tools/weather-lookup.tool.d.ts

@@ -1,7 +1,11 @@
 import { BaseTool, ToolResult } from '@loopstack/common';
-export declare class WeatherLookupTool extends BaseTool {
-    call(args: {
+import type { LoopstackContext } from '@loopstack/common';
+export type WeatherLookupResult = string;
+export declare class WeatherLookupTool extends BaseTool<{
+    city: string;
+}, object, WeatherLookupResult> {
+    protected handle(args: {
         city: string;
-    }): Promise<ToolResult>;
+    }, _ctx: LoopstackContext): Promise<ToolResult<WeatherLookupResult>>;
 }
 //# sourceMappingURL=weather-lookup.tool.d.ts.map

package/dist/tools/weather-lookup.tool.d.ts.map

@@ -1 +1 @@
-{"version":3,"file":"weather-lookup.tool.d.ts","sourceRoot":"","sources":["../../src/tools/weather-lookup.tool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAQ,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAE/D,qBAQa,iBAAkB,SAAQ,QAAQ;IAC7C,IAAI,CAAC,IAAI,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,GAAG,OAAO,CAAC,UAAU,CAAC;CAelD"}
+{"version":3,"file":"weather-lookup.tool.d.ts","sourceRoot":"","sources":["../../src/tools/weather-lookup.tool.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,QAAQ,EAAQ,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAC/D,OAAO,KAAK,EAAE,gBAAgB,EAAE,MAAM,mBAAmB,CAAC;AAE1D,MAAM,MAAM,mBAAmB,GAAG,MAAM,CAAC;AAEzC,qBAOa,iBAAkB,SAAQ,QAAQ,CAAC;IAAE,IAAI,EAAE,MAAM,CAAA;CAAE,EAAE,MAAM,EAAE,mBAAmB,CAAC;cAC5E,MAAM,CAAC,IAAI,EAAE;QAAE,IAAI,EAAE,MAAM,CAAA;KAAE,EAAE,IAAI,EAAE,gBAAgB,GAAG,OAAO,CAAC,UAAU,CAAC,mBAAmB,CAAC,CAAC;CAejH"}

package/dist/tools/weather-lookup.tool.js

@@ -10,7 +10,7 @@ exports.WeatherLookupTool = void 0;
 const zod_1 = require("zod");
 const common_1 = require("@loopstack/common");
 let WeatherLookupTool = class WeatherLookupTool extends common_1.BaseTool {
-    call(args) {
+    async handle(args, _ctx) {
         const forecasts = {
             london: '14°C, cloudy with light rain',
             tokyo: '22°C, sunny with clear skies',
@@ -26,9 +26,8 @@ let WeatherLookupTool = class WeatherLookupTool extends common_1.BaseTool {
 exports.WeatherLookupTool = WeatherLookupTool;
 exports.WeatherLookupTool = WeatherLookupTool = __decorate([
     (0, common_1.Tool)({
-        uiConfig: {
-            description: 'Look up the current weather for a given city. Returns a simulated forecast.',
-        },
+        name: 'weather_lookup',
+        description: 'Look up the current weather for a given city. Returns a simulated forecast.',
         schema: zod_1.z.object({
             city: zod_1.z.string().describe('The city name to look up weather for.'),
         }),

package/dist/tools/weather-lookup.tool.js.map

@@ -1 +1 @@
-{"version":3,"file":"weather-lookup.tool.js","sourceRoot":"","sources":["../../src/tools/weather-lookup.tool.ts"],"names":[],"mappings":";;;;;;;;;AAAA,6BAAwB;AACxB,8CAA+D;AAUxD,IAAM,iBAAiB,GAAvB,MAAM,iBAAkB,SAAQ,iBAAQ;IAC7C,IAAI,CAAC,IAAsB;QAEzB,MAAM,SAAS,GAA2B;YACxC,MAAM,EAAE,8BAA8B;YACtC,KAAK,EAAE,8BAA8B;YACrC,UAAU,EAAE,qBAAqB;SAClC,CAAC;QAEF,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,2BAA2B,CAAC;QAE9D,OAAO,OAAO,CAAC,OAAO,CAAC;YACrB,IAAI,EAAE,cAAc,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE;SAC5C,CAAC,CAAC;IACL,CAAC;CACF,CAAA;AAhBY,8CAAiB;4BAAjB,iBAAiB;IAR7B,IAAA,aAAI,EAAC;QACJ,QAAQ,EAAE;YACR,WAAW,EAAE,6EAA6E;SAC3F;QACD,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC;YACf,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uCAAuC,CAAC;SACnE,CAAC;KACH,CAAC;GACW,iBAAiB,CAgB7B"}
+{"version":3,"file":"weather-lookup.tool.js","sourceRoot":"","sources":["../../src/tools/weather-lookup.tool.ts"],"names":[],"mappings":";;;;;;;;;AAAA,6BAAwB;AACxB,8CAA+D;AAYxD,IAAM,iBAAiB,GAAvB,MAAM,iBAAkB,SAAQ,iBAAuD;IAClF,KAAK,CAAC,MAAM,CAAC,IAAsB,EAAE,IAAsB;QAEnE,MAAM,SAAS,GAA2B;YACxC,MAAM,EAAE,8BAA8B;YACtC,KAAK,EAAE,8BAA8B;YACrC,UAAU,EAAE,qBAAqB;SAClC,CAAC;QAEF,MAAM,GAAG,GAAG,IAAI,CAAC,IAAI,CAAC,WAAW,EAAE,CAAC;QACpC,MAAM,OAAO,GAAG,SAAS,CAAC,GAAG,CAAC,IAAI,2BAA2B,CAAC;QAE9D,OAAO,OAAO,CAAC,OAAO,CAAC;YACrB,IAAI,EAAE,cAAc,IAAI,CAAC,IAAI,KAAK,OAAO,EAAE;SAC5C,CAAC,CAAC;IACL,CAAC;CACF,CAAA;AAhBY,8CAAiB;4BAAjB,iBAAiB;IAP7B,IAAA,aAAI,EAAC;QACJ,IAAI,EAAE,gBAAgB;QACtB,WAAW,EAAE,6EAA6E;QAC1F,MAAM,EAAE,OAAC,CAAC,MAAM,CAAC;YACf,IAAI,EAAE,OAAC,CAAC,MAAM,EAAE,CAAC,QAAQ,CAAC,uCAAuC,CAAC;SACnE,CAAC;KACH,CAAC;GACW,iBAAiB,CAgB7B"}

package/package.json

@@ -9,7 +9,7 @@
     "claude",
     "sub-workflow"
   ],
-  "version": "0.2.1",
+  "version": "0.3.0",
   "license": "MIT",
   "author": {
     "name": "Jakob Klippel",
@@ -30,9 +30,8 @@
     "watch": "nest build --watch"
   },
   "dependencies": {
-    "@loopstack/agent": "^0.4.1",
-    "@loopstack/common": "^0.31.0",
-    "@loopstack/core": "^0.31.0",
+    "@loopstack/agent": "^0.5.0",
+    "@loopstack/common": "^0.32.0",
     "@nestjs/common": "^11.1.19",
     "zod": "^4.3.6"
   },

package/src/__tests__/agent-example.workflow.spec.ts

@@ -8,22 +8,22 @@ import { AgentExampleWorkflow } from '../agent-example.workflow';
 import { CalculatorTool } from '../tools/calculator.tool';
 import { WeatherLookupTool } from '../tools/weather-lookup.tool';
 
+const mockAgentWorkflow = {
+  run: vi.fn(),
+};
+
 describe('AgentExampleWorkflow', () => {
   let module: TestingModule;
   let workflow: AgentExampleWorkflow;
   let processor: WorkflowProcessorService;
 
-  const mockAgent = {
-    run: vi.fn(),
-  };
-
   beforeEach(async () => {
     vi.clearAllMocks();
 
     module = await createWorkflowTest()
       .forWorkflow(AgentExampleWorkflow)
       .withProviders(CalculatorTool, WeatherLookupTool)
-      .withMock(AgentWorkflow, mockAgent)
+      .withMock(AgentWorkflow, mockAgentWorkflow)
       .compile();
 
     workflow = module.get(AgentExampleWorkflow);
@@ -35,7 +35,7 @@ describe('AgentExampleWorkflow', () => {
   });
 
   it('launches AgentWorkflow and stops at running', async () => {
-    mockAgent.run.mockResolvedValue({ workflowId: 'agent-sub-id' });
+    mockAgentWorkflow.run.mockResolvedValue({ workflowId: 'agent-sub-id' });
 
     const result = await processor.process(workflow, {}, createStatelessContext());
 
@@ -43,19 +43,19 @@ describe('AgentExampleWorkflow', () => {
     expect(result.stop).toBe(true);
     expect(result.place).toBe('running');
 
-    expect(mockAgent.run).toHaveBeenCalledWith(
+    expect(mockAgentWorkflow.run).toHaveBeenCalledWith(
       expect.objectContaining({
         system: expect.any(String),
-        tools: ['weatherLookup', 'calculator'],
+        tools: ['weather_lookup', 'calculator'],
         userMessage: expect.any(String),
       }),
-      expect.objectContaining({ alias: 'agent', callback: { transition: 'agentComplete' } }),
+      { callback: { transition: 'agentComplete' } },
     );
 
     expect(result.documents).toEqual(
       expect.arrayContaining([
         expect.objectContaining({
-          className: 'LinkDocument',
+          documentName: 'link',
           content: expect.objectContaining({ workflowId: 'agent-sub-id' }),
         }),
       ]),
@@ -89,7 +89,7 @@ describe('AgentExampleWorkflow', () => {
     expect(result.documents).toEqual(
       expect.arrayContaining([
         expect.objectContaining({
-          className: 'MessageDocument',
+          documentName: 'message',
           content: expect.objectContaining({ content: expect.stringContaining('714') }),
         }),
       ]),

package/src/agent-example.module.ts

@@ -1,12 +1,27 @@
-import { Module } from '@nestjs/common';
+import { DynamicModule, Module } from '@nestjs/common';
 import { AgentModule } from '@loopstack/agent';
+import type { LlmModuleConfig } from '@loopstack/llm-provider-module';
 import { AgentExampleWorkflow } from './agent-example.workflow';
 import { CalculatorTool } from './tools/calculator.tool';
 import { WeatherLookupTool } from './tools/weather-lookup.tool';
 
+const PROVIDERS = [AgentExampleWorkflow, CalculatorTool, WeatherLookupTool];
+
+@Module({})
+class AgentExampleRootModule {}
+
 @Module({
   imports: [AgentModule],
-  providers: [AgentExampleWorkflow, CalculatorTool, WeatherLookupTool],
-  exports: [AgentExampleWorkflow, CalculatorTool, WeatherLookupTool],
+  providers: PROVIDERS,
+  exports: PROVIDERS,
 })
-export class AgentExampleModule {}
+export class AgentExampleModule {
+  static forFeature(config?: { llm?: LlmModuleConfig }): DynamicModule {
+    return {
+      module: AgentExampleRootModule,
+      imports: [AgentModule.forFeature(config)],
+      providers: PROVIDERS,
+      exports: PROVIDERS,
+    };
+  }
+}

package/src/agent-example.workflow.ts

@@ -1,16 +1,17 @@
+import { Inject } from '@nestjs/common';
 import { z } from 'zod';
 import { AgentWorkflow } from '@loopstack/agent';
 import {
   BaseWorkflow,
   CallbackSchema,
-  Final,
-  Initial,
-  InjectWorkflow,
   LinkDocument,
   MessageDocument,
   QueueResult,
+  TEMPLATE_RENDERER,
+  Transition,
   Workflow,
 } from '@loopstack/common';
+import type { TemplateRenderFn } from '@loopstack/common';
 
 const AgentCallbackSchema = CallbackSchema.extend({
   data: z.object({ response: z.string() }),
@@ -19,44 +20,53 @@ const AgentCallbackSchema = CallbackSchema.extend({
 type AgentCallback = z.infer<typeof AgentCallbackSchema>;
 
 @Workflow({
-  uiConfig: __dirname + '/agent-example.ui.yaml',
+  title: 'Agent Example',
+  description: 'Launches a generic agent sub-workflow with weather and calculator tools.',
 })
 export class AgentExampleWorkflow extends BaseWorkflow {
-  @InjectWorkflow() private agent: AgentWorkflow;
+  constructor(
+    private readonly agentWorkflow: AgentWorkflow,
+    @Inject(TEMPLATE_RENDERER) private readonly render: TemplateRenderFn,
+  ) {
+    super();
+  }
 
-  @Initial({ to: 'running' })
-  async start() {
-    const result: QueueResult = await this.agent.run(
+  @Transition({ to: 'running' })
+  async start(state: Record<string, unknown>): Promise<Record<string, unknown>> {
+    const result: QueueResult = await this.agentWorkflow.run(
       {
         system: this.render(__dirname + '/templates/system.md'),
-        tools: ['weatherLookup', 'calculator'],
+        tools: ['weather_lookup', 'calculator'],
         userMessage: "What's the weather in Tokyo? Also, what is 42 * 17?",
       },
-      { alias: 'agent', callback: { transition: 'agentComplete' } },
+      { callback: { transition: 'agentComplete' } },
     );
 
-    await this.repository.save(
+    await this.documentStore.save(
       LinkDocument,
       { label: 'Agent working...', workflowId: result.workflowId, embed: true, expanded: true },
       { id: `link_${result.workflowId}` },
     );
+    return state;
   }
 
-  @Final({
+  @Transition({
     from: 'running',
+    to: 'end',
     wait: true,
     schema: AgentCallbackSchema,
   })
-  async agentComplete(payload: AgentCallback) {
-    await this.repository.save(
+  async agentComplete(state: Record<string, unknown>, payload: AgentCallback): Promise<unknown> {
+    await this.documentStore.save(
       LinkDocument,
       { label: 'Agent complete', status: 'success', workflowId: payload.workflowId },
       { id: `link_${payload.workflowId}` },
     );
 
-    await this.repository.save(MessageDocument, {
+    await this.documentStore.save(MessageDocument, {
       role: 'assistant',
       content: payload.data.response,
     });
+    return {};
   }
 }

package/src/tools/calculator.tool.ts

@@ -1,18 +1,23 @@
 import { z } from 'zod';
 import { BaseTool, Tool, ToolResult } from '@loopstack/common';
+import type { LoopstackContext } from '@loopstack/common';
+
+export type CalculatorResult = string;
 
 @Tool({
-  uiConfig: {
-    description: 'Perform a basic arithmetic calculation. Supports add, subtract, multiply, divide.',
-  },
+  name: 'calculator',
+  description: 'Perform a basic arithmetic calculation. Supports add, subtract, multiply, divide.',
   schema: z.object({
     operation: z.enum(['add', 'subtract', 'multiply', 'divide']).describe('The arithmetic operation.'),
     a: z.number().describe('First operand.'),
     b: z.number().describe('Second operand.'),
   }),
 })
-export class CalculatorTool extends BaseTool {
-  call(args: { operation: string; a: number; b: number }): Promise<ToolResult> {
+export class CalculatorTool extends BaseTool<{ operation: string; a: number; b: number }, object, CalculatorResult> {
+  protected async handle(
+    args: { operation: string; a: number; b: number },
+    _ctx: LoopstackContext,
+  ): Promise<ToolResult<CalculatorResult>> {
     let result: number;
 
     switch (args.operation) {
@@ -27,16 +32,16 @@ export class CalculatorTool extends BaseTool {
         break;
       case 'divide':
         if (args.b === 0) {
-          return Promise.resolve({ data: 'Error: Division by zero', error: 'Cannot divide by zero' });
+          return { data: 'Error: Division by zero', error: 'Cannot divide by zero' };
         }
         result = args.a / args.b;
         break;
       default:
-        return Promise.resolve({ data: `Unknown operation: ${args.operation}`, error: 'Unsupported operation' });
+        return { data: `Unknown operation: ${args.operation}`, error: 'Unsupported operation' };
     }
 
-    return Promise.resolve({
+    return {
       data: `${args.a} ${args.operation} ${args.b} = ${result}`,
-    });
+    };
   }
 }

package/src/tools/weather-lookup.tool.ts

@@ -1,16 +1,18 @@
 import { z } from 'zod';
 import { BaseTool, Tool, ToolResult } from '@loopstack/common';
+import type { LoopstackContext } from '@loopstack/common';
+
+export type WeatherLookupResult = string;
 
 @Tool({
-  uiConfig: {
-    description: 'Look up the current weather for a given city. Returns a simulated forecast.',
-  },
+  name: 'weather_lookup',
+  description: 'Look up the current weather for a given city. Returns a simulated forecast.',
   schema: z.object({
     city: z.string().describe('The city name to look up weather for.'),
   }),
 })
-export class WeatherLookupTool extends BaseTool {
-  call(args: { city: string }): Promise<ToolResult> {
+export class WeatherLookupTool extends BaseTool<{ city: string }, object, WeatherLookupResult> {
+  protected async handle(args: { city: string }, _ctx: LoopstackContext): Promise<ToolResult<WeatherLookupResult>> {
     // Simulated weather data for demonstration purposes
     const forecasts: Record<string, string> = {
       london: '14°C, cloudy with light rain',

package/dist/agent-example.ui.yaml

@@ -1,2 +0,0 @@
-title: 'Agent Example'
-description: Launches a generic agent sub-workflow with weather and calculator tools.

package/src/agent-example.ui.yaml

@@ -1,2 +0,0 @@
-title: 'Agent Example'
-description: Launches a generic agent sub-workflow with weather and calculator tools.