@markusylisiurunen/tau 0.2.101 → 0.2.103

package/dist/core/utils/gemini_speech.js

@@ -0,0 +1,367 @@
+import { z } from "zod";
+const GEMINI_GENERATE_CONTENT_BASE_URL = "https://generativelanguage.googleapis.com/v1beta/models";
+const DEFAULT_GEMINI_SPEECH_REWRITE_MODEL = "gemini-3-flash-preview";
+const DEFAULT_GEMINI_SPEECH_TTS_MODEL = "gemini-3.1-flash-tts-preview";
+const DEFAULT_GEMINI_TTS_VOICE_NAME = "Despina";
+const DEFAULT_TTS_SAMPLE_RATE_HZ = 24000;
+const DEFAULT_TTS_CHANNEL_COUNT = 1;
+const DEFAULT_TTS_BITS_PER_SAMPLE = 16;
+const DEFAULT_TTS_MAX_ATTEMPTS = 3;
+const DEFAULT_TTS_CONCURRENCY = 6;
+const RETRYABLE_TTS_STATUS_CODES = new Set([408, 429, 500, 502, 503, 504]);
+const errorPayloadSchema = z.object({
+    error: z
+        .object({
+        message: z.string().trim().min(1).optional(),
+        status: z.string().trim().min(1).optional(),
+        code: z.number().int().optional(),
+    })
+        .optional(),
+});
+class GeminiApiError extends Error {
+    status;
+    constructor(message, status) {
+        super(message);
+        this.name = "GeminiApiError";
+        this.status = status;
+    }
+}
+class GeminiTtsResponseError extends Error {
+    constructor(message) {
+        super(message);
+        this.name = "GeminiTtsResponseError";
+    }
+}
+export async function generateGeminiSpeech(options) {
+    const sourceText = options.sourceText.trim();
+    if (!sourceText) {
+        throw new Error("speech source text was empty");
+    }
+    const apiKey = options.apiKey.trim();
+    if (!apiKey) {
+        throw new Error("missing Gemini API key");
+    }
+    const fetchImpl = options.fetchImpl ?? fetch;
+    const rewriteModel = options.rewriteModel ?? DEFAULT_GEMINI_SPEECH_REWRITE_MODEL;
+    const ttsModel = options.ttsModel ?? DEFAULT_GEMINI_SPEECH_TTS_MODEL;
+    const voiceName = options.voiceName ?? DEFAULT_GEMINI_TTS_VOICE_NAME;
+    await options.onStageChange?.("rewriting");
+    const spokenText = await rewriteTextForSpeech({
+        apiKey,
+        model: rewriteModel,
+        sourceText,
+        fetchImpl,
+        signal: options.signal,
+    });
+    const spokenParagraphs = splitSpeechParagraphs(spokenText);
+    await options.onStageChange?.("generating");
+    await options.onChunkProgress?.({ ready: 0, total: spokenParagraphs.length });
+    const pcmChunks = await synthesizeSpeechAudioChunks({
+        apiKey,
+        model: ttsModel,
+        voiceName,
+        spokenParagraphs,
+        fetchImpl,
+        signal: options.signal,
+        maxAttempts: options.maxTtsAttempts ?? DEFAULT_TTS_MAX_ATTEMPTS,
+        concurrency: DEFAULT_TTS_CONCURRENCY,
+        onChunkProgress: options.onChunkProgress,
+    });
+    return {
+        spokenText,
+        audio: encodeWaveFile({
+            pcmAudio: Buffer.concat(pcmChunks),
+            sampleRateHz: DEFAULT_TTS_SAMPLE_RATE_HZ,
+            channelCount: DEFAULT_TTS_CHANNEL_COUNT,
+            bitsPerSample: DEFAULT_TTS_BITS_PER_SAMPLE,
+        }),
+        mimeType: "audio/wav",
+    };
+}
+async function rewriteTextForSpeech(args) {
+    const payload = await requestGeminiGenerateContent({
+        apiKey: args.apiKey,
+        model: args.model,
+        fetchImpl: args.fetchImpl,
+        signal: args.signal,
+        body: {
+            contents: [
+                {
+                    parts: [
+                        {
+                            text: buildSpeechRewritePrompt(args.sourceText),
+                        },
+                    ],
+                },
+            ],
+            generationConfig: {
+                thinkingConfig: {
+                    thinkingLevel: "minimal",
+                },
+            },
+        },
+    });
+    const rewrittenText = extractGeminiText(payload).trim();
+    if (!rewrittenText) {
+        throw new Error("Gemini rewrite returned empty text");
+    }
+    return rewrittenText;
+}
+async function synthesizeSpeechAudioChunks(args) {
+    const total = args.spokenParagraphs.length;
+    const results = new Array(total);
+    const concurrency = Math.max(1, Math.trunc(args.concurrency));
+    let nextIndex = 0;
+    let ready = 0;
+    const worker = async () => {
+        while (true) {
+            const index = nextIndex;
+            nextIndex += 1;
+            if (index >= total) {
+                return;
+            }
+            results[index] = await synthesizeSpeechAudioChunk({
+                apiKey: args.apiKey,
+                model: args.model,
+                voiceName: args.voiceName,
+                spokenText: args.spokenParagraphs[index],
+                fetchImpl: args.fetchImpl,
+                signal: args.signal,
+                maxAttempts: args.maxAttempts,
+            });
+            ready += 1;
+            await args.onChunkProgress?.({ ready, total });
+        }
+    };
+    await Promise.all(Array.from({ length: Math.min(concurrency, total) }, () => worker()));
+    return results;
+}
+async function synthesizeSpeechAudioChunk(args) {
+    const maxAttempts = Math.max(1, Math.trunc(args.maxAttempts));
+    let lastError;
+    for (let attempt = 1; attempt <= maxAttempts; attempt++) {
+        try {
+            const payload = await requestGeminiGenerateContent({
+                apiKey: args.apiKey,
+                model: args.model,
+                fetchImpl: args.fetchImpl,
+                signal: args.signal,
+                body: {
+                    contents: [
+                        {
+                            parts: [
+                                {
+                                    text: buildSpeechSynthesisPrompt(args.spokenText),
+                                },
+                            ],
+                        },
+                    ],
+                    generationConfig: {
+                        responseModalities: ["AUDIO"],
+                        temperature: 1,
+                        speechConfig: {
+                            voiceConfig: {
+                                prebuiltVoiceConfig: {
+                                    voiceName: args.voiceName,
+                                },
+                            },
+                        },
+                    },
+                },
+            });
+            const audioData = extractGeminiInlineAudioData(payload);
+            if (!audioData) {
+                throw new GeminiTtsResponseError("Gemini TTS response did not include audio data");
+            }
+            return Buffer.from(audioData, "base64");
+        }
+        catch (error) {
+            lastError = error;
+            if (args.signal?.aborted || !isRetryableTtsError(error) || attempt >= maxAttempts) {
+                throw error;
+            }
+            await waitForRetryDelay(attempt, args.signal);
+        }
+    }
+    throw lastError instanceof Error ? lastError : new Error("Gemini TTS request failed");
+}
+async function requestGeminiGenerateContent(args) {
+    const response = await args.fetchImpl(`${GEMINI_GENERATE_CONTENT_BASE_URL}/${encodeURIComponent(args.model)}:generateContent`, {
+        method: "POST",
+        headers: {
+            "Content-Type": "application/json",
+            "x-goog-api-key": args.apiKey,
+        },
+        body: JSON.stringify(args.body),
+        signal: args.signal,
+    });
+    const responseText = await response.text();
+    let payload;
+    try {
+        payload = responseText ? JSON.parse(responseText) : undefined;
+    }
+    catch {
+        payload = undefined;
+    }
+    if (!response.ok) {
+        const parsed = errorPayloadSchema.safeParse(payload);
+        const fallbackMessage = responseText.trim() || `HTTP ${response.status}`;
+        const message = parsed.success
+            ? (parsed.data.error?.message ?? fallbackMessage)
+            : fallbackMessage;
+        throw new GeminiApiError(message, response.status);
+    }
+    return payload;
+}
+function extractGeminiText(payload) {
+    if (!isObject(payload) || !Array.isArray(payload.candidates)) {
+        return "";
+    }
+    for (const candidate of payload.candidates) {
+        if (!isObject(candidate) ||
+            !isObject(candidate.content) ||
+            !Array.isArray(candidate.content.parts)) {
+            continue;
+        }
+        const text = candidate.content.parts
+            .map((part) => (isObject(part) && typeof part.text === "string" ? part.text : ""))
+            .join("")
+            .trim();
+        if (text) {
+            return text;
+        }
+    }
+    return "";
+}
+function extractGeminiInlineAudioData(payload) {
+    if (!isObject(payload) || !Array.isArray(payload.candidates)) {
+        return undefined;
+    }
+    for (const candidate of payload.candidates) {
+        if (!isObject(candidate) ||
+            !isObject(candidate.content) ||
+            !Array.isArray(candidate.content.parts)) {
+            continue;
+        }
+        for (const part of candidate.content.parts) {
+            if (!isObject(part) ||
+                !isObject(part.inlineData) ||
+                typeof part.inlineData.data !== "string") {
+                continue;
+            }
+            const data = part.inlineData.data.trim();
+            if (data) {
+                return data;
+            }
+        }
+    }
+    return undefined;
+}
+function buildSpeechRewritePrompt(sourceText) {
+    return [
+        "Rewrite the assistant response below so it sounds natural when spoken aloud.",
+        "Keep prose unchanged. Only rewrite spans that are awkward to say aloud, so the listener can follow along with the original text.",
+        "Do not drop, condense, or add content. Preserve meaning, order, and technical accuracy.",
+        "Things that typically need rewriting: file paths, shell commands, code identifiers, markdown structure, XML-like tags, long option lists, and code-heavy formatting.",
+        "For file references, keep the filename and any line or range info that was actually present. Do not add location detail that was not in the original.",
+        "Say code identifiers and version strings as natural words (for example, handleToolUiEvent as 'handle tool UI event', v5.4 as 'version 5.4'). Keep numbers and units natural (for example, 8,192 tokens as 'about eight thousand tokens').",
+        "",
+        "Examples of good rewrites:",
+        '- `src/core/utils/gemini_speech.ts:372` → "gemini_speech.ts, line 372"',
+        '- `src/tui/chat_controller.ts:1819-1855` → "chat_controller.ts, lines 1819 to 1855"',
+        '- `src/core/session/compaction.ts` → "compaction.ts"',
+        '- `/Users/markus/.config/tau/config.json` → "the tau config.json in your home directory"',
+        '- `rg --heading -n -t ts "ToolUiText" src` → "ripgrep for ToolUiText in TypeScript files under src"',
+        '- `<available-skills>` → "the available-skills tag"',
+        "- A markdown bullet list of short items → a natural comma-separated list or short sentences",
+        "",
+        "Write in prose-like paragraphs of sensible spoken length, separated by blank lines.",
+        "Return plain text only, with no markdown fences or commentary.",
+        "",
+        "ASSISTANT RESPONSE:",
+        sourceText,
+    ].join("\n");
+}
+function splitSpeechParagraphs(spokenText) {
+    const paragraphs = spokenText
+        .split(/\n\s*\n/g)
+        .map((paragraph) => paragraph.trim())
+        .filter((paragraph) => paragraph.length > 0);
+    return paragraphs.length > 0 ? paragraphs : [spokenText.trim()];
+}
+function buildSpeechSynthesisPrompt(spokenText) {
+    return [
+        "Synthesize speech audio for the labeled transcript below.",
+        "Speak only the transcript. Do not speak the instructions, labels, or headings.",
+        "",
+        "### DIRECTOR'S NOTES",
+        "Style: Clear, natural, conversational.",
+        "Pacing: Slightly slower than conversational, with deliberate enunciation. The audio will be sped up during playback.",
+        "",
+        "### TRANSCRIPT",
+        spokenText,
+    ].join("\n");
+}
+function isRetryableTtsError(error) {
+    if (error instanceof GeminiTtsResponseError) {
+        return true;
+    }
+    if (error instanceof GeminiApiError) {
+        return error.status !== undefined && RETRYABLE_TTS_STATUS_CODES.has(error.status);
+    }
+    if (error instanceof Error) {
+        return error.name !== "AbortError";
+    }
+    return false;
+}
+async function waitForRetryDelay(attempt, signal) {
+    const durationMs = Math.min(1000, 150 * 2 ** (attempt - 1));
+    await new Promise((resolve, reject) => {
+        const timeout = setTimeout(() => {
+            signal?.removeEventListener("abort", onAbort);
+            resolve();
+        }, durationMs);
+        timeout.unref?.();
+        const onAbort = () => {
+            clearTimeout(timeout);
+            reject(abortError());
+        };
+        if (!signal) {
+            return;
+        }
+        if (signal.aborted) {
+            clearTimeout(timeout);
+            reject(abortError());
+            return;
+        }
+        signal.addEventListener("abort", onAbort, { once: true });
+    });
+}
+function encodeWaveFile(args) {
+    const byteRate = (args.sampleRateHz * args.channelCount * args.bitsPerSample) / 8;
+    const blockAlign = (args.channelCount * args.bitsPerSample) / 8;
+    const header = Buffer.alloc(44);
+    header.write("RIFF", 0);
+    header.writeUInt32LE(36 + args.pcmAudio.length, 4);
+    header.write("WAVE", 8);
+    header.write("fmt ", 12);
+    header.writeUInt32LE(16, 16);
+    header.writeUInt16LE(1, 20);
+    header.writeUInt16LE(args.channelCount, 22);
+    header.writeUInt32LE(args.sampleRateHz, 24);
+    header.writeUInt32LE(byteRate, 28);
+    header.writeUInt16LE(blockAlign, 32);
+    header.writeUInt16LE(args.bitsPerSample, 34);
+    header.write("data", 36);
+    header.writeUInt32LE(args.pcmAudio.length, 40);
+    return Buffer.concat([header, args.pcmAudio]);
+}
+function isObject(value) {
+    return typeof value === "object" && value !== null;
+}
+function abortError() {
+    const error = new Error("The operation was aborted");
+    error.name = "AbortError";
+    return error;
+}
+//# sourceMappingURL=gemini_speech.js.map

