@lowire/loop 0.0.10 → 0.0.12

package/lib/loop.d.ts

@@ -19,12 +19,12 @@ export type LoopEvents = {
     onBeforeTurn?: (params: {
         conversation: types.Conversation;
         totalUsage: types.Usage;
-        budgetTokens: number;
+        budgetTokens?: number;
     }) => PromiseOrValue<'continue' | 'break' | void>;
     onAfterTurn?: (params: {
         assistantMessage: types.AssistantMessage;
         totalUsage: types.Usage;
-        budgetTokens: number;
+        budgetTokens?: number;
     }) => PromiseOrValue<'continue' | 'break' | void>;
     onBeforeToolCall?: (params: {
         assistantMessage: types.AssistantMessage;
@@ -45,7 +45,6 @@ export type LoopOptions = types.CompletionOptions & LoopEvents & {
     tools?: types.Tool[];
     callTool?: types.ToolCallback;
     maxTurns?: number;
-    resultSchema?: types.Schema;
     cache?: {
         messages: types.ReplayCache;
         secrets: Record<string, string>;
@@ -57,10 +56,10 @@ export declare class Loop {
     private _loopOptions;
     private _cacheOutput;
     constructor(loopName: 'openai' | 'github' | 'anthropic' | 'google', options: LoopOptions);
-    run<T>(task: string, runOptions?: Omit<LoopOptions, 'model'> & {
+    run(task: string, runOptions?: Omit<LoopOptions, 'model'> & {
         model?: string;
     }): Promise<{
-        result?: T;
+        result?: types.ToolResult;
         status: 'ok' | 'break';
         usage: types.Usage;
         turns: number;

package/lib/loop.js

@@ -29,12 +29,7 @@ class Loop {
     }
     async run(task, runOptions = {}) {
         const options = { ...this._loopOptions, ...runOptions };
-        const allTools = [...options.tools || []];
-        allTools.push({
-            name: 'report_result',
-            description: 'Report the result of the task.',
-            inputSchema: options.resultSchema ?? defaultResultSchema,
-        });
+        const allTools = [...(options.tools || []).map(wrapToolWithIsDone)];
         const conversation = {
             systemPrompt,
             messages: [
@@ -43,12 +38,12 @@ class Loop {
             tools: allTools,
         };
         const debug = options.debug;
-        let budgetTokens = options.maxTokens ?? 100_000;
+        let budgetTokens = options.maxTokens;
         const totalUsage = { input: 0, output: 0 };
         debug?.('lowire:loop')(`Starting ${this._provider.name} loop`, task);
         const maxTurns = options.maxTurns || 100;
         for (let turns = 0; turns < maxTurns; ++turns) {
-            if (budgetTokens <= 0)
+            if (options.maxTokens && budgetTokens !== undefined && budgetTokens <= 0)
                 throw new Error(`Budget tokens ${options.maxTokens} exhausted`);
             debug?.('lowire:loop')(`Turn ${turns + 1} of (max ${maxTurns})`);
             const caches = options.cache ? {
@@ -68,7 +63,8 @@ class Loop {
             const intent = assistantMessage.content.filter(part => part.type === 'text').map(part => part.text).join('\n');
             totalUsage.input += usage.input;
             totalUsage.output += usage.output;
-            budgetTokens -= usage.input + usage.output;
+            if (budgetTokens !== undefined)
+                budgetTokens -= usage.input + usage.output;
             debug?.('lowire:loop')('Usage', `input: ${usage.input}, output: ${usage.output}`);
             debug?.('lowire:loop')('Assistant', intent, JSON.stringify(assistantMessage.content, null, 2));
             const afterStatus = await options.onAfterTurn?.({ assistantMessage, totalUsage, budgetTokens });
@@ -83,8 +79,6 @@ class Loop {
             for (const toolCall of toolCalls) {
                 const { name, arguments: args } = toolCall;
                 debug?.('lowire:loop')('Call tool', name, JSON.stringify(args, null, 2));
-                if (name === 'report_result')
-                    return { result: args, status: 'ok', usage: totalUsage, turns };
                 const status = await options.onBeforeToolCall?.({ assistantMessage, toolCall });
                 if (status === 'break')
                     return { status: 'break', usage: totalUsage, turns };
@@ -120,6 +114,8 @@ class Loop {
                         continue;
                     }
                     toolCall.result = result;
+                    if (args._is_done)
+                        return { result, status: 'ok', usage: totalUsage, turns };
                 }
                 catch (error) {
                     const errorMessage = `Error while executing tool "${name}": ${error instanceof Error ? error.message : String(error)}\n\nPlease try to recover and complete the task.`;
@@ -152,15 +148,18 @@ class Loop {
     }
 }
 exports.Loop = Loop;
-const defaultResultSchema = {
-    type: 'object',
-    properties: {
-        result: {
-            type: 'string',
-        },
-    },
-    required: ['result'],
-};
+function wrapToolWithIsDone(tool) {
+    const inputSchema = { ...tool.inputSchema };
+    inputSchema.properties = {
+        ...inputSchema.properties,
+        _is_done: { type: 'boolean', description: 'Whether the task is complete. If false, agentic loop will continue to perform the task.' },
+    };
+    inputSchema.required = [...(inputSchema.required || []), '_is_done'];
+    return {
+        ...tool,
+        inputSchema,
+    };
+}
 const systemPrompt = `
 You are an autonomous agent designed to complete tasks by interacting with tools. Perform the user task.
 `;

package/lib/providers/anthropic.js

@@ -19,7 +19,7 @@ exports.Anthropic = void 0;
 class Anthropic {
     name = 'anthropic';
     async complete(conversation, options) {
-        const maxTokens = Math.min(options.maxTokens ?? 32768, 32768);
+        const maxTokens = Math.min(options.maxTokens ?? 32_768, 32_768);
         const response = await create({
             model: options.model,
             max_tokens: maxTokens,

package/lib/types.d.ts

@@ -15,7 +15,7 @@
  */
 export type Schema = {
     type: 'object';
-    properties?: unknown | null;
+    properties?: Record<string, unknown> | null;
     required?: Array<string> | null;
 };
 export type Tool = {

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@lowire/loop",
-  "version": "0.0.10",
+  "version": "0.0.12",
   "description": "Small agentic loop",
   "repository": {
     "type": "git",