@anvia/core 1.0.0-rc.5 → 1.0.0-rc.6

package/dist/{chunk-FCANRQ5Q.js → chunk-QXSVVYQK.js}

@@ -13,10 +13,10 @@ import {
   assertPositiveInteger,
   createMemoryCompactionSummary,
   cumulativeCompactedMessageCount
-} from "./chunk-DNQNOHD5.js";
+} from "./chunk-QXQQ2DBM.js";
 import {
   startAgentRunObservers
-} from "./chunk-E5J4OY4K.js";
+} from "./chunk-7O45T5F2.js";
 import {
   isSkillTool
 } from "./chunk-YK4WAAS4.js";
@@ -44,7 +44,7 @@ import {
   AgentStructuredOutputError,
   AgentToolSuspensionError,
   MaxTurnsError
-} from "./chunk-TEV3OB7N.js";
+} from "./chunk-VGJRXUPT.js";
 import {
   mapWithConcurrency
 } from "./chunk-3RM57ZT2.js";
@@ -56,7 +56,7 @@ import {
   isProviderTool,
   isStreamingCompletionModel,
   textFromAssistantContent
-} from "./chunk-HSA6NL2E.js";
+} from "./chunk-HT4P3SSG.js";
 import {
   parseMessage,
   parseMessages,
@@ -963,6 +963,12 @@ var CompletionStreamAccumulator = class {
         usage: finalResponse.usage,
         rawResponse: finalResponse.rawResponse
       };
+      if (finalResponse.finishReason !== void 0) {
+        mergedResponse.finishReason = finalResponse.finishReason;
+      }
+      if (finalResponse.providerFinishReason !== void 0) {
+        mergedResponse.providerFinishReason = finalResponse.providerFinishReason;
+      }
       if (finalResponse.messageId !== void 0) {
         mergedResponse.messageId = finalResponse.messageId;
       }
@@ -1254,6 +1260,21 @@ var UNLABELED_FENCE_START = "```\n";
 var UNLABELED_FENCE_CRLF_START = "```\r\n";
 var FENCE_END = "\n```";
 var STRUCTURED_OUTPUT_RETRY_PROMPT = "Your previous response was invalid structured output. Return only raw JSON that matches the supplied JSON schema. Do not use Markdown fences or include commentary.";
+var STRUCTURED_OUTPUT_TRUNCATED_RETRY_PROMPT = "Your previous response exceeded the provider output limit. Return substantially shorter raw JSON that matches the supplied JSON schema. Do not use Markdown fences or include commentary.";
+var STRUCTURED_OUTPUT_REPAIR_PREVIEW_MAX_LENGTH = 8192;
+var STRUCTURED_OUTPUT_REPAIR_PREVIEW_MARKER = "\n...[output omitted]...\n";
+function structuredOutputRepairPreview(text) {
+  if (text.length <= STRUCTURED_OUTPUT_REPAIR_PREVIEW_MAX_LENGTH) {
+    return { text, includedOutputLength: text.length };
+  }
+  const availableLength = STRUCTURED_OUTPUT_REPAIR_PREVIEW_MAX_LENGTH - STRUCTURED_OUTPUT_REPAIR_PREVIEW_MARKER.length;
+  const headLength = Math.ceil(availableLength / 2);
+  const tailLength = availableLength - headLength;
+  return {
+    text: `${text.slice(0, headLength)}${STRUCTURED_OUTPUT_REPAIR_PREVIEW_MARKER}${text.slice(-tailLength)}`,
+    includedOutputLength: headLength + tailLength
+  };
+}
 function normalizeStructuredOutput(text) {
   const trimmed = text.trim();
   const opening = fenceOpening(trimmed);
@@ -2938,14 +2959,16 @@ var AgentRun = class _AgentRun {
           }
           structuredRetryUsage = cumulativeUsage;
           this.failedCompletionUsage = structuredRetryUsage;
-          currentRequest = structuredOutputRetryRequest(currentRequest, response);
+          const retryRequest = structuredOutputRetryRequest(request, response, error);
+          currentRequest = retryRequest.request;
           await this.scheduleCompletionRetry(
             error,
             attempt,
             turn,
             false,
             retryOptions,
-            runObservers
+            runObservers,
+            structuredOutputRetryEventAttributes(error, retryRequest)
           );
           continue;
         }
@@ -3029,14 +3052,16 @@ var AgentRun = class _AgentRun {
           args.state.providerErrorUsage = Usage.add(args.state.providerErrorUsage, response.usage);
           args.state.firstDeltaMs = void 0;
           args.state.emittedToolCallIds.clear();
-          currentRequest = structuredOutputRetryRequest(currentRequest, response);
+          const retryRequest = structuredOutputRetryRequest(args.request, response, error);
+          currentRequest = retryRequest.request;
           await this.scheduleCompletionRetry(
             error,
             attempt,
             args.turn,
             true,
             retryOptions,
-            args.runObservers
+            args.runObservers,
+            structuredOutputRetryEventAttributes(error, retryRequest)
           );
           continue;
         }
@@ -3180,6 +3205,10 @@ var AgentRun = class _AgentRun {
       modelId: this.agent.model.modelId,
       usage: { ...response.usage }
     };
+    if (response.finishReason !== void 0) generation.finishReason = response.finishReason;
+    if (response.providerFinishReason !== void 0) {
+      generation.providerFinishReason = response.providerFinishReason;
+    }
     if (response.contextUsage !== void 0) generation.contextUsage = response.contextUsage;
     if (response.sources !== void 0) generation.sources = response.sources;
     if (response.providerToolCalls !== void 0) {
@@ -3225,20 +3254,24 @@ var AgentRun = class _AgentRun {
       streaming
     });
   }
