@argosvix/sdk 0.4.15-alpha.0 → 0.4.17-alpha.0

package/dist/client.js

@@ -1,7 +1,7 @@
 import { calculateCost, calculateCostWithCache } from "./pricing.js";
 import { Recorder } from "./recorder.js";
 import { generateId } from "./ids.js";
-import { getAmbientTraceContext, _registerObservationSink } from "./context.js";
+import { getAmbientTraceContext, getAmbientPromptTag, _registerObservationSink } from "./context.js";
 const wrappedClients = new WeakMap();
 const wrappedGeminiModels = new WeakSet();
 /**
@@ -61,9 +61,11 @@ export function wrap(client, config = {}) {
             break;
         }
     }
-    // 平文 capture の対応範囲(2026-07-02 パリティ解消): 全 4 provider の
-    // 非ストリーミング success path で promptBody / completionBody を抽出する。
-    // streaming 経路は全 provider 未対応(メタデータは記録され、平文のみ省略)。
+    // 平文 capture の対応範囲(2026-07 streaming 対応): 全 4 provider の
+    // 非ストリーミング success path に加え、既存 stream wrapper がある streaming 経路
+    // (OpenAI Chat / Anthropic / Mistral / Gemini legacy + 新 SDK)でも promptBody /
+    // completionBody を抽出する。OpenAI Responses の streaming は stream wrapper 自体が
+    // 未実装(下記 "not yet observed" warn)のため平文も未対応。
     wrappedClients.set(client, recorder);
     return client;
 }
@@ -186,6 +188,13 @@ function buildTags(config, provider, requestArgs) {
         if (native !== undefined)
             tags.userId = native;
     }
+    // withPrompt() の内側なら prompt タグ({name}@v{version})を自動付与。
+    // 明示 tags.prompt が優先(上書きしない)。版別の品質/コスト比較の基盤(2026-07-02 #4)。
+    if (tags.prompt === undefined) {
+        const ambientPrompt = getAmbientPromptTag();
+        if (ambientPrompt !== undefined)
+            tags.prompt = ambientPrompt;
+    }
     return tags;
 }
 /**
@@ -201,8 +210,10 @@ function buildTags(config, provider, requestArgs) {
  *   - anthropic (Messages) = prompt + completion
  *   - mistral (chat.complete) = prompt + completion
  *   - gemini (legacy generateContent + 新 SDK models.generateContent) = prompt + completion
- *   - 全 streaming 経路 = 未対応(captureContent が true でも promptBody /
- *     completionBody は付与されない。メタデータは通常どおり記録される)
+ *   - streaming 経路(2026-07 対応)= 既存 stream wrapper の finalize で promptBody
+ *     (入口 snapshot)+ completionBody(テキスト delta 蓄積、256KB 上限)を付与。
+ *     ツール呼び出し引数の streaming 組み立ては対象外。OpenAI Responses streaming は
+ *     stream wrapper 未実装のため未対応。
  */
 function extractOpenAIChatPromptBody(requestArgs) {
     if (!Array.isArray(requestArgs.messages) || requestArgs.messages.length === 0) {
@@ -462,6 +473,154 @@ function safeStringify(value) {
         return undefined;
     }
 }
+// ============================================================
+// streaming 平文キャプチャ(2026-07 設計、 docs/handoff/streaming-capture-design-2026-07.md)
+//
+// promptBody は既存の extract*PromptBody を stream 入口で snapshot し、
+// completionBody は既存 stream wrapper(usage 集計で全チャンク観測済み)に
+// テキスト delta の蓄積を足すだけで得る(= 新しい傍受点は作らない)。
+// redaction は従来どおり Recorder の flush 時に一括適用される。
+// ============================================================
+/**
+ * 蓄積バッファの上限(= 262,144 bytes = 256KB)。ingest 側の本文上限と整合。
+ * 暴走ストリームで wrapper がホストアプリのメモリを食う事故を構造的に防ぐ。
+ */
+const STREAM_CAPTURE_MAX_BYTES = 262_144;
+const STREAM_TRUNCATED_MARKER = "…[truncated]";
+/**
+ * UTF-16 code unit 走査で UTF-8 バイト長を数える(TextEncoder の毎 delta 割り当てを
+ * 避ける)。surrogate pair は 4 bytes で 1 code point として数える。
+ */
+function utf8ByteLength(text) {
+    let bytes = 0;
+    for (let i = 0; i < text.length; i++) {
+        const code = text.charCodeAt(i);
+        if (code < 0x80)
+            bytes += 1;
+        else if (code < 0x800)
+            bytes += 2;
+        else if (code >= 0xd800 && code <= 0xdbff && i + 1 < text.length) {
+            // surrogate pair = 4 bytes(後続の low surrogate を 1 つ消費)
+            bytes += 4;
+            i++;
+        }
+        else
+            bytes += 3;
+    }
+    return bytes;
+}
+/**
+ * streaming テキスト delta の蓄積バッファ。256KB 上限で超過分は捨てて、
+ * result() が末尾に truncated マーカーを付けた本文を返す。
+ */
+class StreamTextAccumulator {
+    parts = [];
+    bytes = 0;
+    truncated = false;
+    push(text) {
+        if (typeof text !== "string" || text.length === 0 || this.truncated)
+            return;
+        const size = utf8ByteLength(text);
+        if (this.bytes + size <= STREAM_CAPTURE_MAX_BYTES) {
+            this.parts.push(text);
+            this.bytes += size;
+            return;
+        }
+        // 上限をまたぐ delta は入る分だけ code point 境界で継ぎ足し、以降は捨てる。
+        let remaining = STREAM_CAPTURE_MAX_BYTES - this.bytes;
+        let head = "";
+        for (const ch of text) {
+            const chBytes = utf8ByteLength(ch);
+            if (chBytes > remaining)
+                break;
+            head += ch;
+            remaining -= chBytes;
+        }
+        if (head.length > 0) {
+            this.parts.push(head);
+            this.bytes = STREAM_CAPTURE_MAX_BYTES - remaining;
+        }
+        this.truncated = true;
+    }
+    /**
+     * 蓄積本文を返す(空なら undefined = record に載せない)。
+     * interrupted = 例外 / 途中打ち切りで「そこまでの本文」を返すとき、marker を強制付与。
+     */
+    result(interrupted = false) {
+        if (this.parts.length === 0)
+            return undefined;
+        const body = this.parts.join("");
+        return this.truncated || interrupted ? body + STREAM_TRUNCATED_MARKER : body;
+    }
+}
+/**
+ * stream 入口での capture 状態生成。captureContent でないときは undefined を返して
+ * 抽出コストも蓄積コストもゼロに保つ(promptBody 抽出は thunk で遅延)。
+ */
+function buildStreamCapture(config, extractPromptBody) {
+    if (config.captureContent !== true)
+        return undefined;
+    const capture = { acc: new StreamTextAccumulator() };
+    const promptBody = extractPromptBody();
+    if (promptBody !== undefined)
+        capture.promptBody = promptBody;
+    return capture;
+}
+/**
+ * stream finalize / error record に promptBody / completionBody を付与する。
+ * redaction はここではなく Recorder の flush 時に一括適用される(非 stream と同方針)。
+ */
+function applyStreamCapture(record, capture, interrupted) {
+    if (!capture)
+        return;
+    if (capture.promptBody !== undefined)
+        record.promptBody = capture.promptBody;
+    const completionBody = capture.acc.result(interrupted);
+    if (completionBody !== undefined)
+        record.completionBody = completionBody;
+}
+/** OpenAI Chat stream チャンクのテキスト delta(choices[0].delta.content、string のみ)。 */
+function extractOpenAIStreamDeltaText(chunk) {
+    const delta = chunk.choices?.[0]?.delta;
+    const content = delta?.content;
+    return typeof content === "string" && content.length > 0 ? content : undefined;
+}
+/**
+ * Gemini stream チャンクのテキスト delta。新 SDK は chunk.text(string property)、
+ * legacy は chunk.text()(関数、safety block 等で throw しうる)。どちらも無ければ
+ * candidates[0].content.parts[].text を継ぎ足す(JSON fallback はしない = delta のみ)。
+ */
+function extractGeminiStreamChunkText(chunk) {
+    if (!chunk || typeof chunk !== "object")
+        return undefined;
+    const t = chunk.text;
+    if (typeof t === "string" && t.length > 0)
+        return t;
+    if (typeof t === "function") {
+        try {
+            const v = t.call(chunk);
+            if (typeof v === "string" && v.length > 0)
+                return v;
+        }
+        catch {
+            // legacy chunk.text() は candidate なし等で throw する。 蓄積は諦めて続行。
+        }
+    }
+    const candidates = chunk.candidates;
+    if (!Array.isArray(candidates) || candidates.length === 0)
+        return undefined;
+    const parts = candidates[0]
+        ?.content?.parts;
+    if (!Array.isArray(parts))
+        return undefined;
+    const texts = [];
+    for (const p of parts) {
+        const text = p && typeof p === "object" ? p.text : undefined;
+        if (typeof text === "string" && text.length > 0)
+            texts.push(text);
+    }
+    return texts.length > 0 ? texts.join("") : undefined;
+}
 function extractErrorDetails(err) {
     if (!err || typeof err !== "object")
         return undefined;
@@ -571,7 +730,10 @@ function wrapOpenAIChat(client, recorder, config) {
             await recorder.budgetGate.check({ model: requestArgs.model, payload: requestArgs });
             const response = await originalCreate(...args);
             if (isStream) {
-                return wrapOpenAIStream(response, recorder, requestArgs, start, id, callTags, buildTraceMeta(config), usageInjected);
+                return wrapOpenAIStream(response, recorder, requestArgs, start, id, callTags, buildTraceMeta(config), usageInjected, 
+                // streaming 平文キャプチャ = promptBody は入口 snapshot、 completionBody は
+                // delta 蓄積(captureContent=true のときのみ生成)。
+                buildStreamCapture(config, () => extractOpenAIChatPromptBody(requestArgs)));
             }
             const r = response;
             const latencyMs = Date.now() - start;
@@ -643,7 +805,9 @@ callTags,
 traceMeta, 
 // 我々が include_usage を注入したか。 true のとき末尾 usage-only チャンクは host に
 // yield せず観測だけする(consumer の stream 形状を変えない)。
-usageInjected = false) {
+usageInjected = false, 
+// streaming 平文キャプチャ(captureContent=true のときのみ渡る)。
+capture) {
     let finalModel = requestArgs.model || "unknown";
     let promptTokens = 0;
     let completionTokens = 0;
@@ -652,6 +816,8 @@ usageInjected = false) {
     let lastUsage;
     // 監査 Tier 2 = TTFT。 最初に本文 delta を載せたチャンク到達時刻 - 開始時刻。
     let ttftMs;
+    // 正常完走したか(早期 break / abandon では false のまま → 本文に truncated marker)。
+    let completed = false;
     // 監査 2026-06-28 = 消費側が早期 break / abandon すると generator が yield で
     // .return() され、 ループ後の記録が skip されて課金済み呼び出しが record されない
     // (budget gate の spend も漏れる)。 finally + recorded フラグで「正常完走 / error /
@@ -662,7 +828,7 @@ usageInjected = false) {
             return;
         recorded = true;
         const streamCost = calculateCostWithCache("openai", finalModel, promptTokens, completionTokens, cachedReadTokens, 0);
-        recorder.record({
+        const record = {
             id,
             provider: "openai",
             model: finalModel,
@@ -679,7 +845,9 @@ usageInjected = false) {
             tags: callTags,
             ...traceMeta,
             requestMeta: buildOpenAIRequestMeta(requestArgs),
-        });
+        };
+        applyStreamCapture(record, capture, !completed);
+        recorder.record(record);
     };
     try {
         for await (const chunk of stream) {
@@ -688,6 +856,7 @@ usageInjected = false) {
                 // 先頭チャンクでは立てない(= TTFT を過小評価しない、 Codex SHOULD_FIX)。
                 ttftMs = Date.now() - start;
             }
+            capture?.acc.push(extractOpenAIStreamDeltaText(chunk));
             if (chunk.model)
                 finalModel = chunk.model;
             if (chunk.usage) {
@@ -708,12 +877,13 @@ usageInjected = false) {
             }
             yield chunk;
         }
+        completed = true;
         recordOnce();
     }
     catch (err) {
         recorded = true; // error record で確定(finally の recordOnce を抑止)
         const errorDetails = extractErrorDetails(err);
-        recorder.record({
+        const errorRecord = {
             id,
             provider: "openai",
             model: finalModel,
@@ -728,7 +898,10 @@ usageInjected = false) {
             error: err instanceof Error ? err.message : String(err),
             ...(errorDetails ? { errorDetails } : {}),
             requestMeta: buildOpenAIRequestMeta(requestArgs),
-        });
+        };
+        // 例外で切れた場合は「そこまでの本文 + truncated マーカー」を record に載せる。
+        applyStreamCapture(errorRecord, capture, true);
+        recorder.record(errorRecord);
         throw err;
     }
     finally {
@@ -880,7 +1053,7 @@ function wrapAnthropic(client, recorder, config) {
             await recorder.budgetGate.check({ model: requestArgs.model, payload: requestArgs });
             const response = await originalCreate(...args);
             if (isStream) {
-                return wrapAnthropicStream(response, recorder, requestArgs, start, id, callTags, buildTraceMeta(config));
+                return wrapAnthropicStream(response, recorder, requestArgs, start, id, callTags, buildTraceMeta(config), buildStreamCapture(config, () => extractAnthropicPromptBody(requestArgs)));
             }
             const r = response;
             const latencyMs = Date.now() - start;
@@ -942,13 +1115,17 @@ function wrapAnthropic(client, recorder, config) {
         }
     };
 }
-async function* wrapAnthropicStream(stream, recorder, requestArgs, start, id, callTags, traceMeta) {
+async function* wrapAnthropicStream(stream, recorder, requestArgs, start, id, callTags, traceMeta, 
+// streaming 平文キャプチャ(captureContent=true のときのみ渡る)。
+capture) {
     let model = requestArgs.model || "unknown";
     // Anthropic: input_tokens は「非キャッシュ input」。 cache 読取/書込は別計上。
     let inputTokens = 0;
     let cachedReadTokens = 0;
     let cachedWriteTokens = 0;
     let completionTokens = 0;
+    // 正常完走したか(早期 break / abandon では false のまま → 本文に truncated marker)。
+    let completed = false;
     // 監査 2026-06-28 = 早期 break でも観測済みトークンで 1 回だけ記録(OpenAI と同様)。
     let recorded = false;
     const recordOnce = () => {
@@ -957,7 +1134,7 @@ async function* wrapAnthropicStream(stream, recorder, requestArgs, start, id, ca
         recorded = true;
         const promptTokens = inputTokens + cachedReadTokens + cachedWriteTokens;
         const cost = calculateCostWithCache("anthropic", model, promptTokens, completionTokens, cachedReadTokens, cachedWriteTokens);
-        recorder.record({
+        const record = {
             id,
             provider: "anthropic",
             model,
@@ -973,10 +1150,19 @@ async function* wrapAnthropicStream(stream, recorder, requestArgs, start, id, ca
             tags: callTags,
             ...traceMeta,
             requestMeta: buildAnthropicRequestMeta(requestArgs),
-        });
+        };
+        applyStreamCapture(record, capture, !completed);
+        recorder.record(record);
     };
     try {
         for await (const event of stream) {
+            if (capture &&
+                event.type === "content_block_delta" &&
+                event.delta?.type === "text_delta" &&
+                typeof event.delta.text === "string") {
+                // テキスト delta のみ蓄積(ツール引数の input_json_delta 等は対象外)。
+                capture.acc.push(event.delta.text);
+            }
             if (event.type === "message_start" && event.message) {
                 if (event.message.model)
                     model = event.message.model;
@@ -994,13 +1180,14 @@ async function* wrapAnthropicStream(stream, recorder, requestArgs, start, id, ca
             }
             yield event;
         }
+        completed = true;
         recordOnce();
     }
     catch (err) {
         recorded = true;
         const errorDetails = extractErrorDetails(err);
         const promptTokens = inputTokens + cachedReadTokens + cachedWriteTokens;
-        recorder.record({
+        const errorRecord = {
             id,
             provider: "anthropic",
             model,
@@ -1015,7 +1202,10 @@ async function* wrapAnthropicStream(stream, recorder, requestArgs, start, id, ca
             error: err instanceof Error ? err.message : String(err),
             ...(errorDetails ? { errorDetails } : {}),
             requestMeta: buildAnthropicRequestMeta(requestArgs),
-        });
+        };
+        // 例外で切れた場合は「そこまでの本文 + truncated マーカー」を record に載せる。
+        applyStreamCapture(errorRecord, capture, true);
+        recorder.record(errorRecord);
         throw err;
     }
     finally {
@@ -1107,7 +1297,7 @@ function wrapMistral(client, recorder, config) {
             try {
                 await recorder.budgetGate.check({ model: requestArgs.model, payload: requestArgs });
                 const stream = (await originalStream(...args));
-                return wrapMistralStream(stream, recorder, requestArgs, start, id, callTags, buildTraceMeta(config));
+                return wrapMistralStream(stream, recorder, requestArgs, start, id, callTags, buildTraceMeta(config), buildStreamCapture(config, () => extractMistralPromptBody(requestArgs)));
             }
             catch (err) {
                 // H-2: stream initialization error (= auth/validation/connection)
@@ -1133,18 +1323,22 @@ function wrapMistral(client, recorder, config) {
         };
     }
 }
-async function* wrapMistralStream(stream, recorder, requestArgs, start, id, callTags, traceMeta) {
+async function* wrapMistralStream(stream, recorder, requestArgs, start, id, callTags, traceMeta, 
+// streaming 平文キャプチャ(captureContent=true のときのみ渡る)。
+capture) {
     let model = requestArgs.model || "unknown";
     let promptTokens = 0;
     let completionTokens = 0;
     let reportedTotal;
+    // 正常完走したか(早期 break / abandon では false のまま → 本文に truncated marker)。
+    let completed = false;
     // 監査 2026-06-28 = 早期 break でも観測済みトークンで 1 回だけ記録(OpenAI と同様)。
     let recorded = false;
     const recordOnce = () => {
         if (recorded)
             return;
         recorded = true;
-        recorder.record({
+        const record = {
             id,
             provider: "mistral",
             model,
@@ -1157,11 +1351,19 @@ async function* wrapMistralStream(stream, recorder, requestArgs, start, id, call
             tags: callTags,
             ...traceMeta,
             requestMeta: buildMistralRequestMeta(requestArgs),
-        });
+        };
+        applyStreamCapture(record, capture, !completed);
+        recorder.record(record);
     };
     try {
         for await (const chunk of stream) {
             const inner = chunk.data ?? chunk;
+            if (capture) {
+                // Mistral は OpenAI Chat 互換の delta 形状(choices[0].delta.content、string のみ)。
+                const delta = inner.choices?.[0]?.delta;
+                if (typeof delta?.content === "string")
+                    capture.acc.push(delta.content);
+            }
             if (inner.model)
                 model = inner.model;
             if (inner.usage) {
@@ -1173,12 +1375,13 @@ async function* wrapMistralStream(stream, recorder, requestArgs, start, id, call
             }
             yield chunk;
         }
+        completed = true;
         recordOnce();
     }
     catch (err) {
         recorded = true;
         const errorDetails = extractErrorDetails(err);
-        recorder.record({
+        const errorRecord = {
             id,
             provider: "mistral",
             model,
@@ -1193,7 +1396,10 @@ async function* wrapMistralStream(stream, recorder, requestArgs, start, id, call
             error: err instanceof Error ? err.message : String(err),
             ...(errorDetails ? { errorDetails } : {}),
             requestMeta: buildMistralRequestMeta(requestArgs),
-        });
+        };
+        // 例外で切れた場合は「そこまでの本文 + truncated マーカー」を record に載せる。
+        applyStreamCapture(errorRecord, capture, true);
+        recorder.record(errorRecord);
         throw err;
     }
     finally {
@@ -1307,9 +1513,12 @@ function wrapGeminiLegacyModel(model, modelName, recorder, config) {
                 const originalStreamRef = result.stream;
                 // #1 R1: trace 軸は stream 生成時(= 入口、 ambient 有効)に snapshot する。
                 const traceMeta = buildTraceMeta(config);
+                // streaming 平文キャプチャ = promptBody 入口 snapshot + delta 蓄積。
+                const capture = buildStreamCapture(config, () => extractGeminiPromptBody(requestArgs));
                 const wrappedStream = (async function* () {
                     try {
                         for await (const chunk of originalStreamRef) {
+                            capture?.acc.push(extractGeminiStreamChunkText(chunk));
                             yield chunk;
                         }
                         const finalResponse = await result.response;
@@ -1318,7 +1527,7 @@ function wrapGeminiLegacyModel(model, modelName, recorder, config) {
                         const completionTokens = usage?.candidatesTokenCount ?? 0;
                         const cachedReadTokens = usage?.cachedContentTokenCount ?? 0;
                         const cost = calculateCostWithCache("gemini", modelName, promptTokens, completionTokens, cachedReadTokens, 0);
-                        recorder.record({
+                        const record = {
                             id,
                             provider: "gemini",
                             model: modelName,
@@ -1333,11 +1542,13 @@ function wrapGeminiLegacyModel(model, modelName, recorder, config) {
                             tags: callTags,
                             ...traceMeta,
                             requestMeta: buildGeminiRequestMeta(requestArgs),
-                        });
+                        };
+                        applyStreamCapture(record, capture, false);
+                        recorder.record(record);
                     }
                     catch (err) {
                         const errorDetails = extractErrorDetails(err);
-                        recorder.record({
+                        const errorRecord = {
                             id,
                             provider: "gemini",
                             model: modelName,
@@ -1352,7 +1563,10 @@ function wrapGeminiLegacyModel(model, modelName, recorder, config) {
                             error: err instanceof Error ? err.message : String(err),
                             ...(errorDetails ? { errorDetails } : {}),
                             requestMeta: buildGeminiRequestMeta(requestArgs),
-                        });
+                        };
+                        // 例外で切れた場合は「そこまでの本文 + truncated マーカー」を record に載せる。
+                        applyStreamCapture(errorRecord, capture, true);
+                        recorder.record(errorRecord);
                         throw err;
                     }
                 })();
@@ -1464,10 +1678,13 @@ function wrapGeminiNew(client, recorder, config) {
                 // C-2 + H-3 fix: AsyncGenerator wrap with finally-record + error tied to consumption
                 // #1 R1: trace 軸は stream 生成時(= 入口、 ambient 有効)に snapshot する。
                 const traceMeta = buildTraceMeta(config);
+                // streaming 平文キャプチャ = promptBody 入口 snapshot + delta 蓄積。
+                const capture = buildStreamCapture(config, () => extractGeminiPromptBody(requestArgs));
                 return (async function* () {
                     let lastUsage;
                     try {
                         for await (const chunk of stream) {
+                            capture?.acc.push(extractGeminiStreamChunkText(chunk));
                             if (chunk.usageMetadata)
                                 lastUsage = chunk.usageMetadata;
                             yield chunk;
@@ -1476,7 +1693,7 @@ function wrapGeminiNew(client, recorder, config) {
                         const completionTokens = lastUsage?.candidatesTokenCount ?? 0;
                         const cachedReadTokens = lastUsage?.cachedContentTokenCount ?? 0;
                         const cost = calculateCostWithCache("gemini", modelName, promptTokens, completionTokens, cachedReadTokens, 0);
-                        recorder.record({
+                        const record = {
                             id,
                             provider: "gemini",
                             model: modelName,
@@ -1491,11 +1708,13 @@ function wrapGeminiNew(client, recorder, config) {
                             tags: callTags,
                             ...traceMeta,
                             requestMeta: buildGeminiNewRequestMeta(requestArgs),
-                        });
+                        };
+                        applyStreamCapture(record, capture, false);
+                        recorder.record(record);
                     }
                     catch (err) {
                         const errorDetails = extractErrorDetails(err);
-                        recorder.record({
+                        const errorRecord = {
                             id,
                             provider: "gemini",
                             model: modelName,
@@ -1512,7 +1731,10 @@ function wrapGeminiNew(client, recorder, config) {
                             error: err instanceof Error ? err.message : String(err),
                             ...(errorDetails ? { errorDetails } : {}),
                             requestMeta: buildGeminiNewRequestMeta(requestArgs),
-                        });
+                        };
+                        // 例外で切れた場合は「そこまでの本文 + truncated マーカー」を record に載せる。
+                        applyStreamCapture(errorRecord, capture, true);
+                        recorder.record(errorRecord);
                         throw err;
                     }
                 })();