@lowire/loop 0.0.19 → 0.0.20

package/lib/cache.js

@@ -36,7 +36,7 @@ async function cachedCompleteNoSecrets(provider, conversation, caches, options)
     if (!process.env.LOWIRE_NO_CACHE && caches.output[key])
         return caches.output[key];
     if (process.env.LOWIRE_FORCE_CACHE)
-        throw new Error('Cache missing but TL_FORCE_CACHE is set' + JSON.stringify(conversation, null, 2));
+        throw new Error('Cache missing but LOWIRE_FORCE_CACHE is set' + JSON.stringify(conversation, null, 2));
     const result = await provider.complete(conversation, options);
     caches.output[key] = result;
     return result;

package/lib/loop.d.ts

@@ -45,6 +45,8 @@ export type LoopOptions = types.CompletionOptions & LoopEvents & {
     tools?: types.Tool[];
     callTool?: types.ToolCallback;
     maxTurns?: number;
+    maxToolCalls?: number;
+    maxToolCallRetries?: number;
     cache?: types.ReplayCache;
     secrets?: Record<string, string>;
     summarize?: boolean;

package/lib/loop.js

@@ -39,12 +39,16 @@ class Loop {
             tools: allTools,
         };
         const debug = options.debug;
-        let budgetTokens = options.maxTokens;
+        const budget = {
+            tokens: options.maxTokens,
+            toolCalls: options.maxToolCalls,
+            toolCallRetries: options.maxToolCallRetries,
+        };
         const totalUsage = { input: 0, output: 0 };
-        debug?.('lowire:loop')(`Starting ${this._provider.name} loop`, task);
+        debug?.('lowire:loop')(`Starting ${this._provider.name} loop\n${task}`);
         const maxTurns = options.maxTurns || 100;
         for (let turns = 0; turns < maxTurns; ++turns) {
-            if (options.maxTokens && budgetTokens !== undefined && budgetTokens <= 0)
+            if (options.maxTokens && budget.tokens !== undefined && budget.tokens <= 0)
                 throw new Error(`Budget tokens ${options.maxTokens} exhausted`);
             debug?.('lowire:loop')(`Turn ${turns + 1} of (max ${maxTurns})`);
             const caches = options.cache ? {
@@ -52,23 +56,23 @@ class Loop {
                 output: this._cacheOutput,
             } : undefined;
             const summarizedConversation = options.summarize ? this._summarizeConversation(task, conversation, options) : conversation;
-            await options.onBeforeTurn?.({ conversation: summarizedConversation, totalUsage, budgetTokens });
+            await options.onBeforeTurn?.({ conversation: summarizedConversation, totalUsage, budgetTokens: budget.tokens });
             if (abortController?.signal.aborted)
                 return { status: 'break', usage: totalUsage, turns };
             debug?.('lowire:loop')(`Request`, JSON.stringify({ ...summarizedConversation, tools: `${summarizedConversation.tools.length} tools` }, null, 2));
             const { result: assistantMessage, usage } = await (0, cache_1.cachedComplete)(this._provider, summarizedConversation, caches, {
                 ...options,
-                maxTokens: budgetTokens,
+                maxTokens: budget.tokens,
                 signal: abortController?.signal,
             });
             const intent = assistantMessage.content.filter(part => part.type === 'text').map(part => part.text).join('\n');
             totalUsage.input += usage.input;
             totalUsage.output += usage.output;
-            if (budgetTokens !== undefined)
-                budgetTokens -= usage.input + usage.output;
+            if (budget.tokens !== undefined)
+                budget.tokens -= 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));
-            await options.onAfterTurn?.({ assistantMessage, totalUsage, budgetTokens });
+            await options.onAfterTurn?.({ assistantMessage, totalUsage, budgetTokens: budget.tokens });
             if (abortController?.signal.aborted)
                 return { status: 'break', usage: totalUsage, turns };
             conversation.messages.push(assistantMessage);
@@ -78,6 +82,8 @@ class Loop {
                 continue;
             }
             for (const toolCall of toolCalls) {
+                if (budget.toolCalls !== undefined && --budget.toolCalls < 0)
+                    throw new Error(`Failed to perform step, max tool calls (${options.maxToolCalls}) reached`);
                 const { name, arguments: args } = toolCall;
                 debug?.('lowire:loop')('Call tool', name, JSON.stringify(args, null, 2));
                 const status = await options.onBeforeToolCall?.({ assistantMessage, toolCall });
@@ -129,6 +135,11 @@ class Loop {
                     };
                 }
             }
+            const hasErrors = toolCalls.some(toolCall => toolCall.result?.isError);
+            if (!hasErrors)
+                budget.toolCallRetries = options.maxToolCallRetries;
+            if (hasErrors && budget.toolCallRetries !== undefined && --budget.toolCallRetries < 0)
+                throw new Error(`Failed to perform action after ${options.maxToolCallRetries} tool call retries`);
         }
         throw new Error('Failed to perform step, max attempts reached');
     }

package/package.json

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