-  async scheduleCompletionRetry(error, attempt, turn, streaming, options, runObservers) {
+  async scheduleCompletionRetry(error, attempt, turn, streaming, options, runObservers, additionalAttributes) {
     const delayMs = retryDelayMs(options, attempt);
+    const attributes = {
+      turn,
+      attempt,
+      nextAttempt: attempt + 1,
+      maxAttempts: options.maxAttempts,
+      delayMs,
+      streaming,
+      ...retryErrorAttributes(error)
+    };
+    if (additionalAttributes !== void 0) {
+      Object.assign(attributes, additionalAttributes);
+    }
     await runObservers.event({
       name: "completion.retry",
       level: "WARNING",
-      attributes: {
-        turn,
-        attempt,
-        nextAttempt: attempt + 1,
-        maxAttempts: options.maxAttempts,
-        delayMs,
-        streaming,
-        ...retryErrorAttributes(error)
-      }
+      attributes
     });
     await waitForRetry(delayMs, this.abortController.signal);
   }
@@ -3476,7 +3509,27 @@ var AgentRun = class _AgentRun {
     if (this.agent.outputSchema === void 0) return;
     if (response.choice.some((item) => item.type === "tool-call")) return;
     const text = textFromAssistantContent(response.choice);
-    const output = this.parseOutput(text, attempt, usage, { useValidatedOutput: false });
+    const normalized = normalizeStructuredOutput(text);
+    if (response.finishReason === "length") {
+      throw new AgentStructuredOutputError({
+        phase: "truncated",
+        attempt,
+        maxAttempts: this.completionRetryOptions?.maxAttempts ?? 1,
+        outputLength: text.length,
+        normalizedLength: normalized.text.length,
+        outputFormat: normalized.format,
+        attemptUsage: response.usage,
+        usage,
+        finishReason: response.finishReason,
+        providerFinishReason: response.providerFinishReason
+      });
+    }
+    const output = this.parseOutput(text, attempt, usage, {
+      useValidatedOutput: false,
+      attemptUsage: response.usage,
+      finishReason: response.finishReason,
+      providerFinishReason: response.providerFinishReason
+    });
     this.validatedStructuredOutput = { text, output };
   }
   takeFailedCompletionUsage() {
@@ -3503,7 +3556,10 @@ var AgentRun = class _AgentRun {
         outputLength: text.length,
         normalizedLength: normalized.text.length,
         outputFormat: normalized.format,
+        attemptUsage: options.attemptUsage ?? usage,
         usage,
+        finishReason: options.finishReason,
+        providerFinishReason: options.providerFinishReason,
         cause: error
       });
     }
@@ -3517,7 +3573,10 @@ var AgentRun = class _AgentRun {
         outputLength: text.length,
         normalizedLength: normalized.text.length,
         outputFormat: normalized.format,
+        attemptUsage: options.attemptUsage ?? usage,
         usage,
+        finishReason: options.finishReason,
+        providerFinishReason: options.providerFinishReason,
         cause: error
       });
     }
