aiwrapper/dist/process-response-stream.js

A Universal AI Wrapper for JavaScript & TypeScript

import processLinesFromStream from "./lang/process-lines-from-stream.js";
// This would work only in Deno and browsers, not in Node.
let _processResponseStream = (response, onData) => {
    if (response.ok === false) {
        throw new Error(`Response from server was not ok. Status code: ${response.status}.`);
    }
    const reader = response.body.getReader();
    let decoder = new TextDecoder("utf-8");
    let rawData = "";
    return reader.read().then(function processStream(result) {
        if (result.done || result.value === undefined) {
            return Promise.resolve();
        }
        rawData += decoder.decode(result.value, {
            stream: true,
        });
        // Process each complete message (messages are devived by newlines)
        let lastIndex = rawData.lastIndexOf("\n");
        if (lastIndex > -1) {
            processLinesFromStream(rawData.slice(0, lastIndex), onData);
            rawData = rawData.slice(lastIndex + 1);
        }
        return reader.read().then(processStream);
    });
};
export const setProcessResponseStreamImpl = (impl) => {
    _processResponseStream = impl;
};
export const processResponseStream = (response, onProgress) => {
    return _processResponseStream(response, onProgress);
};
//# sourceMappingURL=process-response-stream.js.map