@loopstack/agent-examples 0.1.1 → 0.1.3

package/README.md

@@ -38,6 +38,26 @@ npm install @loopstack/agent-examples
 import { AgentExamplesModule } from '@loopstack/agent-examples';
 ```
 
+## Required app-module configuration
+
+All four examples use LLM tools (`LlmGenerateTextTool`, `LlmDelegateToolCallsTool`, etc.) from `@loopstack/llm-provider-module`. That module is `@Global` and must be configured once in your root module to set the default model:
+
+```typescript
+import { Module } from '@nestjs/common';
+import { AgentExamplesModule } from '@loopstack/agent-examples';
+import { LlmProviderModule } from '@loopstack/llm-provider-module';
+import { LoopstackModule } from '@loopstack/loopstack-module';
+
+@Module({
+  imports: [LoopstackModule.forRoot(), LlmProviderModule.forRoot({ model: 'claude-sonnet-4-6' }), AgentExamplesModule],
+})
+export class AppModule {}
+```
+
+`AgentExamplesModule` re-imports `ClaudeModule` (provider), `AgentModule` (the `AgentWorkflow` / `ChatAgentWorkflow` classes), `CodeAgentModule`, `McpModule`, and `RemoteClientModule.forFeature({ type: 'sandbox' })`. `LlmProviderModule.forRoot(...)` sets the default model the tools dispatch to.
+
+The Code Agent example additionally requires `RemoteClientModule.forRoot({ environments })` — see [Code Agent → Setup](#code-agent) below.
+
 ## Environment
 
 ```bash

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

@@ -1,6 +1,8 @@
+import { z } from 'zod';
 import { BaseTool, ToolEnvelope } from '@loopstack/common';
 import type { RunContext } from '@loopstack/common';
 export type CalculatorResult = string;
+export declare const CalculatorResultSchema: z.ZodString;
 export declare class CalculatorTool extends BaseTool<{
     operation: string;
     a: number;

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,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAEpD,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,UAAU,GACf,OAAO,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;CA2B3C"}
+{"version":3,"file":"calculator.tool.d.ts","sourceRoot":"","sources":["../../src/tools/calculator.tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,QAAQ,EAAQ,YAAY,EAAE,MAAM,mBAAmB,CAAC;AACjE,OAAO,KAAK,EAAE,UAAU,EAAE,MAAM,mBAAmB,CAAC;AAEpD,MAAM,MAAM,gBAAgB,GAAG,MAAM,CAAC;AAEtC,eAAO,MAAM,sBAAsB,aAAa,CAAC;AAEjD,qBAWa,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,UAAU,GACf,OAAO,CAAC,YAAY,CAAC,gBAAgB,CAAC,CAAC;CA2B3C"}

package/dist/tools/calculator.tool.js

@@ -6,9 +6,10 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
     return c > 3 && r && Object.defineProperty(target, key, r), r;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.CalculatorTool = void 0;
+exports.CalculatorTool = exports.CalculatorResultSchema = void 0;
 const zod_1 = require("zod");
 const common_1 = require("@loopstack/common");
+exports.CalculatorResultSchema = zod_1.z.string();
 let CalculatorTool = class CalculatorTool extends common_1.BaseTool {
     async handle(args, _ctx) {
         let result;
@@ -46,6 +47,8 @@ exports.CalculatorTool = CalculatorTool = __decorate([
             a: zod_1.z.number().describe('First operand.'),
             b: zod_1.z.number().describe('Second operand.'),
         }),
+        resultSchema: exports.CalculatorResultSchema,
+        effects: 'none',
     })
 ], CalculatorTool);
 //# sourceMappingURL=calculator.tool.js.map

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,8CAAiE;AAc1D,IAAM,cAAc,GAApB,MAAM,cAAe,SAAQ,iBAA+E;IACvG,KAAK,CAAC,MAAM,CACpB,IAAiD,EACjD,IAAgB;QAEhB,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"}
+{"version":3,"file":"calculator.tool.js","sourceRoot":"","sources":["../../src/tools/calculator.tool.ts"],"names":[],"mappings":";;;;;;;;;AAAA,6BAAwB;AACxB,8CAAiE;AAKpD,QAAA,sBAAsB,GAAG,OAAC,CAAC,MAAM,EAAE,CAAC;AAa1C,IAAM,cAAc,GAApB,MAAM,cAAe,SAAQ,iBAA+E;IACvG,KAAK,CAAC,MAAM,CACpB,IAAiD,EACjD,IAAgB;QAEhB,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;IAX1B,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;QACF,YAAY,EAAE,8BAAsB;QACpC,OAAO,EAAE,MAAM;KAChB,CAAC;GACW,cAAc,CA+B1B"}

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

@@ -1,5 +1,7 @@
+import { z } from 'zod';
 import { BaseTool, ToolEnvelope } from '@loopstack/common';
 export type WeatherLookupResult = string;