@@ -3984,10 +4043,14 @@ function responseStreamEvents(turn, response, includeProviderArtifacts = true) {
 function generationArtifacts(messages) {
   const sources = /* @__PURE__ */ new Map();
   const providerToolCalls = /* @__PURE__ */ new Map();
+  let finishReason;
+  let providerFinishReason;
   let contextUsage;
   for (const message of messages) {
     const metadata = getAssistantGenerationMetadata(message);
     if (metadata !== void 0) {
+      finishReason = metadata.finishReason;
+      providerFinishReason = metadata.providerFinishReason;
       contextUsage = metadata.contextUsage;
     }
     for (const source of metadata?.sources ?? []) {
@@ -4003,6 +4066,10 @@ function generationArtifacts(messages) {
   if (providerToolCalls.size > 0) {
     artifacts.providerToolCalls = [...providerToolCalls.values()];
   }
+  if (finishReason !== void 0) artifacts.finishReason = finishReason;
+  if (providerFinishReason !== void 0) {
+    artifacts.providerFinishReason = providerFinishReason;
+  }
   if (contextUsage !== void 0) artifacts.contextUsage = contextUsage;
   return artifacts;
 }
@@ -4049,19 +4116,75 @@ function latestAssistantText(messages) {
   }
   return "";
 }
-function structuredOutputRetryRequest(request, response) {
-  const invalidResponse = {
-    role: "assistant",
-    content: [...response.choice]
-  };
+function structuredOutputRetryRequest(request, response, error) {
+  const truncated = error instanceof AgentStructuredOutputError && error.phase === "truncated";
   const correction = {
     role: "user",
-    content: STRUCTURED_OUTPUT_RETRY_PROMPT
+    content: truncated ? STRUCTURED_OUTPUT_TRUNCATED_RETRY_PROMPT : STRUCTURED_OUTPUT_RETRY_PROMPT
+  };
+  if (truncated) {
+    return {
+      request: {
+        ...request,
+        chatHistory: [...request.chatHistory, correction]
+      },
+      previousResponse: "omitted",
+      includedOutputLength: 0
+    };
+  }
+  const preview = structuredOutputRepairPreview(textFromAssistantContent(response.choice));
+  if (preview.includedOutputLength === 0) {
+    return {
+      request: {
+        ...request,
+        chatHistory: [...request.chatHistory, correction]
+      },
+      previousResponse: "omitted",
+      includedOutputLength: 0
+    };
+  }
+  const invalidResponse = {
+    role: "assistant",
+    content: [{ type: "text", text: preview.text }]
   };
   return {
-    ...request,
-    chatHistory: [...request.chatHistory, invalidResponse, correction]
+    request: {
+      ...request,
+      chatHistory: [...request.chatHistory, invalidResponse, correction]
+    },
+    previousResponse: "preview",
+    includedOutputLength: preview.includedOutputLength
+  };
+}
+function structuredOutputRetryEventAttributes(error, retryRequest) {
+  const attributes = {
+    previousResponse: retryRequest.previousResponse,
+    includedOutputLength: retryRequest.includedOutputLength
   };
+  if (!(error instanceof AgentStructuredOutputError)) return attributes;
+  Object.assign(attributes, {
+    failurePhase: error.phase,
+    outputLength: error.outputLength,
+    normalizedLength: error.normalizedLength,
+    attemptUsage: usageEventValue(error.attemptUsage),
+    cumulativeUsage: usageEventValue(error.usage)
+  });
+  if (error.finishReason !== void 0) attributes.finishReason = error.finishReason;
+  if (error.providerFinishReason !== void 0) {
+    attributes.providerFinishReason = error.providerFinishReason;
+  }
+  return attributes;
+}
+function usageEventValue(usage) {
+  const value = {
+    inputTokens: usage.inputTokens,
+    outputTokens: usage.outputTokens,
+    totalTokens: usage.totalTokens,
+    cachedInputTokens: usage.cachedInputTokens,
+    cacheCreationInputTokens: usage.cacheCreationInputTokens
+  };
+  if (usage.details !== void 0) value.details = { ...usage.details };
+  return value;
 }
 function providerMessages(messages) {
   const providerMessages2 = [];
@@ -4706,4 +4829,4 @@ export {
   markResolvedAgentOptions,
   Agent
 };
-//# sourceMappingURL=chunk-FCANRQ5Q.js.map
+//# sourceMappingURL=chunk-QXSVVYQK.js.map