@limo-labs/deity 0.2.1 → 0.2.2

package/dist/index.cjs

@@ -2876,6 +2876,7 @@ var DEFAULT_LLM_LOOP_CONFIG = {
   // 2 minutes
   verbose: false
 };
+var DEADLINE_PROMPT = `\u26A0\uFE0F DEADLINE REACHED: You have reached the maximum allowed rounds or time limit. You MUST immediately provide your final answer based on the information gathered so far. Do NOT call any more tools. Respond with your best final output NOW.`;
 async function executeLLMLoop(adapter, initialMessages, tools, config, ctx, loopConfig = {}, validator) {
   const cfg = { ...DEFAULT_LLM_LOOP_CONFIG, ...loopConfig };
   const messagesWithMemory = await injectRelevantMemories(
@@ -2889,6 +2890,7 @@ async function executeLLMLoop(adapter, initialMessages, tools, config, ctx, loop
   let toolCallsExecuted = 0;
   let response;
   let toolCallsThisRound = [];
+  const startTime = Date.now();
   while (rounds < cfg.maxToolRounds) {
     rounds++;
     toolCallsThisRound = [];
@@ -2936,10 +2938,6 @@ async function executeLLMLoop(adapter, initialMessages, tools, config, ctx, loop
       }
       break;
     }
-    if (rounds >= cfg.maxToolRounds) {
-      errors.push(`Maximum tool rounds (${cfg.maxToolRounds}) exceeded`);
-      break;
-    }
     for (const toolCall of response.toolCalls) {
       const alreadyExecuted = toolCall._alreadyExecuted;
       if (alreadyExecuted) {
@@ -3007,6 +3005,23 @@ async function executeLLMLoop(adapter, initialMessages, tools, config, ctx, loop
     if (allPreExecuted) {
       break;
     }
+    const isTimeout = cfg.timeout > 0 && Date.now() - startTime >= cfg.timeout;
+    const isLastRound = rounds >= cfg.maxToolRounds;
+    if (isTimeout || isLastRound) {
+      messages.push({
+        role: "user",
+        content: DEADLINE_PROMPT
+      });
+      try {
+        response = await adapter.generate(messages, void 0, config, ctx);
+        messages.push({ role: "assistant", content: response.content });
+      } catch (_error) {
+      }
+      errors.push(
+        isTimeout ? `Timeout reached (${cfg.timeout}ms)` : `Maximum tool rounds (${cfg.maxToolRounds}) reached`
+      );
+      break;
+    }
   }
   return {
     response,
@@ -3246,6 +3261,17 @@ async function executeWithRetry(component, ctx, executeFn) {
     }
     const shouldRetry = retryConfig.shouldRetry ? retryConfig.shouldRetry(attempts, output, ctx) : attempts < retryConfig.maxAttempts;
     if (!shouldRetry) {
+      const lastOutput = outputHistory[outputHistory.length - 1];
+      if (lastOutput !== void 0) {
+        return {
+          output: lastOutput,
+          success: true,
+          attempts,
+          errors: validation.errors ?? ["Validation failed (used best-effort output)"],
+          validationHistory,
+          outputHistory
+        };
+      }
       return {
         success: false,
         attempts,
@@ -4344,7 +4370,7 @@ zod.z.object({
   sessionId: zod.z.string().min(1, "Session ID cannot be empty"),
   workflowName: zod.z.string().min(1, "Workflow name cannot be empty"),
   pausedAt: zod.z.string().datetime("Must be ISO 8601 datetime"),
-  version: zod.z.string().regex(/^\d+\.\d+\.\d+$/, "Must be semver format (e.g., 4.0.0)"),
+  version: zod.z.string().regex(/^\d+\.\d+\.\d+$/, "Must be semver format (e.g., 1.0.0)"),
   state: PausedSessionStateSchema,
   metadata: PausedSessionMetadataSchema
 });
@@ -4360,7 +4386,7 @@ var SessionSerializer = class {
   metadata;
   constructor() {
     this.metadata = {
-      version: "4.0.0",
+      version: "1.0.0",
       timestamp: /* @__PURE__ */ new Date(),
       specialFields: []
     };
@@ -5252,6 +5278,7 @@ function createLoopNode(child, maxIterations) {
 
 exports.Agent = Agent;
 exports.Conditional = Conditional;
+exports.DEADLINE_PROMPT = DEADLINE_PROMPT;
 exports.DEBUG_ENABLED = DEBUG_ENABLED;
 exports.DebugLogger = DebugLogger;
 exports.ForEach = ForEach;