package/dist/core/utils/gemini_speech.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"gemini_speech.js","sourceRoot":"","sources":["../../../src/core/utils/gemini_speech.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,MAAM,gCAAgC,GAAG,yDAAyD,CAAC;AACnG,MAAM,mCAAmC,GAAG,wBAAwB,CAAC;AACrE,MAAM,+BAA+B,GAAG,8BAA8B,CAAC;AACvE,MAAM,6BAA6B,GAAG,SAAS,CAAC;AAChD,MAAM,0BAA0B,GAAG,KAAK,CAAC;AACzC,MAAM,yBAAyB,GAAG,CAAC,CAAC;AACpC,MAAM,2BAA2B,GAAG,EAAE,CAAC;AACvC,MAAM,wBAAwB,GAAG,CAAC,CAAC;AACnC,MAAM,uBAAuB,GAAG,CAAC,CAAC;AAClC,MAAM,0BAA0B,GAAG,IAAI,GAAG,CAAC,CAAC,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,EAAE,GAAG,CAAC,CAAC,CAAC;AAE3E,MAAM,kBAAkB,GAAG,CAAC,CAAC,MAAM,CAAC;IAClC,KAAK,EAAE,CAAC;SACL,MAAM,CAAC;QACN,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE;QAC5C,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,QAAQ,EAAE;QAC3C,IAAI,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,QAAQ,EAAE;KAClC,CAAC;SACD,QAAQ,EAAE;CACd,CAAC,CAAC;AA4BH,MAAM,cAAe,SAAQ,KAAK;IACvB,MAAM,CAAU;IAEzB,YAAY,OAAe,EAAE,MAAe;QAC1C,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,gBAAgB,CAAC;QAC7B,IAAI,CAAC,MAAM,GAAG,MAAM,CAAC;IACvB,CAAC;CACF;AAED,MAAM,sBAAuB,SAAQ,KAAK;IACxC,YAAY,OAAe;QACzB,KAAK,CAAC,OAAO,CAAC,CAAC;QACf,IAAI,CAAC,IAAI,GAAG,wBAAwB,CAAC;IACvC,CAAC;CACF;AAED,MAAM,CAAC,KAAK,UAAU,oBAAoB,CACxC,OAA4B;IAE5B,MAAM,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC;IAC7C,IAAI,CAAC,UAAU,EAAE,CAAC;QAChB,MAAM,IAAI,KAAK,CAAC,8BAA8B,CAAC,CAAC;IAClD,CAAC;IAED,MAAM,MAAM,GAAG,OAAO,CAAC,MAAM,CAAC,IAAI,EAAE,CAAC;IACrC,IAAI,CAAC,MAAM,EAAE,CAAC;QACZ,MAAM,IAAI,KAAK,CAAC,wBAAwB,CAAC,CAAC;IAC5C,CAAC;IAED,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,KAAK,CAAC;IAC7C,MAAM,YAAY,GAAG,OAAO,CAAC,YAAY,IAAI,mCAAmC,CAAC;IACjF,MAAM,QAAQ,GAAG,OAAO,CAAC,QAAQ,IAAI,+BAA+B,CAAC;IACrE,MAAM,SAAS,GAAG,OAAO,CAAC,SAAS,IAAI,6BAA6B,CAAC;IACrE,MAAM,OAAO,CAAC,aAAa,EAAE,CAAC,WAAW,CAAC,CAAC;IAC3C,MAAM,UAAU,GAAG,MAAM,oBAAoB,CAAC;QAC5C,MAAM;QACN,KAAK,EAAE,YAAY;QACnB,UAAU;QACV,SAAS;QACT,MAAM,EAAE,OAAO,CAAC,MAAM;KACvB,CAAC,CAAC;IAEH,MAAM,gBAAgB,GAAG,qBAAqB,CAAC,UAAU,CAAC,CAAC;IAC3D,MAAM,OAAO,CAAC,aAAa,EAAE,CAAC,YAAY,CAAC,CAAC;IAC5C,MAAM,OAAO,CAAC,eAAe,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,gBAAgB,CAAC,MAAM,EAAE,CAAC,CAAC;IAC9E,MAAM,SAAS,GAAG,MAAM,2BAA2B,CAAC;QAClD,MAAM;QACN,KAAK,EAAE,QAAQ;QACf,SAAS;QACT,gBAAgB;QAChB,SAAS;QACT,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,WAAW,EAAE,OAAO,CAAC,cAAc,IAAI,wBAAwB;QAC/D,WAAW,EAAE,uBAAuB;QACpC,eAAe,EAAE,OAAO,CAAC,eAAe;KACzC,CAAC,CAAC;IAEH,OAAO;QACL,UAAU;QACV,KAAK,EAAE,cAAc,CAAC;YACpB,QAAQ,EAAE,MAAM,CAAC,MAAM,CAAC,SAAS,CAAC;YAClC,YAAY,EAAE,0BAA0B;YACxC,YAAY,EAAE,yBAAyB;YACvC,aAAa,EAAE,2BAA2B;SAC3C,CAAC;QACF,QAAQ,EAAE,WAAW;KACtB,CAAC;AACJ,CAAC;AAUD,KAAK,UAAU,oBAAoB,CAAC,IAA8B;IAChE,MAAM,OAAO,GAAG,MAAM,4BAA4B,CAAC;QACjD,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,KAAK,EAAE,IAAI,CAAC,KAAK;QACjB,SAAS,EAAE,IAAI,CAAC,SAAS;QACzB,MAAM,EAAE,IAAI,CAAC,MAAM;QACnB,IAAI,EAAE;YACJ,QAAQ,EAAE;gBACR;oBACE,KAAK,EAAE;wBACL;4BACE,IAAI,EAAE,wBAAwB,CAAC,IAAI,CAAC,UAAU,CAAC;yBAChD;qBACF;iBACF;aACF;YACD,gBAAgB,EAAE;gBAChB,cAAc,EAAE;oBACd,aAAa,EAAE,SAAS;iBACzB;aACF;SACF;KACF,CAAC,CAAC;IAEH,MAAM,aAAa,GAAG,iBAAiB,CAAC,OAAO,CAAC,CAAC,IAAI,EAAE,CAAC;IACxD,IAAI,CAAC,aAAa,EAAE,CAAC;QACnB,MAAM,IAAI,KAAK,CAAC,oCAAoC,CAAC,CAAC;IACxD,CAAC;IACD,OAAO,aAAa,CAAC;AACvB,CAAC;AAcD,KAAK,UAAU,2BAA2B,CACxC,IAAqC;IAErC,MAAM,KAAK,GAAG,IAAI,CAAC,gBAAgB,CAAC,MAAM,CAAC;IAC3C,MAAM,OAAO,GAAG,IAAI,KAAK,CAAS,KAAK,CAAC,CAAC;IACzC,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;IAC9D,IAAI,SAAS,GAAG,CAAC,CAAC;IAClB,IAAI,KAAK,GAAG,CAAC,CAAC;IAEd,MAAM,MAAM,GAAG,KAAK,IAAmB,EAAE;QACvC,OAAO,IAAI,EAAE,CAAC;YACZ,MAAM,KAAK,GAAG,SAAS,CAAC;YACxB,SAAS,IAAI,CAAC,CAAC;YACf,IAAI,KAAK,IAAI,KAAK,EAAE,CAAC;gBACnB,OAAO;YACT,CAAC;YAED,OAAO,CAAC,KAAK,CAAC,GAAG,MAAM,0BAA0B,CAAC;gBAChD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,UAAU,EAAE,IAAI,CAAC,gBAAgB,CAAC,KAAK,CAAE;gBACzC,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,WAAW,EAAE,IAAI,CAAC,WAAW;aAC9B,CAAC,CAAC;YACH,KAAK,IAAI,CAAC,CAAC;YACX,MAAM,IAAI,CAAC,eAAe,EAAE,CAAC,EAAE,KAAK,EAAE,KAAK,EAAE,CAAC,CAAC;QACjD,CAAC;IACH,CAAC,CAAC;IAEF,MAAM,OAAO,CAAC,GAAG,CAAC,KAAK,CAAC,IAAI,CAAC,EAAE,MAAM,EAAE,IAAI,CAAC,GAAG,CAAC,WAAW,EAAE,KAAK,CAAC,EAAE,EAAE,GAAG,EAAE,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACxF,OAAO,OAAO,CAAC;AACjB,CAAC;AAYD,KAAK,UAAU,0BAA0B,CAAC,IAAoC;IAC5E,MAAM,WAAW,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,WAAW,CAAC,CAAC,CAAC;IAC9D,IAAI,SAAkB,CAAC;IAEvB,KAAK,IAAI,OAAO,GAAG,CAAC,EAAE,OAAO,IAAI,WAAW,EAAE,OAAO,EAAE,EAAE,CAAC;QACxD,IAAI,CAAC;YACH,MAAM,OAAO,GAAG,MAAM,4BAA4B,CAAC;gBACjD,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,KAAK,EAAE,IAAI,CAAC,KAAK;gBACjB,SAAS,EAAE,IAAI,CAAC,SAAS;gBACzB,MAAM,EAAE,IAAI,CAAC,MAAM;gBACnB,IAAI,EAAE;oBACJ,QAAQ,EAAE;wBACR;4BACE,KAAK,EAAE;gCACL;oCACE,IAAI,EAAE,0BAA0B,CAAC,IAAI,CAAC,UAAU,CAAC;iCAClD;6BACF;yBACF;qBACF;oBACD,gBAAgB,EAAE;wBAChB,kBAAkB,EAAE,CAAC,OAAO,CAAC;wBAC7B,WAAW,EAAE,CAAC;wBACd,YAAY,EAAE;4BACZ,WAAW,EAAE;gCACX,mBAAmB,EAAE;oCACnB,SAAS,EAAE,IAAI,CAAC,SAAS;iCAC1B;6BACF;yBACF;qBACF;iBACF;aACF,CAAC,CAAC;YAEH,MAAM,SAAS,GAAG,4BAA4B,CAAC,OAAO,CAAC,CAAC;YACxD,IAAI,CAAC,SAAS,EAAE,CAAC;gBACf,MAAM,IAAI,sBAAsB,CAAC,gDAAgD,CAAC,CAAC;YACrF,CAAC;YAED,OAAO,MAAM,CAAC,IAAI,CAAC,SAAS,EAAE,QAAQ,CAAC,CAAC;QAC1C,CAAC;QAAC,OAAO,KAAK,EAAE,CAAC;YACf,SAAS,GAAG,KAAK,CAAC;YAClB,IAAI,IAAI,CAAC,MAAM,EAAE,OAAO,IAAI,CAAC,mBAAmB,CAAC,KAAK,CAAC,IAAI,OAAO,IAAI,WAAW,EAAE,CAAC;gBAClF,MAAM,KAAK,CAAC;YACd,CAAC;YACD,MAAM,iBAAiB,CAAC,OAAO,EAAE,IAAI,CAAC,MAAM,CAAC,CAAC;QAChD,CAAC;IACH,CAAC;IAED,MAAM,SAAS,YAAY,KAAK,CAAC,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;AACxF,CAAC;AAUD,KAAK,UAAU,4BAA4B,CACzC,IAAsC;IAEtC,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CACnC,GAAG,gCAAgC,IAAI,kBAAkB,CAAC,IAAI,CAAC,KAAK,CAAC,kBAAkB,EACvF;QACE,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,cAAc,EAAE,kBAAkB;YAClC,gBAAgB,EAAE,IAAI,CAAC,MAAM;SAC9B;QACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,IAAI,CAAC;QAC/B,MAAM,EAAE,IAAI,CAAC,MAAM;KACpB,CACF,CAAC;IAEF,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC3C,IAAI,OAAgB,CAAC;IACrB,IAAI,CAAC;QACH,OAAO,GAAG,YAAY,CAAC,CAAC,CAAE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAa,CAAC,CAAC,CAAC,SAAS,CAAC;IAC7E,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,GAAG,SAAS,CAAC;IACtB,CAAC;IAED,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,MAAM,GAAG,kBAAkB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QACrD,MAAM,eAAe,GAAG,YAAY,CAAC,IAAI,EAAE,IAAI,QAAQ,QAAQ,CAAC,MAAM,EAAE,CAAC;QACzE,MAAM,OAAO,GAAG,MAAM,CAAC,OAAO;YAC5B,CAAC,CAAC,CAAC,MAAM,CAAC,IAAI,CAAC,KAAK,EAAE,OAAO,IAAI,eAAe,CAAC;YACjD,CAAC,CAAC,eAAe,CAAC;QACpB,MAAM,IAAI,cAAc,CAAC,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC;IACrD,CAAC;IAED,OAAO,OAAO,CAAC;AACjB,CAAC;AAED,SAAS,iBAAiB,CAAC,OAAgB;IACzC,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QAC7D,OAAO,EAAE,CAAC;IACZ,CAAC;IAED,KAAK,MAAM,SAAS,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;QAC3C,IACE,CAAC,QAAQ,CAAC,SAAS,CAAC;YACpB,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC;YAC5B,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,EACvC,CAAC;YACD,SAAS;QACX,CAAC;QAED,MAAM,IAAI,GAAG,SAAS,CAAC,OAAO,CAAC,KAAK;aACjC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC,QAAQ,CAAC,IAAI,CAAC,IAAI,OAAO,IAAI,CAAC,IAAI,KAAK,QAAQ,CAAC,CAAC,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC;aACjF,IAAI,CAAC,EAAE,CAAC;aACR,IAAI,EAAE,CAAC;QACV,IAAI,IAAI,EAAE,CAAC;YACT,OAAO,IAAI,CAAC;QACd,CAAC;IACH,CAAC;IAED,OAAO,EAAE,CAAC;AACZ,CAAC;AAED,SAAS,4BAA4B,CAAC,OAAgB;IACpD,IAAI,CAAC,QAAQ,CAAC,OAAO,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,EAAE,CAAC;QAC7D,OAAO,SAAS,CAAC;IACnB,CAAC;IAED,KAAK,MAAM,SAAS,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;QAC3C,IACE,CAAC,QAAQ,CAAC,SAAS,CAAC;YACpB,CAAC,QAAQ,CAAC,SAAS,CAAC,OAAO,CAAC;YAC5B,CAAC,KAAK,CAAC,OAAO,CAAC,SAAS,CAAC,OAAO,CAAC,KAAK,CAAC,EACvC,CAAC;YACD,SAAS;QACX,CAAC;QAED,KAAK,MAAM,IAAI,IAAI,SAAS,CAAC,OAAO,CAAC,KAAK,EAAE,CAAC;YAC3C,IACE,CAAC,QAAQ,CAAC,IAAI,CAAC;gBACf,CAAC,QAAQ,CAAC,IAAI,CAAC,UAAU,CAAC;gBAC1B,OAAO,IAAI,CAAC,UAAU,CAAC,IAAI,KAAK,QAAQ,EACxC,CAAC;gBACD,SAAS;YACX,CAAC;YACD,MAAM,IAAI,GAAG,IAAI,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,EAAE,CAAC;YACzC,IAAI,IAAI,EAAE,CAAC;gBACT,OAAO,IAAI,CAAC;YACd,CAAC;QACH,CAAC;IACH,CAAC;IAED,OAAO,SAAS,CAAC;AACnB,CAAC;AAED,SAAS,wBAAwB,CAAC,UAAkB;IAClD,OAAO;QACL,8EAA8E;QAC9E,kIAAkI;QAClI,yFAAyF;QACzF,sKAAsK;QACtK,uJAAuJ;QACvJ,2OAA2O;QAC3O,EAAE;QACF,4BAA4B;QAC5B,wEAAwE;QACxE,qFAAqF;QACrF,sDAAsD;QACtD,0FAA0F;QAC1F,qGAAqG;QACrG,qDAAqD;QACrD,6FAA6F;QAC7F,EAAE;QACF,qFAAqF;QACrF,gEAAgE;QAChE,EAAE;QACF,qBAAqB;QACrB,UAAU;KACX,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,SAAS,qBAAqB,CAAC,UAAkB;IAC/C,MAAM,UAAU,GAAG,UAAU;SAC1B,KAAK,CAAC,UAAU,CAAC;SACjB,GAAG,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,IAAI,EAAE,CAAC;SACpC,MAAM,CAAC,CAAC,SAAS,EAAE,EAAE,CAAC,SAAS,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC;IAE/C,OAAO,UAAU,CAAC,MAAM,GAAG,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC,UAAU,CAAC,IAAI,EAAE,CAAC,CAAC;AAClE,CAAC;AAED,SAAS,0BAA0B,CAAC,UAAkB;IACpD,OAAO;QACL,2DAA2D;QAC3D,gFAAgF;QAChF,EAAE;QACF,sBAAsB;QACtB,wCAAwC;QACxC,sHAAsH;QACtH,EAAE;QACF,gBAAgB;QAChB,UAAU;KACX,CAAC,IAAI,CAAC,IAAI,CAAC,CAAC;AACf,CAAC;AAED,SAAS,mBAAmB,CAAC,KAAc;IACzC,IAAI,KAAK,YAAY,sBAAsB,EAAE,CAAC;QAC5C,OAAO,IAAI,CAAC;IACd,CAAC;IAED,IAAI,KAAK,YAAY,cAAc,EAAE,CAAC;QACpC,OAAO,KAAK,CAAC,MAAM,KAAK,SAAS,IAAI,0BAA0B,CAAC,GAAG,CAAC,KAAK,CAAC,MAAM,CAAC,CAAC;IACpF,CAAC;IAED,IAAI,KAAK,YAAY,KAAK,EAAE,CAAC;QAC3B,OAAO,KAAK,CAAC,IAAI,KAAK,YAAY,CAAC;IACrC,CAAC;IAED,OAAO,KAAK,CAAC;AACf,CAAC;AAED,KAAK,UAAU,iBAAiB,CAAC,OAAe,EAAE,MAAoB;IACpE,MAAM,UAAU,GAAG,IAAI,CAAC,GAAG,CAAC,IAAI,EAAE,GAAG,GAAG,CAAC,IAAI,CAAC,OAAO,GAAG,CAAC,CAAC,CAAC,CAAC;IAC5D,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC1C,MAAM,OAAO,GAAG,UAAU,CAAC,GAAG,EAAE;YAC9B,MAAM,EAAE,mBAAmB,CAAC,OAAO,EAAE,OAAO,CAAC,CAAC;YAC9C,OAAO,EAAE,CAAC;QACZ,CAAC,EAAE,UAAU,CAAC,CAAC;QACf,OAAO,CAAC,KAAK,EAAE,EAAE,CAAC;QAElB,MAAM,OAAO,GAAG,GAAG,EAAE;YACnB,YAAY,CAAC,OAAO,CAAC,CAAC;YACtB,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC;QACvB,CAAC,CAAC;QAEF,IAAI,CAAC,MAAM,EAAE,CAAC;YACZ,OAAO;QACT,CAAC;QAED,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;YACnB,YAAY,CAAC,OAAO,CAAC,CAAC;YACtB,MAAM,CAAC,UAAU,EAAE,CAAC,CAAC;YACrB,OAAO;QACT,CAAC;QAED,MAAM,CAAC,gBAAgB,CAAC,OAAO,EAAE,OAAO,EAAE,EAAE,IAAI,EAAE,IAAI,EAAE,CAAC,CAAC;IAC5D,CAAC,CAAC,CAAC;AACL,CAAC;AAED,SAAS,cAAc,CAAC,IAKvB;IACC,MAAM,QAAQ,GAAG,CAAC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IAClF,MAAM,UAAU,GAAG,CAAC,IAAI,CAAC,YAAY,GAAG,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC;IAChE,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;IAEhC,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACxB,MAAM,CAAC,aAAa,CAAC,EAAE,GAAG,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACnD,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,CAAC,CAAC,CAAC;IACxB,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACzB,MAAM,CAAC,aAAa,CAAC,EAAE,EAAE,EAAE,CAAC,CAAC;IAC7B,MAAM,CAAC,aAAa,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC;IAC5B,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;IAC5C,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,YAAY,EAAE,EAAE,CAAC,CAAC;IAC5C,MAAM,CAAC,aAAa,CAAC,QAAQ,EAAE,EAAE,CAAC,CAAC;IACnC,MAAM,CAAC,aAAa,CAAC,UAAU,EAAE,EAAE,CAAC,CAAC;IACrC,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,aAAa,EAAE,EAAE,CAAC,CAAC;IAC7C,MAAM,CAAC,KAAK,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IACzB,MAAM,CAAC,aAAa,CAAC,IAAI,CAAC,QAAQ,CAAC,MAAM,EAAE,EAAE,CAAC,CAAC;IAE/C,OAAO,MAAM,CAAC,MAAM,CAAC,CAAC,MAAM,EAAE,IAAI,CAAC,QAAQ,CAAC,CAAC,CAAC;AAChD,CAAC;AAED,SAAS,QAAQ,CAAC,KAAc;IAC9B,OAAO,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,KAAK,IAAI,CAAC;AACrD,CAAC;AAED,SAAS,UAAU;IACjB,MAAM,KAAK,GAAG,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;IACrD,KAAK,CAAC,IAAI,GAAG,YAAY,CAAC;IAC1B,OAAO,KAAK,CAAC;AACf,CAAC"}