+export declare const WeatherLookupResultSchema: z.ZodString;
 export declare class WeatherLookupTool extends BaseTool<{
     city: string;
 }, object, WeatherLookupResult> {

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,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjE,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,GAAG,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC;CAe3F"}
+{"version":3,"file":"weather-lookup.tool.d.ts","sourceRoot":"","sources":["../../src/tools/weather-lookup.tool.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AACxB,OAAO,EAAE,QAAQ,EAAQ,YAAY,EAAE,MAAM,mBAAmB,CAAC;AAEjE,MAAM,MAAM,mBAAmB,GAAG,MAAM,CAAC;AAEzC,eAAO,MAAM,yBAAyB,aAAa,CAAC;AAEpD,qBASa,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,GAAG,OAAO,CAAC,YAAY,CAAC,mBAAmB,CAAC,CAAC;CAe3F"}

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

@@ -6,9 +6,10 @@ var __decorate = (this && this.__decorate) || function (decorators, target, key,
     return c > 3 && r && Object.defineProperty(target, key, r), r;
 };
 Object.defineProperty(exports, "__esModule", { value: true });
-exports.WeatherLookupTool = void 0;
+exports.WeatherLookupTool = exports.WeatherLookupResultSchema = void 0;
 const zod_1 = require("zod");
 const common_1 = require("@loopstack/common");
+exports.WeatherLookupResultSchema = zod_1.z.string();
 let WeatherLookupTool = class WeatherLookupTool extends common_1.BaseTool {
     async handle(args) {
         const forecasts = {
@@ -31,6 +32,8 @@ exports.WeatherLookupTool = WeatherLookupTool = __decorate([
         schema: zod_1.z.object({
             city: zod_1.z.string().describe('The city name to look up weather for.'),
         }),
+        resultSchema: exports.WeatherLookupResultSchema,
+        effects: 'none',
     })
 ], WeatherLookupTool);
 //# sourceMappingURL=weather-lookup.tool.js.map

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,8CAAiE;AAW1D,IAAM,iBAAiB,GAAvB,MAAM,iBAAkB,SAAQ,iBAAuD;IAClF,KAAK,CAAC,MAAM,CAAC,IAAsB;QAE3C,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"}
+{"version":3,"file":"weather-lookup.tool.js","sourceRoot":"","sources":["../../src/tools/weather-lookup.tool.ts"],"names":[],"mappings":";;;;;;;;;AAAA,6BAAwB;AACxB,8CAAiE;AAIpD,QAAA,yBAAyB,GAAG,OAAC,CAAC,MAAM,EAAE,CAAC;AAW7C,IAAM,iBAAiB,GAAvB,MAAM,iBAAkB,SAAQ,iBAAuD;IAClF,KAAK,CAAC,MAAM,CAAC,IAAsB;QAE3C,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;IAT7B,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;QACF,YAAY,EAAE,iCAAyB;QACvC,OAAO,EAAE,MAAM;KAChB,CAAC;GACW,iBAAiB,CAgB7B"}

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

@@ -1 +1 @@
-{"version":3,"file":"agent-example.workflow.d.ts","sourceRoot":"","sources":["../../../src/workflows/agent/agent-example.workflow.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,WAAW,EAAqB,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACtF,OAAO,EAAE,YAAY,EAAc,KAAK,eAAe,EAAY,MAAM,mBAAmB,CAAC;AAG7F,qBAKa,oBAAqB,SAAQ,YAAY;IACxC,OAAO,CAAC,QAAQ,CAAC,aAAa;gBAAb,aAAa,EAAE,aAAa;IAKnD,KAAK;IAiBL,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,KAAK,EAAE,eAAe,CAAC,WAAW,CAAC;CAMzF"}
+{"version":3,"file":"agent-example.workflow.d.ts","sourceRoot":"","sources":["../../../src/workflows/agent/agent-example.workflow.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,KAAK,WAAW,EAAqB,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACtF,OAAO,EAAE,YAAY,EAAc,KAAK,eAAe,EAAY,MAAM,mBAAmB,CAAC;AAG7F,qBAKa,oBAAqB,SAAQ,YAAY;IACxC,OAAO,CAAC,QAAQ,CAAC,aAAa;gBAAb,aAAa,EAAE,aAAa;IAKnD,KAAK;IAiBL,aAAa,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,KAAK,EAAE,eAAe,CAAC,WAAW,CAAC;CAOzF"}

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

@@ -32,6 +32,7 @@ let AgentExampleWorkflow = class AgentExampleWorkflow extends common_1.BaseWorkf
             role: 'assistant',
             text: input.data.response,
         });
+        this.setResult(input.data);
     }
 };
 exports.AgentExampleWorkflow = AgentExampleWorkflow;

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

@@ -1 +1 @@
-{"version":3,"file":"agent-example.workflow.js","sourceRoot":"","sources":["../../../src/workflows/agent/agent-example.workflow.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,yCAAiC;AACjC,4CAAsF;AACtF,8CAA6F;AAC7F,wEAAoE;AAO7D,IAAM,oBAAoB,GAA1B,MAAM,oBAAqB,SAAQ,qBAAY;IACvB;IAA7B,YAA6B,aAA4B;QACvD,KAAK,EAAE,CAAC;QADmB,kBAAa,GAAb,aAAa,CAAe;IAEzD,CAAC;IAGK,AAAN,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAC1B;YACE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAA,gBAAI,EAAC,SAAS,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;YAC9D,KAAK,EAAE,CAAC,gBAAgB,EAAE,YAAY,CAAC;YACvC,WAAW,EAAE,qDAAqD;SACnE,EACD,EAAE,QAAQ,EAAE,EAAE,UAAU,EAAE,eAAe,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,CACjF,CAAC;IACJ,CAAC;IAQK,AAAN,KAAK,CAAC,aAAa,CAAC,MAA+B,EAAE,KAAmC;QACtF,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,wCAAkB,EAAE;YAChD,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;SAC1B,CAAC,CAAC;IACL,CAAC;CACF,CAAA;AA7BY,oDAAoB;AAMzB;IADL,IAAA,mBAAU,EAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC;;;;iDAU7B;AAQK;IANL,IAAA,mBAAU,EAAC;QACV,IAAI,EAAE,SAAS;QACf,EAAE,EAAE,KAAK;QACT,IAAI,EAAE,IAAI;QACV,MAAM,EAAE,yBAAiB;KAC1B,CAAC;;;;yDAMD;+BA5BU,oBAAoB;IALhC,IAAA,iBAAQ,EAAC;QACR,KAAK,EAAE,6BAA6B;QACpC,WAAW,EACT,mHAAmH;KACtH,CAAC;qCAE4C,qBAAa;GAD9C,oBAAoB,CA6BhC"}
+{"version":3,"file":"agent-example.workflow.js","sourceRoot":"","sources":["../../../src/workflows/agent/agent-example.workflow.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,yCAAiC;AACjC,4CAAsF;AACtF,8CAA6F;AAC7F,wEAAoE;AAO7D,IAAM,oBAAoB,GAA1B,MAAM,oBAAqB,SAAQ,qBAAY;IACvB;IAA7B,YAA6B,aAA4B;QACvD,KAAK,EAAE,CAAC;QADmB,kBAAa,GAAb,aAAa,CAAe;IAEzD,CAAC;IAGK,AAAN,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAC1B;YACE,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAA,gBAAI,EAAC,SAAS,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;YAC9D,KAAK,EAAE,CAAC,gBAAgB,EAAE,YAAY,CAAC;YACvC,WAAW,EAAE,qDAAqD;SACnE,EACD,EAAE,QAAQ,EAAE,EAAE,UAAU,EAAE,eAAe,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,UAAU,EAAE,CACjF,CAAC;IACJ,CAAC;IAQK,AAAN,KAAK,CAAC,aAAa,CAAC,MAA+B,EAAE,KAAmC;QACtF,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,wCAAkB,EAAE;YAChD,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;SAC1B,CAAC,CAAC;QACH,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC;IAC7B,CAAC;CACF,CAAA;AA9BY,oDAAoB;AAMzB;IADL,IAAA,mBAAU,EAAC,EAAE,EAAE,EAAE,SAAS,EAAE,CAAC;;;;iDAU7B;AAQK;IANL,IAAA,mBAAU,EAAC;QACV,IAAI,EAAE,SAAS;QACf,EAAE,EAAE,KAAK;QACT,IAAI,EAAE,IAAI;QACV,MAAM,EAAE,yBAAiB;KAC1B,CAAC;;;;yDAOD;+BA7BU,oBAAoB;IALhC,IAAA,iBAAQ,EAAC;QACR,KAAK,EAAE,6BAA6B;QACpC,WAAW,EACT,mHAAmH;KACtH,CAAC;qCAE4C,qBAAa;GAD9C,oBAAoB,CA8BhC"}

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

@@ -5,8 +5,8 @@ export declare class CodeAgentExampleWorkflow extends BaseWorkflow {
     private readonly agentWorkflow;
     private readonly environments;
     constructor(agentWorkflow: AgentWorkflow, environments: EnvironmentService);
-    verifyEnvironment(_state: Record<string, unknown>): Promise<void>;
-    startExploration(_state: Record<string, unknown>): Promise<void>;
+    verifyEnvironment(): Promise<void>;
+    startExploration(): Promise<void>;
     exploreComplete(_state: Record<string, unknown>, input: TransitionInput<AgentResult>): Promise<void>;
 }
 //# sourceMappingURL=code-agent-example.workflow.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"code-agent-example.workflow.d.ts","sourceRoot":"","sources":["../../../src/workflows/code-agent/code-agent-example.workflow.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,WAAW,EAAqB,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACtF,OAAO,EAAE,YAAY,EAA+B,KAAK,eAAe,EAAY,MAAM,mBAAmB,CAAC;AAC9G,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAY9D,qBAIa,wBAAyB,SAAQ,YAAY;IAEtD,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,YAAY;gBADZ,aAAa,EAAE,aAAa,EAC5B,YAAY,EAAE,kBAAkB;IAM7C,iBAAiB,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAKjD,gBAAgB,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC;IAiBhD,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,KAAK,EAAE,eAAe,CAAC,WAAW,CAAC;CAM3F"}
+{"version":3,"file":"code-agent-example.workflow.d.ts","sourceRoot":"","sources":["../../../src/workflows/code-agent/code-agent-example.workflow.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,KAAK,WAAW,EAAqB,aAAa,EAAE,MAAM,kBAAkB,CAAC;AACtF,OAAO,EAAE,YAAY,EAA+B,KAAK,eAAe,EAAY,MAAM,mBAAmB,CAAC;AAC9G,OAAO,EAAE,kBAAkB,EAAE,MAAM,0BAA0B,CAAC;AAY9D,qBAIa,wBAAyB,SAAQ,YAAY;IAEtD,OAAO,CAAC,QAAQ,CAAC,aAAa;IAC9B,OAAO,CAAC,QAAQ,CAAC,YAAY;gBADZ,aAAa,EAAE,aAAa,EAC5B,YAAY,EAAE,kBAAkB;IAM7C,iBAAiB;IAKjB,gBAAgB;IAiBhB,eAAe,CAAC,MAAM,EAAE,MAAM,CAAC,MAAM,EAAE,OAAO,CAAC,EAAE,KAAK,EAAE,eAAe,CAAC,WAAW,CAAC;CAM3F"}

package/dist/workflows/code-agent/code-agent-example.workflow.js

@@ -23,10 +23,10 @@ let CodeAgentExampleWorkflow = class CodeAgentExampleWorkflow extends common_1.B
         this.agentWorkflow = agentWorkflow;
         this.environments = environments;
     }
-    async verifyEnvironment(_state) {
+    async verifyEnvironment() {
         await this.environments.assertReachable('sandbox');
     }
-    async startExploration(_state) {
+    async startExploration() {
         await this.agentWorkflow.run({
             system: 'You are a codebase exploration agent. Search and read source code to answer the question thoroughly.',
             tools: ['glob', 'grep', 'read'],
@@ -44,13 +44,13 @@ exports.CodeAgentExampleWorkflow = CodeAgentExampleWorkflow;
 __decorate([
     (0, common_1.Transition)({ to: 'verified' }),
     __metadata("design:type", Function),
-    __metadata("design:paramtypes", [Object]),
+    __metadata("design:paramtypes", []),
     __metadata("design:returntype", Promise)
 ], CodeAgentExampleWorkflow.prototype, "verifyEnvironment", null);
 __decorate([
     (0, common_1.Transition)({ from: 'verified', to: 'exploring' }),
     __metadata("design:type", Function),
-    __metadata("design:paramtypes", [Object]),
+    __metadata("design:paramtypes", []),
     __metadata("design:returntype", Promise)
 ], CodeAgentExampleWorkflow.prototype, "startExploration", null);
 __decorate([

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

@@ -1 +1 @@
-{"version":3,"file":"code-agent-example.workflow.js","sourceRoot":"","sources":["../../../src/workflows/code-agent/code-agent-example.workflow.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,4CAAsF;AACtF,8CAA8G;AAC9G,4DAA8D;AAE9D,MAAM,oBAAoB,GAAG;mEACsC,CAAC;AAa7D,IAAM,wBAAwB,GAA9B,MAAM,wBAAyB,SAAQ,qBAAY;IAErC;IACA;IAFnB,YACmB,aAA4B,EAC5B,YAAgC;QAEjD,KAAK,EAAE,CAAC;QAHS,kBAAa,GAAb,aAAa,CAAe;QAC5B,iBAAY,GAAZ,YAAY,CAAoB;IAGnD,CAAC;IAGK,AAAN,KAAK,CAAC,iBAAiB,CAAC,MAA+B;QACrD,MAAM,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;IACrD,CAAC;IAGK,AAAN,KAAK,CAAC,gBAAgB,CAAC,MAA+B;QACpD,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAC1B;YACE,MAAM,EAAE,sGAAsG;YAC9G,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;YAC/B,WAAW,EAAE,oBAAoB;SAClC,EACD,EAAE,QAAQ,EAAE,EAAE,UAAU,EAAE,iBAAiB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAChG,CAAC;IACJ,CAAC;IAQK,AAAN,KAAK,CAAC,eAAe,CAAC,MAA+B,EAAE,KAAmC;QACxF,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,wBAAe,EAAE;YAC7C,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;SAC1B,CAAC,CAAC;IACL,CAAC;CACF,CAAA;AArCY,4DAAwB;AAS7B;IADL,IAAA,mBAAU,EAAC,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC;;;;iEAG9B;AAGK;IADL,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,WAAW,EAAE,CAAC;;;;gEAUjD;AAQK;IANL,IAAA,mBAAU,EAAC;QACV,IAAI,EAAE,WAAW;QACjB,EAAE,EAAE,KAAK;QACT,IAAI,EAAE,IAAI;QACV,MAAM,EAAE,yBAAiB;KAC1B,CAAC;;;;+DAMD;mCApCU,wBAAwB;IAJpC,IAAA,iBAAQ,EAAC;QACR,KAAK,EAAE,4BAA4B;QACnC,WAAW,EAAE,yFAAyF;KACvG,CAAC;qCAGkC,qBAAa;QACd,kCAAkB;GAHxC,wBAAwB,CAqCpC"}
+{"version":3,"file":"code-agent-example.workflow.js","sourceRoot":"","sources":["../../../src/workflows/code-agent/code-agent-example.workflow.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,4CAAsF;AACtF,8CAA8G;AAC9G,4DAA8D;AAE9D,MAAM,oBAAoB,GAAG;mEACsC,CAAC;AAa7D,IAAM,wBAAwB,GAA9B,MAAM,wBAAyB,SAAQ,qBAAY;IAErC;IACA;IAFnB,YACmB,aAA4B,EAC5B,YAAgC;QAEjD,KAAK,EAAE,CAAC;QAHS,kBAAa,GAAb,aAAa,CAAe;QAC5B,iBAAY,GAAZ,YAAY,CAAoB;IAGnD,CAAC;IAGK,AAAN,KAAK,CAAC,iBAAiB;QACrB,MAAM,IAAI,CAAC,YAAY,CAAC,eAAe,CAAC,SAAS,CAAC,CAAC;IACrD,CAAC;IAGK,AAAN,KAAK,CAAC,gBAAgB;QACpB,MAAM,IAAI,CAAC,aAAa,CAAC,GAAG,CAC1B;YACE,MAAM,EAAE,sGAAsG;YAC9G,KAAK,EAAE,CAAC,MAAM,EAAE,MAAM,EAAE,MAAM,CAAC;YAC/B,WAAW,EAAE,oBAAoB;SAClC,EACD,EAAE,QAAQ,EAAE,EAAE,UAAU,EAAE,iBAAiB,EAAE,EAAE,IAAI,EAAE,QAAQ,EAAE,KAAK,EAAE,uBAAuB,EAAE,CAChG,CAAC;IACJ,CAAC;IAQK,AAAN,KAAK,CAAC,eAAe,CAAC,MAA+B,EAAE,KAAmC;QACxF,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,wBAAe,EAAE;YAC7C,IAAI,EAAE,WAAW;YACjB,IAAI,EAAE,KAAK,CAAC,IAAI,CAAC,QAAQ;SAC1B,CAAC,CAAC;IACL,CAAC;CACF,CAAA;AArCY,4DAAwB;AAS7B;IADL,IAAA,mBAAU,EAAC,EAAE,EAAE,EAAE,UAAU,EAAE,CAAC;;;;iEAG9B;AAGK;IADL,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,UAAU,EAAE,EAAE,EAAE,WAAW,EAAE,CAAC;;;;gEAUjD;AAQK;IANL,IAAA,mBAAU,EAAC;QACV,IAAI,EAAE,WAAW;QACjB,EAAE,EAAE,KAAK;QACT,IAAI,EAAE,IAAI;QACV,MAAM,EAAE,yBAAiB;KAC1B,CAAC;;;;+DAMD;mCApCU,wBAAwB;IAJpC,IAAA,iBAAQ,EAAC;QACR,KAAK,EAAE,4BAA4B;QACnC,WAAW,EAAE,yFAAyF;KACvG,CAAC;qCAGkC,qBAAa;QACd,kCAAkB;GAHxC,wBAAwB,CAqCpC"}

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

@@ -12,19 +12,19 @@ export declare class CustomAgentExampleWorkflow extends BaseWorkflow {
     private readonly llmDelegateToolCalls;
     private readonly llmUpdateToolResult;
     constructor(llmGenerateText: LlmGenerateTextTool, llmDelegateToolCalls: LlmDelegateToolCallsTool, llmUpdateToolResult: LlmUpdateToolResultTool);
-    setup(_state: CustomAgentState): Promise<void>;
+    setup(): Promise<void>;
     llmTurn(state: CustomAgentState): Promise<void>;
-    notifyOverBudget(_state: CustomAgentState): Promise<void>;
-    wrapUp(_state: CustomAgentState): Promise<void>;
+    notifyOverBudget(): Promise<void>;
+    wrapUp(): Promise<void>;
     executeToolCalls(state: CustomAgentState): Promise<void>;
     toolResultReceived(state: CustomAgentState, input: TransitionInput): Promise<void>;
-    toolsComplete(_state: CustomAgentState): void;
-    respond(_state: CustomAgentState): void;
+    toolsComplete(): void;
+    respond(): void;
     private hasBudget;
     private isOverBudget;
     private hasToolCalls;
     private allToolsComplete;
-    private isEndTurn;
+    private isDone;
 }
 export {};
 //# sourceMappingURL=custom-agent-example.workflow.d.ts.map

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

@@ -1 +1 @@
-{"version":3,"file":"custom-agent-example.workflow.d.ts","sourceRoot":"","sources":["../../../src/workflows/custom-agent/custom-agent-example.workflow.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAiD,MAAM,mBAAmB,CAAC;AAChG,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,KAAK,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,MAAM,gCAAgC,CAAC;AAC/F,OAAO,EACL,wBAAwB,EACxB,mBAAmB,EAEnB,uBAAuB,EACxB,MAAM,gCAAgC,CAAC;AAIxC,UAAU,gBAAgB;IACxB,SAAS,CAAC,EAAE,qBAAqB,CAAC;IAClC,cAAc,CAAC,EAAE,iBAAiB,CAAC;IACnC,SAAS,EAAE,MAAM,CAAC;CACnB;AAcD,qBAKa,0BAA2B,SAAQ,YAAY;IAExD,OAAO,CAAC,QAAQ,CAAC,eAAe;IAChC,OAAO,CAAC,QAAQ,CAAC,oBAAoB;IACrC,OAAO,CAAC,QAAQ,CAAC,mBAAmB;gBAFnB,eAAe,EAAE,mBAAmB,EACpC,oBAAoB,EAAE,wBAAwB,EAC9C,mBAAmB,EAAE,uBAAuB;IAUzD,KAAK,CAAC,MAAM,EAAE,gBAAgB;IA0B9B,OAAO,CAAC,KAAK,EAAE,gBAAgB;IA2B/B,gBAAgB,CAAC,MAAM,EAAE,gBAAgB;IAezC,MAAM,CAAC,MAAM,EAAE,gBAAgB;IAoB/B,gBAAgB,CAAC,KAAK,EAAE,gBAAgB;IAcxC,kBAAkB,CAAC,KAAK,EAAE,gBAAgB,EAAE,KAAK,EAAE,eAAe;IAcxE,aAAa,CAAC,MAAM,EAAE,gBAAgB;IAStC,OAAO,CAAC,MAAM,EAAE,gBAAgB;IAGhC,OAAO,CAAC,SAAS;IAKjB,OAAO,CAAC,YAAY;IAKpB,OAAO,CAAC,YAAY;IAKpB,OAAO,CAAC,gBAAgB;IAKxB,OAAO,CAAC,SAAS;CAGlB"}
+{"version":3,"file":"custom-agent-example.workflow.d.ts","sourceRoot":"","sources":["../../../src/workflows/custom-agent/custom-agent-example.workflow.ts"],"names":[],"mappings":"AACA,OAAO,EAAE,YAAY,EAAiD,MAAM,mBAAmB,CAAC;AAChG,OAAO,KAAK,EAAE,eAAe,EAAE,MAAM,mBAAmB,CAAC;AACzD,OAAO,KAAK,EAAE,iBAAiB,EAAE,qBAAqB,EAAE,MAAM,gCAAgC,CAAC;AAC/F,OAAO,EACL,wBAAwB,EACxB,mBAAmB,EAEnB,uBAAuB,EACxB,MAAM,gCAAgC,CAAC;AAIxC,UAAU,gBAAgB;IACxB,SAAS,CAAC,EAAE,qBAAqB,CAAC;IAClC,cAAc,CAAC,EAAE,iBAAiB,CAAC;IACnC,SAAS,EAAE,MAAM,CAAC;CACnB;AAcD,qBAKa,0BAA2B,SAAQ,YAAY;IAExD,OAAO,CAAC,QAAQ,CAAC,eAAe;IAChC,OAAO,CAAC,QAAQ,CAAC,oBAAoB;IACrC,OAAO,CAAC,QAAQ,CAAC,mBAAmB;gBAFnB,eAAe,EAAE,mBAAmB,EACpC,oBAAoB,EAAE,wBAAwB,EAC9C,mBAAmB,EAAE,uBAAuB;IAUzD,KAAK;IA0BL,OAAO,CAAC,KAAK,EAAE,gBAAgB;IA2B/B,gBAAgB;IAehB,MAAM;IAoBN,gBAAgB,CAAC,KAAK,EAAE,gBAAgB;IAcxC,kBAAkB,CAAC,KAAK,EAAE,gBAAgB,EAAE,KAAK,EAAE,eAAe;IAcxE,aAAa;IASb,OAAO;IAGP,OAAO,CAAC,SAAS;IAKjB,OAAO,CAAC,YAAY;IAKpB,OAAO,CAAC,YAAY;IAKpB,OAAO,CAAC,gBAAgB;IAKxB,OAAO,CAAC,MAAM;CAGf"}

package/dist/workflows/custom-agent/custom-agent-example.workflow.js

@@ -24,7 +24,7 @@ let CustomAgentExampleWorkflow = class CustomAgentExampleWorkflow extends common
         this.llmDelegateToolCalls = llmDelegateToolCalls;
         this.llmUpdateToolResult = llmUpdateToolResult;
     }
-    async setup(_state) {
+    async setup() {
         await this.documentStore.save(common_1.MarkdownDocument, {
             markdown: '# Custom Agent Example\n\n' +
                 `This agent has a budget of **${MAX_TURNS} tool-using turns**. ` +
@@ -51,14 +51,14 @@ let CustomAgentExampleWorkflow = class CustomAgentExampleWorkflow extends common
             llmResult: result.data,
         });
     }
-    async notifyOverBudget(_state) {
+    async notifyOverBudget() {
         await this.documentStore.save(llm_provider_module_1.LlmMessageDocument, {
             role: 'user',
             text: `You have used your full turn budget of ${MAX_TURNS} tool-using turns and can no longer call tools. ` +
                 'Summarise what you found so far in plain text and answer based on that.',
         });
     }
-    async wrapUp(_state) {
+    async wrapUp() {
         await this.llmGenerateText.call({}, {
             config: {
                 provider: 'claude',
@@ -81,8 +81,8 @@ let CustomAgentExampleWorkflow = class CustomAgentExampleWorkflow extends common
         });
         this.assignState({ delegateResult: result.data });
     }
-    toolsComplete(_state) { }
-    respond(_state) { }
+    toolsComplete() { }
+    respond() { }
     hasBudget(state) {
         return state.turnCount < MAX_TURNS;
     }
@@ -95,7 +95,7 @@ let CustomAgentExampleWorkflow = class CustomAgentExampleWorkflow extends common
     allToolsComplete(state) {
         return !!state.delegateResult?.allCompleted;
     }
-    isEndTurn(state) {
+    isDone(state) {
         return state.llmResult?.message.stopReason === 'end_turn';
     }
 };
@@ -103,7 +103,7 @@ exports.CustomAgentExampleWorkflow = CustomAgentExampleWorkflow;
 __decorate([
     (0, common_1.Transition)({ to: 'ready' }),
     __metadata("design:type", Function),
-    __metadata("design:paramtypes", [Object]),
+    __metadata("design:paramtypes", []),
     __metadata("design:returntype", Promise)
 ], CustomAgentExampleWorkflow.prototype, "setup", null);
 __decorate([
@@ -117,13 +117,13 @@ __decorate([
     (0, common_1.Transition)({ from: 'ready', to: 'over_budget' }),
     (0, common_1.Guard)('isOverBudget'),
     __metadata("design:type", Function),
-    __metadata("design:paramtypes", [Object]),
+    __metadata("design:paramtypes", []),
     __metadata("design:returntype", Promise)
 ], CustomAgentExampleWorkflow.prototype, "notifyOverBudget", null);
 __decorate([
     (0, common_1.Transition)({ from: 'over_budget', to: 'end' }),
     __metadata("design:type", Function),
-    __metadata("design:paramtypes", [Object]),
+    __metadata("design:paramtypes", []),
     __metadata("design:returntype", Promise)
 ], CustomAgentExampleWorkflow.prototype, "wrapUp", null);
 __decorate([
@@ -143,14 +143,14 @@ __decorate([
     (0, common_1.Transition)({ from: 'awaiting_tools', to: 'ready' }),
     (0, common_1.Guard)('allToolsComplete'),
     __metadata("design:type", Function),
-    __metadata("design:paramtypes", [Object]),
+    __metadata("design:paramtypes", []),
     __metadata("design:returntype", void 0)
 ], CustomAgentExampleWorkflow.prototype, "toolsComplete", null);
 __decorate([
     (0, common_1.Transition)({ from: 'prompt_executed', to: 'end' }),
-    (0, common_1.Guard)('isEndTurn'),
+    (0, common_1.Guard)('isDone'),
     __metadata("design:type", Function),
-    __metadata("design:paramtypes", [Object]),
+    __metadata("design:paramtypes", []),
     __metadata("design:returntype", void 0)
 ], CustomAgentExampleWorkflow.prototype, "respond", null);
 exports.CustomAgentExampleWorkflow = CustomAgentExampleWorkflow = __decorate([

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

@@ -1 +1 @@
-{"version":3,"file":"custom-agent-example.workflow.js","sourceRoot":"","sources":["../../../src/workflows/custom-agent/custom-agent-example.workflow.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,yCAAiC;AACjC,8CAAgG;AAGhG,wEAKwC;AAExC,MAAM,SAAS,GAAG,CAAC,CAAC;AAyBb,IAAM,0BAA0B,GAAhC,MAAM,0BAA2B,SAAQ,qBAAY;IAEvC;IACA;IACA;IAHnB,YACmB,eAAoC,EACpC,oBAA8C,EAC9C,mBAA4C;QAE7D,KAAK,EAAE,CAAC;QAJS,oBAAe,GAAf,eAAe,CAAqB;QACpC,yBAAoB,GAApB,oBAAoB,CAA0B;QAC9C,wBAAmB,GAAnB,mBAAmB,CAAyB;IAG/D,CAAC;IAOK,AAAN,KAAK,CAAC,KAAK,CAAC,MAAwB;QAClC,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,yBAAgB,EAAE;YAC9C,QAAQ,EACN,4BAA4B;gBAC5B,gCAAgC,SAAS,uBAAuB;gBAChE,2FAA2F;SAC9F,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,wCAAkB,EAAE;YAChD,IAAI,EAAE,MAAM;YACZ,IAAI,EACF,2FAA2F;gBAC3F,8EAA8E;gBAC9E,+EAA+E;SAClF,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;IACrC,CAAC;IASK,AAAN,KAAK,CAAC,OAAO,CAAC,KAAuB;QACnC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAC5C,EAAE,EACF;YACE,MAAM,EAAE;gBACN,QAAQ,EAAE,QAAQ;gBAClB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAA,gBAAI,EAAC,SAAS,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;gBAC9D,KAAK,EAAE,CAAC,gBAAgB,EAAE,YAAY,CAAC;aACxC;SACF,CACF,CAAC;QAEF,IAAI,CAAC,WAAW,CAAC;YACf,SAAS,EAAE,KAAK,CAAC,SAAS,GAAG,CAAC;YAC9B,SAAS,EAAE,MAAM,CAAC,IAAI;SACvB,CAAC,CAAC;IACL,CAAC;IAWK,AAAN,KAAK,CAAC,gBAAgB,CAAC,MAAwB;QAC7C,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,wCAAkB,EAAE;YAChD,IAAI,EAAE,MAAM;YACZ,IAAI,EACF,0CAA0C,SAAS,kDAAkD;gBACrG,yEAAyE;SAC5E,CAAC,CAAC;IACL,CAAC;IAQK,AAAN,KAAK,CAAC,MAAM,CAAC,MAAwB;QACnC,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAC7B,EAAE,EACF;YACE,MAAM,EAAE;gBACN,QAAQ,EAAE,QAAQ;gBAClB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAA,gBAAI,EAAC,SAAS,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;gBAC/D,KAAK,EAAE,EAAE;aACV;SACF,CACF,CAAC;IACJ,CAAC;IASK,AAAN,KAAK,CAAC,gBAAgB,CAAC,KAAuB;QAC5C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC;YAClD,OAAO,EAAE,KAAK,CAAC,SAAU,CAAC,OAAO;YACjC,QAAQ,EAAE,EAAE,UAAU,EAAE,oBAAoB,EAAE;SAC/C,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC;IAQK,AAAN,KAAK,CAAC,kBAAkB,CAAC,KAAuB,EAAE,KAAsB;QACtE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;YACjD,cAAc,EAAE,KAAK,CAAC,cAAe;YACrC,aAAa,EAAE,KAAK;SACrB,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC;IAQD,aAAa,CAAC,MAAwB,IAAG,CAAC;IAS1C,OAAO,CAAC,MAAwB,IAAG,CAAC;IAG5B,SAAS,CAAC,KAAuB;QACvC,OAAO,KAAK,CAAC,SAAS,GAAG,SAAS,CAAC;IACrC,CAAC;IAGO,YAAY,CAAC,KAAuB;QAC1C,OAAO,KAAK,CAAC,SAAS,IAAI,SAAS,CAAC;IACtC,CAAC;IAGO,YAAY,CAAC,KAAuB;QAC1C,OAAO,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,UAAU,KAAK,UAAU,CAAC;IAC5D,CAAC;IAGO,gBAAgB,CAAC,KAAuB;QAC9C,OAAO,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,YAAY,CAAC;IAC9C,CAAC;IAGO,SAAS,CAAC,KAAuB;QACvC,OAAO,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,UAAU,KAAK,UAAU,CAAC;IAC5D,CAAC;CACF,CAAA;AArKY,gEAA0B;AAc/B;IADL,IAAA,mBAAU,EAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC;;;;uDAkB3B;AASK;IAFL,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,iBAAiB,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IAClE,IAAA,cAAK,EAAC,WAAW,CAAC;;;;yDAiBlB;AAWK;IAFL,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC;IAChD,IAAA,cAAK,EAAC,cAAc,CAAC;;;;kEAQrB;AAQK;IADL,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;;;;wDAY9C;AASK;IAFL,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE,EAAE,gBAAgB,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IAC3E,IAAA,cAAK,EAAC,cAAc,CAAC;;;;kEAOrB;AAQK;IADL,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;;;;oEAOxE;AAQD;IAFC,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC;IACnD,IAAA,cAAK,EAAC,kBAAkB,CAAC;;;;+DACgB;AAS1C;IAFC,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;IAClD,IAAA,cAAK,EAAC,WAAW,CAAC;;;;yDACiB;qCA3IzB,0BAA0B;IALtC,IAAA,iBAAQ,EAAC;QACR,KAAK,EAAE,8BAA8B;QACrC,WAAW,EACT,uPAAuP;KAC1P,CAAC;qCAGoC,yCAAmB;QACd,8CAAwB;QACzB,6CAAuB;GAJpD,0BAA0B,CAqKtC"}
+{"version":3,"file":"custom-agent-example.workflow.js","sourceRoot":"","sources":["../../../src/workflows/custom-agent/custom-agent-example.workflow.ts"],"names":[],"mappings":";;;;;;;;;;;;AAAA,yCAAiC;AACjC,8CAAgG;AAGhG,wEAKwC;AAExC,MAAM,SAAS,GAAG,CAAC,CAAC;AAyBb,IAAM,0BAA0B,GAAhC,MAAM,0BAA2B,SAAQ,qBAAY;IAEvC;IACA;IACA;IAHnB,YACmB,eAAoC,EACpC,oBAA8C,EAC9C,mBAA4C;QAE7D,KAAK,EAAE,CAAC;QAJS,oBAAe,GAAf,eAAe,CAAqB;QACpC,yBAAoB,GAApB,oBAAoB,CAA0B;QAC9C,wBAAmB,GAAnB,mBAAmB,CAAyB;IAG/D,CAAC;IAOK,AAAN,KAAK,CAAC,KAAK;QACT,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,yBAAgB,EAAE;YAC9C,QAAQ,EACN,4BAA4B;gBAC5B,gCAAgC,SAAS,uBAAuB;gBAChE,2FAA2F;SAC9F,CAAC,CAAC;QAEH,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,wCAAkB,EAAE;YAChD,IAAI,EAAE,MAAM;YACZ,IAAI,EACF,2FAA2F;gBAC3F,8EAA8E;gBAC9E,+EAA+E;SAClF,CAAC,CAAC;QAEH,IAAI,CAAC,WAAW,CAAC,EAAE,SAAS,EAAE,CAAC,EAAE,CAAC,CAAC;IACrC,CAAC;IASK,AAAN,KAAK,CAAC,OAAO,CAAC,KAAuB;QACnC,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAC5C,EAAE,EACF;YACE,MAAM,EAAE;gBACN,QAAQ,EAAE,QAAQ;gBAClB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAA,gBAAI,EAAC,SAAS,EAAE,WAAW,EAAE,WAAW,CAAC,CAAC;gBAC9D,KAAK,EAAE,CAAC,gBAAgB,EAAE,YAAY,CAAC;aACxC;SACF,CACF,CAAC;QAEF,IAAI,CAAC,WAAW,CAAC;YACf,SAAS,EAAE,KAAK,CAAC,SAAS,GAAG,CAAC;YAC9B,SAAS,EAAE,MAAM,CAAC,IAAI;SACvB,CAAC,CAAC;IACL,CAAC;IAWK,AAAN,KAAK,CAAC,gBAAgB;QACpB,MAAM,IAAI,CAAC,aAAa,CAAC,IAAI,CAAC,wCAAkB,EAAE;YAChD,IAAI,EAAE,MAAM;YACZ,IAAI,EACF,0CAA0C,SAAS,kDAAkD;gBACrG,yEAAyE;SAC5E,CAAC,CAAC;IACL,CAAC;IAQK,AAAN,KAAK,CAAC,MAAM;QACV,MAAM,IAAI,CAAC,eAAe,CAAC,IAAI,CAC7B,EAAE,EACF;YACE,MAAM,EAAE;gBACN,QAAQ,EAAE,QAAQ;gBAClB,MAAM,EAAE,IAAI,CAAC,MAAM,CAAC,IAAA,gBAAI,EAAC,SAAS,EAAE,WAAW,EAAE,YAAY,CAAC,CAAC;gBAC/D,KAAK,EAAE,EAAE;aACV;SACF,CACF,CAAC;IACJ,CAAC;IASK,AAAN,KAAK,CAAC,gBAAgB,CAAC,KAAuB;QAC5C,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,oBAAoB,CAAC,IAAI,CAAC;YAClD,OAAO,EAAE,KAAK,CAAC,SAAU,CAAC,OAAO;YACjC,QAAQ,EAAE,EAAE,UAAU,EAAE,oBAAoB,EAAE;SAC/C,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC;IAQK,AAAN,KAAK,CAAC,kBAAkB,CAAC,KAAuB,EAAE,KAAsB;QACtE,MAAM,MAAM,GAAG,MAAM,IAAI,CAAC,mBAAmB,CAAC,IAAI,CAAC;YACjD,cAAc,EAAE,KAAK,CAAC,cAAe;YACrC,aAAa,EAAE,KAAK;SACrB,CAAC,CAAC;QACH,IAAI,CAAC,WAAW,CAAC,EAAE,cAAc,EAAE,MAAM,CAAC,IAAI,EAAE,CAAC,CAAC;IACpD,CAAC;IAQD,aAAa,KAAI,CAAC;IASlB,OAAO,KAAI,CAAC;IAGJ,SAAS,CAAC,KAAuB;QACvC,OAAO,KAAK,CAAC,SAAS,GAAG,SAAS,CAAC;IACrC,CAAC;IAGO,YAAY,CAAC,KAAuB;QAC1C,OAAO,KAAK,CAAC,SAAS,IAAI,SAAS,CAAC;IACtC,CAAC;IAGO,YAAY,CAAC,KAAuB;QAC1C,OAAO,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,UAAU,KAAK,UAAU,CAAC;IAC5D,CAAC;IAGO,gBAAgB,CAAC,KAAuB;QAC9C,OAAO,CAAC,CAAC,KAAK,CAAC,cAAc,EAAE,YAAY,CAAC;IAC9C,CAAC;IAGO,MAAM,CAAC,KAAuB;QACpC,OAAO,KAAK,CAAC,SAAS,EAAE,OAAO,CAAC,UAAU,KAAK,UAAU,CAAC;IAC5D,CAAC;CACF,CAAA;AArKY,gEAA0B;AAc/B;IADL,IAAA,mBAAU,EAAC,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC;;;;uDAkB3B;AASK;IAFL,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,iBAAiB,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IAClE,IAAA,cAAK,EAAC,WAAW,CAAC;;;;yDAiBlB;AAWK;IAFL,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,OAAO,EAAE,EAAE,EAAE,aAAa,EAAE,CAAC;IAChD,IAAA,cAAK,EAAC,cAAc,CAAC;;;;kEAQrB;AAQK;IADL,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,aAAa,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;;;;wDAY9C;AASK;IAFL,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE,EAAE,gBAAgB,EAAE,QAAQ,EAAE,EAAE,EAAE,CAAC;IAC3E,IAAA,cAAK,EAAC,cAAc,CAAC;;;;kEAOrB;AAQK;IADL,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE,EAAE,gBAAgB,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC;;;;oEAOxE;AAQD;IAFC,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,EAAE,EAAE,OAAO,EAAE,CAAC;IACnD,IAAA,cAAK,EAAC,kBAAkB,CAAC;;;;+DACR;AASlB;IAFC,IAAA,mBAAU,EAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,EAAE,EAAE,KAAK,EAAE,CAAC;IAClD,IAAA,cAAK,EAAC,QAAQ,CAAC;;;;yDACJ;qCA3ID,0BAA0B;IALtC,IAAA,iBAAQ,EAAC;QACR,KAAK,EAAE,8BAA8B;QACrC,WAAW,EACT,uPAAuP;KAC1P,CAAC;qCAGoC,yCAAmB;QACd,8CAAwB;QACzB,6CAAuB;GAJpD,0BAA0B,CAqKtC"}

package/package.json

@@ -11,7 +11,7 @@
     "mcp",
     "workflow"
   ],
-  "version": "0.1.1",
+  "version": "0.1.3",
   "license": "MIT",
   "author": {
     "name": "Jakob Klippel",
@@ -39,7 +39,7 @@
     "@loopstack/llm-provider-module": "*",
     "@loopstack/mcp-module": "*",
     "@loopstack/remote-client": "*",
-    "@nestjs/common": "^11.1.19",
+    "@nestjs/common": "^12.0.1",
     "zod": "^4.3.6"
   },
   "files": [

package/src/tools/calculator.tool.ts

@@ -4,6 +4,8 @@ import type { RunContext } from '@loopstack/common';
 
 export type CalculatorResult = string;
 
+export const CalculatorResultSchema = z.string();
+
 @Tool({
   name: 'calculator',
   description: 'Perform a basic arithmetic calculation. Supports add, subtract, multiply, divide.',
@@ -12,6 +14,8 @@ export type CalculatorResult = string;
     a: z.number().describe('First operand.'),
     b: z.number().describe('Second operand.'),
   }),
+  resultSchema: CalculatorResultSchema,
+  effects: 'none',
 })
 export class CalculatorTool extends BaseTool<{ operation: string; a: number; b: number }, object, CalculatorResult> {
   protected async handle(

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

@@ -3,12 +3,16 @@ import { BaseTool, Tool, ToolEnvelope } from '@loopstack/common';
 
 export type WeatherLookupResult = string;
 
+export const WeatherLookupResultSchema = z.string();
+
 @Tool({
   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.'),
   }),
+  resultSchema: WeatherLookupResultSchema,
+  effects: 'none',
 })
 export class WeatherLookupTool extends BaseTool<{ city: string }, object, WeatherLookupResult> {
   protected async handle(args: { city: string }): Promise<ToolEnvelope<WeatherLookupResult>> {

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

@@ -0,0 +1,60 @@
+import { describe, expect, it } from 'vitest';
+import { AgentModule } from '@loopstack/agent';
+import { LlmGenerateTextTool, LlmProviderModule } from '@loopstack/llm-provider-module';
+import { replay, runWorkflow } from '@loopstack/testing';
+import { CalculatorTool } from '../../../tools/calculator.tool';
+import { WeatherLookupTool } from '../../../tools/weather-lookup.tool';
+import { AgentExampleWorkflow } from '../agent-example.workflow';
+
+/**
+ * The canonical agent test: replay only the LLM (`replayTools: [LlmGenerateTextTool]`) so the
+ * model turns are scripted, while the delegation machinery and the real `weather_lookup` /
+ * `calculator` tools run live. The scripted turns drive one tool-use round then a final answer.
+ */
+const llmTurn = (message: { id: string; role: string; text: string; blocks: unknown[]; stopReason: string }) => ({
+  tool: 'llm_generate_text',
+  envelope: {
+    data: { message, response: {} },
+    metadata: { provider: 'claude', model: 'claude-sonnet-4-6' },
+    documents: [{ documentName: 'llm_message', content: message, options: { meta: { provider: 'claude' } } }],
+  },
+});
+
+const TOOL_TURN = llmTurn({
+  id: 'msg_1',
+  role: 'assistant',
+  text: '',
+  blocks: [
+    { type: 'tool_call', id: 'w1', name: 'weather_lookup', args: { city: 'Tokyo' } },
+    { type: 'tool_call', id: 'c1', name: 'calculator', args: { operation: 'multiply', a: 42, b: 17 } },
+  ],
+  stopReason: 'tool_use',
+});
+
+const FINAL_TURN = llmTurn({
+  id: 'msg_2',
+  role: 'assistant',
+  text: 'Tokyo is sunny, and 42 * 17 = 714.',
+  blocks: [{ type: 'text', text: 'Tokyo is sunny, and 42 * 17 = 714.' }],
+  stopReason: 'end_turn',
+});
+
+describe('AgentExampleWorkflow', () => {
+  it('replays the LLM while the real tools run in the live delegation', async () => {
+    const run = await runWorkflow(AgentExampleWorkflow, undefined, {
+      imports: [LlmProviderModule, AgentModule],
+      providers: [WeatherLookupTool, CalculatorTool],
+      replayTools: [LlmGenerateTextTool],
+      replay: replay({ version: 3, recordings: [TOOL_TURN, FINAL_TURN] }),
+    });
+
+    expect(run.error).toBeUndefined();
+    expect(run.status).toBe('completed');
+    expect(run.path).toEqual(['start', 'agentComplete']);
+    expect(run.result).toEqual({ response: 'Tokyo is sunny, and 42 * 17 = 714.' });
+
+    // The agent sub-workflow ran the loop to completion (outcome, not internals).
+    expect(run.children[0].workflowName).toBe('agent');
+    expect(run.children[0].status).toBe('completed');
+  });
+});

package/src/workflows/agent/agent-example.workflow.ts

@@ -36,5 +36,6 @@ export class AgentExampleWorkflow extends BaseWorkflow {
       role: 'assistant',
       text: input.data.response,
     });
+    this.setResult(input.data);
   }
 }

package/src/workflows/code-agent/code-agent-example.workflow.ts

@@ -25,12 +25,12 @@ export class CodeAgentExampleWorkflow extends BaseWorkflow {
   }
 
   @Transition({ to: 'verified' })
-  async verifyEnvironment(_state: Record<string, unknown>) {
+  async verifyEnvironment() {
     await this.environments.assertReachable('sandbox');
   }
 
   @Transition({ from: 'verified', to: 'exploring' })
-  async startExploration(_state: Record<string, unknown>) {
+  async startExploration() {
     await this.agentWorkflow.run(
       {
         system: 'You are a codebase exploration agent. Search and read source code to answer the question thoroughly.',

package/src/workflows/custom-agent/__tests__/custom-agent-example.spec.ts

@@ -0,0 +1,69 @@
+import { describe, expect, it } from 'vitest';
+import { LlmGenerateTextTool, LlmProviderModule } from '@loopstack/llm-provider-module';
+import { replay, runWorkflow } from '@loopstack/testing';
+import { CalculatorTool } from '../../../tools/calculator.tool';
+import { WeatherLookupTool } from '../../../tools/weather-lookup.tool';
+import { CustomAgentExampleWorkflow } from '../custom-agent-example.workflow';
+
+/**
+ * A hand-rolled agent loop with a deterministic turn budget. Scripting the LLM turns lets a
+ * test drive the loop *past* its budget and assert the control flow the generic AgentWorkflow
+ * can't express: after MAX_TURNS tool-using turns the workflow forces a no-tools wrap-up turn.
+ * The LLM is replayed; the delegation and the real weather tool run live.
+ */
+const toolUse = (id: string, city: string) => ({
+  tool: 'llm_generate_text',
+  envelope: {
+    data: {
+      message: {
+        id,
+        role: 'assistant',
+        text: '',
+        blocks: [{ type: 'tool_call', id: `w_${id}`, name: 'weather_lookup', args: { city } }],
+        stopReason: 'tool_use',
+      },
+      response: {},
+    },
+  },
+});
+
+const wrapUp = () => ({
+  tool: 'llm_generate_text',
+  envelope: {
+    data: {
+      message: {
+        id: 'final',
+        role: 'assistant',
+        text: 'Budget reached — none of the cities I checked were warm enough.',
+        blocks: [{ type: 'text', text: 'Budget reached — none of the cities I checked were warm enough.' }],
+        stopReason: 'end_turn',
+      },
+      response: {},
+    },
+  },
+});
+
+describe('CustomAgentExampleWorkflow', () => {
+  it('runs two tool-using turns then forces a no-tools wrap-up when the budget is spent', async () => {
+    const run = await runWorkflow(CustomAgentExampleWorkflow, undefined, {
+      imports: [LlmProviderModule],
+      providers: [WeatherLookupTool, CalculatorTool],
+      replayTools: [LlmGenerateTextTool],
+      // Two tool-using turns exhaust the 2-turn budget, then the wrap-up turn (no tools).
+      replay: replay({ version: 3, recordings: [toolUse('t1', 'London'), toolUse('t2', 'New York'), wrapUp()] }),
+    });
+
+    expect(run.error).toBeUndefined();
+    expect(run.status).toBe('completed');
+    expect(run.place).toBe('end');
+
+    // The budget guard fired: exactly two tool-using turns, then the forced wrap-up path.
+    expect(run.path.filter((t) => t === 'llmTurn')).toHaveLength(2);
+    expect(run.path).toContain('notifyOverBudget');
+    expect(run.path).toContain('wrapUp');
+
+    // The over-budget notice was shown to the model before the wrap-up turn.
+    const texts = run.documents.map((d) => (d.content as { text?: string }).text ?? '');
+    expect(texts.some((t) => t.includes('used your full turn budget'))).toBe(true);
+  });
+});

package/src/workflows/custom-agent/custom-agent-example.workflow.ts

@@ -48,7 +48,7 @@ export class CustomAgentExampleWorkflow extends BaseWorkflow {
    * conversation with the user's request, and initialises the turn counter.
    */
   @Transition({ to: 'ready' })
-  async setup(_state: CustomAgentState) {
+  async setup() {
     await this.documentStore.save(MarkdownDocument, {
       markdown:
         '# Custom Agent Example\n\n' +
@@ -101,7 +101,7 @@ export class CustomAgentExampleWorkflow extends BaseWorkflow {
    */
   @Transition({ from: 'ready', to: 'over_budget' })
   @Guard('isOverBudget')
-  async notifyOverBudget(_state: CustomAgentState) {
+  async notifyOverBudget() {
     await this.documentStore.save(LlmMessageDocument, {
       role: 'user',
       text:
@@ -116,7 +116,7 @@ export class CustomAgentExampleWorkflow extends BaseWorkflow {
    * summary based on whatever results were gathered before the budget hit.
    */
   @Transition({ from: 'over_budget', to: 'end' })
-  async wrapUp(_state: CustomAgentState) {
+  async wrapUp() {
     await this.llmGenerateText.call(
       {},
       {
@@ -164,7 +164,7 @@ export class CustomAgentExampleWorkflow extends BaseWorkflow {
    */
   @Transition({ from: 'awaiting_tools', to: 'ready' })
   @Guard('allToolsComplete')
-  toolsComplete(_state: CustomAgentState) {}
+  toolsComplete() {}
 
   /**
    * The LLM emitted `end_turn` instead of more tool calls — it considers the
@@ -172,8 +172,8 @@ export class CustomAgentExampleWorkflow extends BaseWorkflow {
    * just close the workflow.
    */
   @Transition({ from: 'prompt_executed', to: 'end' })
-  @Guard('isEndTurn')
-  respond(_state: CustomAgentState) {}
+  @Guard('isDone')
+  respond() {}
 
   /** True while the model is still allowed to take another tool-using turn. */
   private hasBudget(state: CustomAgentState): boolean {
@@ -196,7 +196,7 @@ export class CustomAgentExampleWorkflow extends BaseWorkflow {
   }
 
   /** True when the last LLM response was a final plain-text answer (no tools). */
-  private isEndTurn(state: CustomAgentState): boolean {
+  private isDone(state: CustomAgentState): boolean {
     return state.llmResult?.message.stopReason === 'end_turn';
   }
 }