@musnows/scriverse 0.8.0 → 0.8.1

package/dist/ai.js

@@ -5617,6 +5617,7 @@ export class AiManager {
         });
         let activeSecrets = [];
         let streamedContent = "";
+        let streamedPartialContent = "";
         let trackedInputTokens = 0;
         let trackedOutputTokens = 0;
         let trackedCachedInputTokens = 0;
@@ -5694,6 +5695,7 @@ export class AiManager {
                     traceRound.attempts.push(traceAttempt);
                     saveTrace();
                     logger.info("ai.call.attempt_started", { callId, attempt, maximumAttempts, toolChoice, purpose });
+                    let streamedRoundContent = "";
                     try {
                         const candidate = await this.scheduleProviderRequest(provider, input.signal, async () => {
                             const controller = new AbortController();
@@ -5748,7 +5750,8 @@ export class AiManager {
                                 }
                                 const payload = await this.readCompletionStream(response, protocol, activeSecrets, (delta) => {
                                     attemptEmitted = true;
-                                    streamedContent += delta;
+                                    streamedRoundContent += delta;
+                                    streamedPartialContent += delta;
                                     onDelta?.(delta);
                                 }, (delta) => {
                                     attemptEmitted = true;
@@ -5801,6 +5804,23 @@ export class AiManager {
                         if (candidate.ok) {
                             const parsed = candidate.payload;
                             completionDelivery.set(parsed, candidate.delivery);
+                            if (candidate.delivery === "sse" && purpose === "generation" && streamedRoundContent.length > 0) {
+                                const currentChoice = parsed.choices?.[0];
+                                if (currentChoice?.message?.tool_calls?.length) {
+                                    const step = {
+                                        id: id("process"),
+                                        type: "intermediate",
+                                        round: processRound,
+                                        content: streamedRoundContent,
+                                        createdAt: now()
+                                    };
+                                    processSteps.push(step);
+                                    input.onProcessStep?.(step);
+                                }
+                                else {
+                                    streamedContent += streamedRoundContent;
+                                }
+                            }
                             traceAttempt.completedAt = now();
                             traceAttempt.status = "completed";
                             traceAttempt.httpStatus = candidate.status;
@@ -6141,7 +6161,7 @@ export class AiManager {
          SET status = 'failed', failure = ?, output_chars = ?, input_tokens = ?, output_tokens = ?,
              cached_input_tokens = ?, cache_eligible_input_tokens = ?, cache_usage_available = ?,
              token_usage_source = ?, completed_at = ?
-         WHERE id = ?`, message, streamedContent.length, trackedInputTokens, trackedOutputTokens, trackedCachedInputTokens, trackedCacheEligibleInputTokens, trackedCacheEligibleInputTokens > 0 ? 1 : 0, trackedUsageSource(), now(), callId);
+         WHERE id = ?`, message, (streamedPartialContent || streamedContent).length, trackedInputTokens, trackedOutputTokens, trackedCachedInputTokens, trackedCacheEligibleInputTokens, trackedCacheEligibleInputTokens > 0 ? 1 : 0, trackedUsageSource(), now(), callId);
             logger.error("ai.call.failed", {
                 callId,
                 workId: input.workId,