package/dist/core/utils/git.js

@@ -1,5 +1,5 @@
 import { spawnSync } from "node:child_process";
-import { relative, resolve } from "node:path";
+import { resolve } from "node:path";
 export function getGitRoot(cwd) {
     try {
         const res = spawnSync("git", ["rev-parse", "--show-toplevel"], {
@@ -19,11 +19,6 @@ export function getGitRoot(cwd) {
     }
 }
 export function resolvePromptGitRoot(args) {
-    const hostCwd = args.hostCwd ?? args.cwd;
-    const hostGitRoot = getGitRoot(hostCwd);
-    if (!hostGitRoot)
-        return undefined;
-    const rel = relative(hostCwd, hostGitRoot);
-    return resolve(args.cwd, rel);
+    return getGitRoot(args.cwd);
 }
 //# sourceMappingURL=git.js.map

package/dist/core/utils/git.js.map

@@ -1 +1 @@
-{"version":3,"file":"git.js","sourceRoot":"","sources":["../../../src/core/utils/git.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,QAAQ,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAE9C,MAAM,UAAU,UAAU,CAAC,GAAW;IACpC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC,KAAK,EAAE,CAAC,WAAW,EAAE,iBAAiB,CAAC,EAAE;YAC7D,GAAG;YACH,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,CAAC;SACpC,CAAC,CAAC;QAEH,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QAEvC,MAAM,IAAI,GAAG,CAAC,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACvC,IAAI,CAAC,IAAI;YAAE,OAAO,SAAS,CAAC;QAE5B,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,IAAuC;IAC1E,MAAM,OAAO,GAAG,IAAI,CAAC,OAAO,IAAI,IAAI,CAAC,GAAG,CAAC;IACzC,MAAM,WAAW,GAAG,UAAU,CAAC,OAAO,CAAC,CAAC;IACxC,IAAI,CAAC,WAAW;QAAE,OAAO,SAAS,CAAC;IAEnC,MAAM,GAAG,GAAG,QAAQ,CAAC,OAAO,EAAE,WAAW,CAAC,CAAC;IAC3C,OAAO,OAAO,CAAC,IAAI,CAAC,GAAG,EAAE,GAAG,CAAC,CAAC;AAChC,CAAC"}
+{"version":3,"file":"git.js","sourceRoot":"","sources":["../../../src/core/utils/git.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,SAAS,EAAE,MAAM,oBAAoB,CAAC;AAC/C,OAAO,EAAE,OAAO,EAAE,MAAM,WAAW,CAAC;AAEpC,MAAM,UAAU,UAAU,CAAC,GAAW;IACpC,IAAI,CAAC;QACH,MAAM,GAAG,GAAG,SAAS,CAAC,KAAK,EAAE,CAAC,WAAW,EAAE,iBAAiB,CAAC,EAAE;YAC7D,GAAG;YACH,QAAQ,EAAE,OAAO;YACjB,KAAK,EAAE,CAAC,QAAQ,EAAE,MAAM,EAAE,QAAQ,CAAC;SACpC,CAAC,CAAC;QAEH,IAAI,GAAG,CAAC,MAAM,KAAK,CAAC;YAAE,OAAO,SAAS,CAAC;QAEvC,MAAM,IAAI,GAAG,CAAC,GAAG,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,IAAI,EAAE,CAAC;QACvC,IAAI,CAAC,IAAI;YAAE,OAAO,SAAS,CAAC;QAE5B,OAAO,OAAO,CAAC,IAAI,CAAC,CAAC;IACvB,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,MAAM,UAAU,oBAAoB,CAAC,IAAqB;IACxD,OAAO,UAAU,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC;AAC9B,CAAC"}

package/dist/core/utils/mistral_document_ocr.js

@@ -0,0 +1,186 @@
+import { z } from "zod";
+const MISTRAL_FILES_URL = "https://api.mistral.ai/v1/files";
+const MISTRAL_DOCUMENT_OCR_URL = "https://api.mistral.ai/v1/ocr";
+const DEFAULT_MISTRAL_DOCUMENT_OCR_MODEL = "mistral-ocr-latest";
+const DEFAULT_MISTRAL_DOCUMENT_MIME_TYPE = "application/pdf";
+const errorSchema = z.union([
+    z.object({ message: z.string() }),
+    z.object({ detail: z.string() }),
+    z.object({ error: z.object({ message: z.string() }) }),
+]);
+const uploadSuccessSchema = z.object({
+    id: z.string().uuid(),
+});
+const tableSchema = z
+    .object({
+    id: z.string().min(1),
+    content: z.string(),
+    format: z.string(),
+})
+    .passthrough();
+const imageSchema = z
+    .object({
+    id: z.string().min(1),
+    top_left_x: z.number().int().nonnegative().nullable().optional(),
+    top_left_y: z.number().int().nonnegative().nullable().optional(),
+    bottom_right_x: z.number().int().nonnegative().nullable().optional(),
+    bottom_right_y: z.number().int().nonnegative().nullable().optional(),
+})
+    .passthrough();
+const successSchema = z.object({
+    pages: z
+        .array(z
+        .object({
+        index: z.number().int().nonnegative(),
+        markdown: z.string(),
+        tables: z.array(tableSchema).optional(),
+        images: z.array(imageSchema).optional(),
+    })
+        .passthrough())
+        .min(1),
+});
+function parseResponsePayload(responseText) {
+    try {
+        return responseText ? JSON.parse(responseText) : undefined;
+    }
+    catch {
+        return undefined;
+    }
+}
+function parseErrorMessage(responseText, payload, status) {
+    const parsed = errorSchema.safeParse(payload);
+    if (parsed.success) {
+        if ("message" in parsed.data) {
+            return parsed.data.message;
+        }
+        if ("detail" in parsed.data) {
+            return parsed.data.detail;
+        }
+        return parsed.data.error.message;
+    }
+    return responseText.trim() || `HTTP ${status}`;
+}
+async function uploadMistralFile(args) {
+    const formData = new FormData();
+    formData.append("purpose", "ocr");
+    formData.append("file", new Blob([Uint8Array.from(args.document)], {
+        type: args.mimeType,
+    }), args.fileName);
+    const response = await args.fetchImpl(MISTRAL_FILES_URL, {
+        method: "POST",
+        headers: {
+            Authorization: `Bearer ${args.apiKey}`,
+        },
+        body: formData,
+    });
+    const responseText = await response.text();
+    const payload = parseResponsePayload(responseText);
+    if (!response.ok) {
+        throw new Error(parseErrorMessage(responseText, payload, response.status));
+    }
+    const parsed = uploadSuccessSchema.safeParse(payload);
+    if (!parsed.success) {
+        throw new Error("unexpected Mistral file upload response format");
+    }
+    return parsed.data.id;
+}
+async function deleteMistralFile(args) {
+    try {
+        const response = await args.fetchImpl(`${MISTRAL_FILES_URL}/${args.fileId}`, {
+            method: "DELETE",
+            headers: {
+                Authorization: `Bearer ${args.apiKey}`,
+            },
+        });
+        if (response.ok) {
+            return undefined;
+        }
+        const responseText = await response.text();
+        const payload = parseResponsePayload(responseText);
+        return `failed to delete the uploaded PDF from Mistral: ${parseErrorMessage(responseText, payload, response.status)}`;
+    }
+    catch (error) {
+        return `failed to delete the uploaded PDF from Mistral: ${error instanceof Error ? error.message : String(error)}`;
+    }
+}
+export async function ocrMistralDocument(options) {
+    const fetchFn = options.fetchImpl ?? fetch;
+    const fileId = await uploadMistralFile({
+        apiKey: options.apiKey,
+        document: options.document,
+        fileName: options.fileName,
+        mimeType: options.mimeType ?? DEFAULT_MISTRAL_DOCUMENT_MIME_TYPE,
+        fetchImpl: fetchFn,
+    });
+    let pages;
+    let operationError;
+    try {
+        const response = await fetchFn(MISTRAL_DOCUMENT_OCR_URL, {
+            method: "POST",
+            headers: {
+                Authorization: `Bearer ${options.apiKey}`,
+                "Content-Type": "application/json",
+            },
+            body: JSON.stringify({
+                model: options.model ?? DEFAULT_MISTRAL_DOCUMENT_OCR_MODEL,
+                document: {
+                    type: "file",
+                    file_id: fileId,
+                },
+                table_format: "markdown",
+            }),
+        });
+        const responseText = await response.text();
+        const payload = parseResponsePayload(responseText);
+        if (!response.ok) {
+            throw new Error(parseErrorMessage(responseText, payload, response.status));
+        }
+        const parsed = successSchema.safeParse(payload);
+        if (!parsed.success) {
+            throw new Error("unexpected Mistral OCR response format");
+        }
+        const sortedPages = [...parsed.data.pages].sort((a, b) => a.index - b.index);
+        for (const [expectedIndex, page] of sortedPages.entries()) {
+            if (page.index !== expectedIndex) {
+                throw new Error("unexpected Mistral OCR page indexes");
+            }
+        }
+        pages = sortedPages.map((page) => ({
+            markdown: page.markdown,
+            tables: (page.tables ?? []).map((table) => ({
+                id: table.id,
+                content: table.content,
+                format: table.format,
+            })),
+            images: (page.images ?? []).map((image) => ({
+                id: image.id,
+                topLeftX: image.top_left_x,
+                topLeftY: image.top_left_y,
+                bottomRightX: image.bottom_right_x,
+                bottomRightY: image.bottom_right_y,
+            })),
+        }));
+    }
+    catch (error) {
+        operationError = error instanceof Error ? error : new Error(String(error));
+    }
+    const cleanupWarning = await deleteMistralFile({
+        apiKey: options.apiKey,
+        fileId,
+        fetchImpl: fetchFn,
+    });
+    if (operationError) {
+        if (cleanupWarning) {
+            throw new Error(`${operationError.message} (${cleanupWarning})`);
+        }
+        throw operationError;
+    }
+    if (!pages) {
+        throw new Error("Mistral OCR completed without pages");
+    }
+    return {
+        pages,
+        cleanupWarning,
+    };
+}
+//# sourceMappingURL=mistral_document_ocr.js.map

package/dist/core/utils/mistral_document_ocr.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"mistral_document_ocr.js","sourceRoot":"","sources":["../../../src/core/utils/mistral_document_ocr.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,CAAC,EAAE,MAAM,KAAK,CAAC;AAExB,MAAM,iBAAiB,GAAG,iCAAiC,CAAC;AAC5D,MAAM,wBAAwB,GAAG,+BAA+B,CAAC;AACjE,MAAM,kCAAkC,GAAG,oBAAoB,CAAC;AAChE,MAAM,kCAAkC,GAAG,iBAAiB,CAAC;AAE7D,MAAM,WAAW,GAAG,CAAC,CAAC,KAAK,CAAC;IAC1B,CAAC,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;IACjC,CAAC,CAAC,MAAM,CAAC,EAAE,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC;IAChC,CAAC,CAAC,MAAM,CAAC,EAAE,KAAK,EAAE,CAAC,CAAC,MAAM,CAAC,EAAE,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE,EAAE,CAAC,EAAE,CAAC;CACvD,CAAC,CAAC;AAEH,MAAM,mBAAmB,GAAG,CAAC,CAAC,MAAM,CAAC;IACnC,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,IAAI,EAAE;CACtB,CAAC,CAAC;AAEH,MAAM,WAAW,GAAG,CAAC;KAClB,MAAM,CAAC;IACN,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACrB,OAAO,EAAE,CAAC,CAAC,MAAM,EAAE;IACnB,MAAM,EAAE,CAAC,CAAC,MAAM,EAAE;CACnB,CAAC;KACD,WAAW,EAAE,CAAC;AAEjB,MAAM,WAAW,GAAG,CAAC;KAClB,MAAM,CAAC;IACN,EAAE,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,CAAC,CAAC,CAAC;IACrB,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;IAChE,UAAU,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;IAChE,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;IACpE,cAAc,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,WAAW,EAAE,CAAC,QAAQ,EAAE,CAAC,QAAQ,EAAE;CACrE,CAAC;KACD,WAAW,EAAE,CAAC;AAEjB,MAAM,aAAa,GAAG,CAAC,CAAC,MAAM,CAAC;IAC7B,KAAK,EAAE,CAAC;SACL,KAAK,CACJ,CAAC;SACE,MAAM,CAAC;QACN,KAAK,EAAE,CAAC,CAAC,MAAM,EAAE,CAAC,GAAG,EAAE,CAAC,WAAW,EAAE;QACrC,QAAQ,EAAE,CAAC,CAAC,MAAM,EAAE;QACpB,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE;QACvC,MAAM,EAAE,CAAC,CAAC,KAAK,CAAC,WAAW,CAAC,CAAC,QAAQ,EAAE;KACxC,CAAC;SACD,WAAW,EAAE,CACjB;SACA,GAAG,CAAC,CAAC,CAAC;CACV,CAAC,CAAC;AAoCH,SAAS,oBAAoB,CAAC,YAAoB;IAChD,IAAI,CAAC;QACH,OAAO,YAAY,CAAC,CAAC,CAAE,IAAI,CAAC,KAAK,CAAC,YAAY,CAAa,CAAC,CAAC,CAAC,SAAS,CAAC;IAC1E,CAAC;IAAC,MAAM,CAAC;QACP,OAAO,SAAS,CAAC;IACnB,CAAC;AACH,CAAC;AAED,SAAS,iBAAiB,CAAC,YAAoB,EAAE,OAAgB,EAAE,MAAc;IAC/E,MAAM,MAAM,GAAG,WAAW,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IAC9C,IAAI,MAAM,CAAC,OAAO,EAAE,CAAC;QACnB,IAAI,SAAS,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAC7B,OAAO,MAAM,CAAC,IAAI,CAAC,OAAO,CAAC;QAC7B,CAAC;QACD,IAAI,QAAQ,IAAI,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5B,OAAO,MAAM,CAAC,IAAI,CAAC,MAAM,CAAC;QAC5B,CAAC;QACD,OAAO,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC;IACnC,CAAC;IAED,OAAO,YAAY,CAAC,IAAI,EAAE,IAAI,QAAQ,MAAM,EAAE,CAAC;AACjD,CAAC;AAED,KAAK,UAAU,iBAAiB,CAAC,IAMhC;IACC,MAAM,QAAQ,GAAG,IAAI,QAAQ,EAAE,CAAC;IAChC,QAAQ,CAAC,MAAM,CAAC,SAAS,EAAE,KAAK,CAAC,CAAC;IAClC,QAAQ,CAAC,MAAM,CACb,MAAM,EACN,IAAI,IAAI,CAAC,CAAC,UAAU,CAAC,IAAI,CAAC,IAAI,CAAC,QAAQ,CAAC,CAAC,EAAE;QACzC,IAAI,EAAE,IAAI,CAAC,QAAQ;KACpB,CAAC,EACF,IAAI,CAAC,QAAQ,CACd,CAAC;IAEF,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,iBAAiB,EAAE;QACvD,MAAM,EAAE,MAAM;QACd,OAAO,EAAE;YACP,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;SACvC;QACD,IAAI,EAAE,QAAQ;KACf,CAAC,CAAC;IAEH,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;IAC3C,MAAM,OAAO,GAAG,oBAAoB,CAAC,YAAY,CAAC,CAAC;IACnD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;QACjB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,YAAY,EAAE,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;IAC7E,CAAC;IAED,MAAM,MAAM,GAAG,mBAAmB,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;IACtD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;QACpB,MAAM,IAAI,KAAK,CAAC,gDAAgD,CAAC,CAAC;IACpE,CAAC;IAED,OAAO,MAAM,CAAC,IAAI,CAAC,EAAE,CAAC;AACxB,CAAC;AAED,KAAK,UAAU,iBAAiB,CAAC,IAIhC;IACC,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,IAAI,CAAC,SAAS,CAAC,GAAG,iBAAiB,IAAI,IAAI,CAAC,MAAM,EAAE,EAAE;YAC3E,MAAM,EAAE,QAAQ;YAChB,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,IAAI,CAAC,MAAM,EAAE;aACvC;SACF,CAAC,CAAC;QAEH,IAAI,QAAQ,CAAC,EAAE,EAAE,CAAC;YAChB,OAAO,SAAS,CAAC;QACnB,CAAC;QAED,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC3C,MAAM,OAAO,GAAG,oBAAoB,CAAC,YAAY,CAAC,CAAC;QACnD,OAAO,mDAAmD,iBAAiB,CAAC,YAAY,EAAE,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,EAAE,CAAC;IACxH,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,OAAO,mDAAmD,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC;IACrH,CAAC;AACH,CAAC;AAED,MAAM,CAAC,KAAK,UAAU,kBAAkB,CACtC,OAAkC;IAElC,MAAM,OAAO,GAAG,OAAO,CAAC,SAAS,IAAI,KAAK,CAAC;IAC3C,MAAM,MAAM,GAAG,MAAM,iBAAiB,CAAC;QACrC,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ;QAC1B,QAAQ,EAAE,OAAO,CAAC,QAAQ,IAAI,kCAAkC;QAChE,SAAS,EAAE,OAAO;KACnB,CAAC,CAAC;IAEH,IAAI,KAA2C,CAAC;IAChD,IAAI,cAAiC,CAAC;IAEtC,IAAI,CAAC;QACH,MAAM,QAAQ,GAAG,MAAM,OAAO,CAAC,wBAAwB,EAAE;YACvD,MAAM,EAAE,MAAM;YACd,OAAO,EAAE;gBACP,aAAa,EAAE,UAAU,OAAO,CAAC,MAAM,EAAE;gBACzC,cAAc,EAAE,kBAAkB;aACnC;YACD,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC;gBACnB,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,kCAAkC;gBAC1D,QAAQ,EAAE;oBACR,IAAI,EAAE,MAAM;oBACZ,OAAO,EAAE,MAAM;iBAChB;gBACD,YAAY,EAAE,UAAU;aACzB,CAAC;SACH,CAAC,CAAC;QAEH,MAAM,YAAY,GAAG,MAAM,QAAQ,CAAC,IAAI,EAAE,CAAC;QAC3C,MAAM,OAAO,GAAG,oBAAoB,CAAC,YAAY,CAAC,CAAC;QACnD,IAAI,CAAC,QAAQ,CAAC,EAAE,EAAE,CAAC;YACjB,MAAM,IAAI,KAAK,CAAC,iBAAiB,CAAC,YAAY,EAAE,OAAO,EAAE,QAAQ,CAAC,MAAM,CAAC,CAAC,CAAC;QAC7E,CAAC;QAED,MAAM,MAAM,GAAG,aAAa,CAAC,SAAS,CAAC,OAAO,CAAC,CAAC;QAChD,IAAI,CAAC,MAAM,CAAC,OAAO,EAAE,CAAC;YACpB,MAAM,IAAI,KAAK,CAAC,wCAAwC,CAAC,CAAC;QAC5D,CAAC;QAED,MAAM,WAAW,GAAG,CAAC,GAAG,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,CAAC,KAAK,GAAG,CAAC,CAAC,KAAK,CAAC,CAAC;QAC7E,KAAK,MAAM,CAAC,aAAa,EAAE,IAAI,CAAC,IAAI,WAAW,CAAC,OAAO,EAAE,EAAE,CAAC;YAC1D,IAAI,IAAI,CAAC,KAAK,KAAK,aAAa,EAAE,CAAC;gBACjC,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;YACzD,CAAC;QACH,CAAC;QAED,KAAK,GAAG,WAAW,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,EAAE,CAAC,CAAC;YACjC,QAAQ,EAAE,IAAI,CAAC,QAAQ;YACvB,MAAM,EAAE,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAC1C,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,OAAO,EAAE,KAAK,CAAC,OAAO;gBACtB,MAAM,EAAE,KAAK,CAAC,MAAM;aACrB,CAAC,CAAC;YACH,MAAM,EAAE,CAAC,IAAI,CAAC,MAAM,IAAI,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,KAAK,EAAE,EAAE,CAAC,CAAC;gBAC1C,EAAE,EAAE,KAAK,CAAC,EAAE;gBACZ,QAAQ,EAAE,KAAK,CAAC,UAAU;gBAC1B,QAAQ,EAAE,KAAK,CAAC,UAAU;gBAC1B,YAAY,EAAE,KAAK,CAAC,cAAc;gBAClC,YAAY,EAAE,KAAK,CAAC,cAAc;aACnC,CAAC,CAAC;SACJ,CAAC,CAAC,CAAC;IACN,CAAC;IAAC,OAAO,KAAK,EAAE,CAAC;QACf,cAAc,GAAG,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,CAAC,CAAC,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC,CAAC;IAC7E,CAAC;IAED,MAAM,cAAc,GAAG,MAAM,iBAAiB,CAAC;QAC7C,MAAM,EAAE,OAAO,CAAC,MAAM;QACtB,MAAM;QACN,SAAS,EAAE,OAAO;KACnB,CAAC,CAAC;IAEH,IAAI,cAAc,EAAE,CAAC;QACnB,IAAI,cAAc,EAAE,CAAC;YACnB,MAAM,IAAI,KAAK,CAAC,GAAG,cAAc,CAAC,OAAO,KAAK,cAAc,GAAG,CAAC,CAAC;QACnE,CAAC;QACD,MAAM,cAAc,CAAC;IACvB,CAAC;IAED,IAAI,CAAC,KAAK,EAAE,CAAC;QACX,MAAM,IAAI,KAAK,CAAC,qCAAqC,CAAC,CAAC;IACzD,CAAC;IAED,OAAO;QACL,KAAK;QACL,cAAc;KACf,CAAC;AACJ,CAAC"}

package/dist/core/version.js

@@ -1,2 +1,2 @@
-export const APP_VERSION = "0.2.101";
+export const APP_VERSION = "0.2.103";
 //# sourceMappingURL=version.js.map