opencode-rehydar 0.14.1 → 0.14.3

package/dist/proxy/providers/streaming-tool-loop.js

@@ -0,0 +1,223 @@
+function object(value) {
+    if (value === null || typeof value !== 'object' || Array.isArray(value)) {
+        throw new Error('Invalid streaming response object');
+    }
+    return value;
+}
+function index(value) {
+    if (typeof value !== 'number' || !Number.isSafeInteger(value) || value < 0) {
+        throw new Error('Invalid streaming response index');
+    }
+    return value;
+}
+function append(target, key, value) {
+    if (typeof value === 'string')
+        target[key] = String(target[key] ?? '') + value;
+}
+function event(data, type = 'message') {
+    return { event: type, data: JSON.stringify(data) };
+}
+function validateTool(id, name, args) {
+    if (typeof id !== 'string' || id === '' || typeof name !== 'string' || name === '') {
+        throw new Error('Incomplete streamed tool call');
+    }
+    object(JSON.parse(args));
+}
+export const openAIStreamingToolLoop = {
+    createAccumulator() {
+        const metadata = {};
+        const message = { role: 'assistant', content: '' };
+        const calls = new Map();
+        let finishReason;
+        return {
+            push(data) {
+                const chunk = object(data);
+                if (chunk.error !== undefined)
+                    throw new Error('Upstream streaming error');
+                for (const key of ['id', 'model', 'created', 'system_fingerprint', 'usage']) {
+                    if (chunk[key] !== undefined)
+                        metadata[key] = chunk[key];
+                }
+                if (!Array.isArray(chunk.choices))
+                    throw new Error('Missing streaming choices');
+                for (const value of chunk.choices) {
+                    const choice = object(value);
+                    if (index(choice.index) !== 0)
+                        throw new Error('Streaming tool loops require a single completion choice');
+                    const delta = object(choice.delta);
+                    if (choice.finish_reason !== undefined && choice.finish_reason !== null)
+                        finishReason = choice.finish_reason;
+                    for (const key of ['content', 'refusal', 'reasoning_content'])
+                        append(message, key, delta[key]);
+                    if (Array.isArray(delta.tool_calls))
+                        for (const item of delta.tool_calls) {
+                            const fragment = object(item);
+                            const i = index(fragment.index);
+                            const call = calls.get(i) ?? { id: '', type: 'function', function: { name: '', arguments: '' } };
+                            append(call, 'id', fragment.id);
+                            if (fragment.function !== undefined) {
+                                const fn = object(fragment.function);
+                                append(call.function, 'name', fn.name);
+                                append(call.function, 'arguments', fn.arguments);
+                            }
+                            calls.set(i, call);
+                        }
+                }
+            },
+            finish() {
+                if (finishReason === undefined)
+                    throw new Error('Truncated streaming completion');
+                if (calls.size > 0) {
+                    if (finishReason !== 'tool_calls')
+                        throw new Error('Streamed tool calls did not complete');
+                    const tools = [...calls].sort(([a], [b]) => a - b).map(([, call]) => call);
+                    if (new Set(tools.map(call => call.id)).size !== tools.length)
+                        throw new Error('Duplicate streamed tool call ID');
+                    for (const call of tools)
+                        validateTool(call.id, call.function.name, call.function.arguments);
+                    message.tool_calls = tools;
+                }
+                return { ...metadata, object: 'chat.completion', choices: [{ index: 0, message, finish_reason: finishReason }] };
+            },
+        };
+    },
+    *encode(body) {
+        const response = object(body);
+        if (!Array.isArray(response.choices))
+            throw new Error('Missing completion choices');
+        const metadata = { ...response, object: 'chat.completion.chunk', usage: undefined };
+        for (const value of response.choices) {
+            const choice = object(value);
+            const message = object(choice.message);
+            const delta = { ...message };
+            if (Array.isArray(message.tool_calls))
+                delta.tool_calls = message.tool_calls.map((call, i) => ({ ...object(call), index: i }));
+            yield event({ ...metadata, choices: [{ index: choice.index ?? 0, delta, finish_reason: null }] });
+            yield event({ ...metadata, choices: [{ index: choice.index ?? 0, delta: {}, finish_reason: choice.finish_reason }] });
+        }
+        if (response.usage !== undefined)
+            yield event({ ...metadata, choices: [], usage: response.usage });
+        yield { event: 'message', data: '[DONE]' };
+    },
+};
+export const anthropicStreamingToolLoop = {
+    createAccumulator() {
+        let message;
+        const blocks = new Map();
+        const args = new Map();
+        const stoppedBlocks = new Set();
+        let stopped = false;
+        return {
+            push(data) {
+                const value = object(data);
+                if (value.type === 'error')
+                    throw new Error('Upstream streaming error');
+                switch (value.type) {
+                    case 'message_start':
+                        if (message !== undefined)
+                            throw new Error('Duplicate message start');
+                        message = { ...object(value.message) };
+                        break;
+                    case 'content_block_start': {
+                        const i = index(value.index);
+                        if (blocks.has(i))
+                            throw new Error('Duplicate content block');
+                        blocks.set(i, { ...object(value.content_block) });
+                        break;
+                    }
+                    case 'content_block_delta': {
+                        const i = index(value.index);
+                        const block = blocks.get(i);
+                        if (block === undefined || stoppedBlocks.has(i))
+                            throw new Error('Unexpected content delta');
+                        const delta = object(value.delta);
+                        if (delta.type === 'input_json_delta') {
+                            if (typeof delta.partial_json !== 'string')
+                                throw new Error('Invalid tool argument delta');
+                            args.set(i, (args.get(i) ?? '') + delta.partial_json);
+                        }
+                        else if (delta.type === 'text_delta')
+                            append(block, 'text', delta.text);
+                        else if (delta.type === 'thinking_delta')
+                            append(block, 'thinking', delta.thinking);
+                        else if (delta.type === 'signature_delta')
+                            append(block, 'signature', delta.signature);
+                        else
+                            throw new Error('Unsupported content delta in streaming tool loop');
+                        break;
+                    }
+                    case 'content_block_stop': {
+                        const i = index(value.index);
+                        if (!blocks.has(i) || stoppedBlocks.has(i))
+                            throw new Error('Unexpected content block stop');
+                        stoppedBlocks.add(i);
+                        break;
+                    }
+                    case 'message_delta':
+                        if (message === undefined)
+                            throw new Error('Missing message start');
+                        Object.assign(message, object(value.delta));
+                        if (value.usage !== undefined)
+                            message.usage = { ...object(message.usage ?? {}), ...object(value.usage) };
+                        break;
+                    case 'message_stop':
+                        stopped = true;
+                        break;
+                }
+            },
+            finish() {
+                if (message === undefined || !stopped)
+                    throw new Error('Truncated streaming message');
+                const content = [...blocks].sort(([a], [b]) => a - b).map(([i, block]) => {
+                    if (!stoppedBlocks.has(i))
+                        throw new Error('Unfinished content block');
+                    if (block.type === 'tool_use') {
+                        if (message.stop_reason !== 'tool_use')
+                            throw new Error('Streamed tool calls did not complete');
+                        const json = args.get(i) ?? JSON.stringify(block.input ?? {});
+                        validateTool(block.id, block.name, json);
+                        block.input = JSON.parse(json);
+                    }
+                    return block;
+                });
+                const ids = content.filter(block => block.type === 'tool_use').map(block => block.id);
+                if (new Set(ids).size !== ids.length)
+                    throw new Error('Duplicate streamed tool call ID');
+                return { ...message, content };
+            },
+        };
+    },
+    *encode(body) {
+        const response = object(body);
+        if (!Array.isArray(response.content))
+            throw new Error('Missing message content');
+        yield event({ type: 'message_start', message: { ...response, content: [], stop_reason: null, stop_sequence: null } }, 'message_start');
+        for (const [i, item] of response.content.entries()) {
+            const block = object(item);
+            const start = { ...block };
+            const deltas = [];
+            if (block.type === 'text') {
+                start.text = '';
+                deltas.push({ type: 'text_delta', text: block.text });
+            }
+            if (block.type === 'tool_use') {
+                start.input = {};
+                deltas.push({ type: 'input_json_delta', partial_json: JSON.stringify(block.input) });
+            }
+            if (block.type === 'thinking') {
+                start.thinking = '';
+                start.signature = '';
+                deltas.push({ type: 'thinking_delta', thinking: block.thinking });
+                if (typeof block.signature === 'string')
+                    deltas.push({ type: 'signature_delta', signature: block.signature });
+            }
+            yield event({ type: 'content_block_start', index: i, content_block: start }, 'content_block_start');
+            for (const delta of deltas)
+                yield event({ type: 'content_block_delta', index: i, delta }, 'content_block_delta');
+            yield event({ type: 'content_block_stop', index: i }, 'content_block_stop');
+        }
+        yield event({ type: 'message_delta', delta: { stop_reason: response.stop_reason, stop_sequence: response.stop_sequence }, usage: response.usage ?? {} }, 'message_delta');
+        yield event({ type: 'message_stop' }, 'message_stop');
+    },
+};
+//# sourceMappingURL=streaming-tool-loop.js.map

package/dist/proxy/providers/streaming-tool-loop.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"streaming-tool-loop.js","sourceRoot":"","sources":["../../../src/proxy/providers/streaming-tool-loop.ts"],"names":[],"mappings":"AASA,SAAS,MAAM,CAAC,KAAc;IAC5B,IAAI,KAAK,KAAK,IAAI,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;QACxE,MAAM,IAAI,KAAK,CAAC,mCAAmC,CAAC,CAAC;IACvD,CAAC;IACD,OAAO,KAAoB,CAAC;AAC9B,CAAC;AACD,SAAS,KAAK,CAAC,KAAc;IAC3B,IAAI,OAAO,KAAK,KAAK,QAAQ,IAAI,CAAC,MAAM,CAAC,aAAa,CAAC,KAAK,CAAC,IAAI,KAAK,GAAG,CAAC,EAAE,CAAC;QAC3E,MAAM,IAAI,KAAK,CAAC,kCAAkC,CAAC,CAAC;IACtD,CAAC;IACD,OAAO,KAAK,CAAC;AACf,CAAC;AACD,SAAS,MAAM,CAAC,MAAmB,EAAE,GAAW,EAAE,KAAc;IAC9D,IAAI,OAAO,KAAK,KAAK,QAAQ;QAAE,MAAM,CAAC,GAAG,CAAC,GAAG,MAAM,CAAC,MAAM,CAAC,GAAG,CAAC,IAAI,EAAE,CAAC,GAAG,KAAK,CAAC;AACjF,CAAC;AACD,SAAS,KAAK,CAAC,IAAa,EAAE,IAAI,GAAG,SAAS;IAC5C,OAAO,EAAE,KAAK,EAAE,IAAI,EAAE,IAAI,EAAE,IAAI,CAAC,SAAS,CAAC,IAAI,CAAC,EAAE,CAAC;AACrD,CAAC;AACD,SAAS,YAAY,CAAC,EAAW,EAAE,IAAa,EAAE,IAAY;IAC5D,IAAI,OAAO,EAAE,KAAK,QAAQ,IAAI,EAAE,KAAK,EAAE,IAAI,OAAO,IAAI,KAAK,QAAQ,IAAI,IAAI,KAAK,EAAE,EAAE,CAAC;QACnF,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;IACnD,CAAC;IACD,MAAM,CAAC,IAAI,CAAC,KAAK,CAAC,IAAI,CAAC,CAAC,CAAC;AAC3B,CAAC;AAED,MAAM,CAAC,MAAM,uBAAuB,GAA6B;IAC/D,iBAAiB;QACf,MAAM,QAAQ,GAAgB,EAAE,CAAC;QACjC,MAAM,OAAO,GAAgB,EAAE,IAAI,EAAE,WAAW,EAAE,OAAO,EAAE,EAAE,EAAE,CAAC;QAChE,MAAM,KAAK,GAAG,IAAI,GAAG,EAAuF,CAAC;QAC7G,IAAI,YAAqB,CAAC;QAC1B,OAAO;YACL,IAAI,CAAC,IAAI;gBACP,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC3B,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS;oBAAE,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;gBAC3E,KAAK,MAAM,GAAG,IAAI,CAAC,IAAI,EAAE,OAAO,EAAE,SAAS,EAAE,oBAAoB,EAAE,OAAO,CAAC,EAAE,CAAC;oBAC5E,IAAI,KAAK,CAAC,GAAG,CAAC,KAAK,SAAS;wBAAE,QAAQ,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC;gBAC3D,CAAC;gBACD,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,OAAO,CAAC;oBAAE,MAAM,IAAI,KAAK,CAAC,2BAA2B,CAAC,CAAC;gBAChF,KAAK,MAAM,KAAK,IAAI,KAAK,CAAC,OAAO,EAAE,CAAC;oBAClC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;oBAC7B,IAAI,KAAK,CAAC,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC;wBAAE,MAAM,IAAI,KAAK,CAAC,yDAAyD,CAAC,CAAC;oBAC1G,MAAM,KAAK,GAAG,MAAM,CAAC,MAAM,CAAC,KAAK,CAAC,CAAC;oBACnC,IAAI,MAAM,CAAC,aAAa,KAAK,SAAS,IAAI,MAAM,CAAC,aAAa,KAAK,IAAI;wBAAE,YAAY,GAAG,MAAM,CAAC,aAAa,CAAC;oBAC7G,KAAK,MAAM,GAAG,IAAI,CAAC,SAAS,EAAE,SAAS,EAAE,mBAAmB,CAAC;wBAAE,MAAM,CAAC,OAAO,EAAE,GAAG,EAAE,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC;oBAChG,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,UAAU,CAAC;wBAAE,KAAK,MAAM,IAAI,IAAI,KAAK,CAAC,UAAU,EAAE,CAAC;4BACzE,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;4BAC9B,MAAM,CAAC,GAAG,KAAK,CAAC,QAAQ,CAAC,KAAK,CAAC,CAAC;4BAChC,MAAM,IAAI,GAAG,KAAK,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,EAAE,EAAE,EAAE,EAAE,IAAI,EAAE,UAAU,EAAE,QAAQ,EAAE,EAAE,IAAI,EAAE,EAAE,EAAE,SAAS,EAAE,EAAE,EAAE,EAAE,CAAC;4BACjG,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,QAAQ,CAAC,EAAE,CAAC,CAAC;4BAChC,IAAI,QAAQ,CAAC,QAAQ,KAAK,SAAS,EAAE,CAAC;gCACpC,MAAM,EAAE,GAAG,MAAM,CAAC,QAAQ,CAAC,QAAQ,CAAC,CAAC;gCACrC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,MAAM,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;gCACvC,MAAM,CAAC,IAAI,CAAC,QAAQ,EAAE,WAAW,EAAE,EAAE,CAAC,SAAS,CAAC,CAAC;4BACnD,CAAC;4BACD,KAAK,CAAC,GAAG,CAAC,CAAC,EAAE,IAAI,CAAC,CAAC;wBACrB,CAAC;gBACH,CAAC;YACH,CAAC;YACD,MAAM;gBACJ,IAAI,YAAY,KAAK,SAAS;oBAAE,MAAM,IAAI,KAAK,CAAC,gCAAgC,CAAC,CAAC;gBAClF,IAAI,KAAK,CAAC,IAAI,GAAG,CAAC,EAAE,CAAC;oBACnB,IAAI,YAAY,KAAK,YAAY;wBAAE,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;oBAC3F,MAAM,KAAK,GAAG,CAAC,GAAG,KAAK,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,EAAE,IAAI,CAAC,EAAE,EAAE,CAAC,IAAI,CAAC,CAAC;oBAC3E,IAAI,IAAI,GAAG,CAAC,KAAK,CAAC,GAAG,CAAC,IAAI,CAAC,EAAE,CAAC,IAAI,CAAC,EAAE,CAAC,CAAC,CAAC,IAAI,KAAK,KAAK,CAAC,MAAM;wBAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;oBAClH,KAAK,MAAM,IAAI,IAAI,KAAK;wBAAE,YAAY,CAAC,IAAI,CAAC,EAAE,EAAE,IAAI,CAAC,QAAQ,CAAC,IAAI,EAAE,IAAI,CAAC,QAAQ,CAAC,SAAS,CAAC,CAAC;oBAC7F,OAAO,CAAC,UAAU,GAAG,KAAK,CAAC;gBAC7B,CAAC;gBACD,OAAO,EAAE,GAAG,QAAQ,EAAE,MAAM,EAAE,iBAAiB,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,OAAO,EAAE,aAAa,EAAE,YAAY,EAAE,CAAC,EAAE,CAAC;YACnH,CAAC;SACF,CAAC;IACJ,CAAC;IACD,CAAC,MAAM,CAAC,IAAI;QACV,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,4BAA4B,CAAC,CAAC;QACpF,MAAM,QAAQ,GAAG,EAAE,GAAG,QAAQ,EAAE,MAAM,EAAE,uBAAuB,EAAE,KAAK,EAAE,SAAS,EAAE,CAAC;QACpF,KAAK,MAAM,KAAK,IAAI,QAAQ,CAAC,OAAO,EAAE,CAAC;YACrC,MAAM,MAAM,GAAG,MAAM,CAAC,KAAK,CAAC,CAAC;YAC7B,MAAM,OAAO,GAAG,MAAM,CAAC,MAAM,CAAC,OAAO,CAAC,CAAC;YACvC,MAAM,KAAK,GAAG,EAAE,GAAG,OAAO,EAAE,CAAC;YAC7B,IAAI,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC;gBAAE,KAAK,CAAC,UAAU,GAAG,OAAO,CAAC,UAAU,CAAC,GAAG,CAAC,CAAC,IAAI,EAAE,CAAC,EAAE,EAAE,CAAC,CAAC,EAAE,GAAG,MAAM,CAAC,IAAI,CAAC,EAAE,KAAK,EAAE,CAAC,EAAE,CAAC,CAAC,CAAC;YAC/H,MAAM,KAAK,CAAC,EAAE,GAAG,QAAQ,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,CAAC,EAAE,KAAK,EAAE,aAAa,EAAE,IAAI,EAAE,CAAC,EAAE,CAAC,CAAC;YAClG,MAAM,KAAK,CAAC,EAAE,GAAG,QAAQ,EAAE,OAAO,EAAE,CAAC,EAAE,KAAK,EAAE,MAAM,CAAC,KAAK,IAAI,CAAC,EAAE,KAAK,EAAE,EAAE,EAAE,aAAa,EAAE,MAAM,CAAC,aAAa,EAAE,CAAC,EAAE,CAAC,CAAC;QACxH,CAAC;QACD,IAAI,QAAQ,CAAC,KAAK,KAAK,SAAS;YAAE,MAAM,KAAK,CAAC,EAAE,GAAG,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,EAAE,CAAC,CAAC;QACnG,MAAM,EAAE,KAAK,EAAE,SAAS,EAAE,IAAI,EAAE,QAAQ,EAAE,CAAC;IAC7C,CAAC;CACF,CAAC;AAEF,MAAM,CAAC,MAAM,0BAA0B,GAA6B;IAClE,iBAAiB;QACf,IAAI,OAAgC,CAAC;QACrC,MAAM,MAAM,GAAG,IAAI,GAAG,EAAuB,CAAC;QAC9C,MAAM,IAAI,GAAG,IAAI,GAAG,EAAkB,CAAC;QACvC,MAAM,aAAa,GAAG,IAAI,GAAG,EAAU,CAAC;QACxC,IAAI,OAAO,GAAG,KAAK,CAAC;QACpB,OAAO;YACL,IAAI,CAAC,IAAI;gBACP,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;gBAC3B,IAAI,KAAK,CAAC,IAAI,KAAK,OAAO;oBAAE,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;gBACxE,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;oBACnB,KAAK,eAAe;wBAClB,IAAI,OAAO,KAAK,SAAS;4BAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;wBACtE,OAAO,GAAG,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,OAAO,CAAC,EAAE,CAAC;wBAAC,MAAM;oBAChD,KAAK,qBAAqB,CAAC,CAAC,CAAC;wBAC3B,MAAM,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;wBAC7B,IAAI,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC;4BAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;wBAC9D,MAAM,CAAC,GAAG,CAAC,CAAC,EAAE,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,aAAa,CAAC,EAAE,CAAC,CAAC;wBAClD,MAAM;oBACR,CAAC;oBACD,KAAK,qBAAqB,CAAC,CAAC,CAAC;wBAC3B,MAAM,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;wBAC7B,MAAM,KAAK,GAAG,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;wBAC5B,IAAI,KAAK,KAAK,SAAS,IAAI,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC;4BAAE,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;wBAC7F,MAAM,KAAK,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;wBAClC,IAAI,KAAK,CAAC,IAAI,KAAK,kBAAkB,EAAE,CAAC;4BACtC,IAAI,OAAO,KAAK,CAAC,YAAY,KAAK,QAAQ;gCAAE,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;4BAC3F,IAAI,CAAC,GAAG,CAAC,CAAC,EAAE,CAAC,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,EAAE,CAAC,GAAG,KAAK,CAAC,YAAY,CAAC,CAAC;wBACxD,CAAC;6BAAM,IAAI,KAAK,CAAC,IAAI,KAAK,YAAY;4BAAE,MAAM,CAAC,KAAK,EAAE,MAAM,EAAE,KAAK,CAAC,IAAI,CAAC,CAAC;6BACrE,IAAI,KAAK,CAAC,IAAI,KAAK,gBAAgB;4BAAE,MAAM,CAAC,KAAK,EAAE,UAAU,EAAE,KAAK,CAAC,QAAQ,CAAC,CAAC;6BAC/E,IAAI,KAAK,CAAC,IAAI,KAAK,iBAAiB;4BAAE,MAAM,CAAC,KAAK,EAAE,WAAW,EAAE,KAAK,CAAC,SAAS,CAAC,CAAC;;4BAClF,MAAM,IAAI,KAAK,CAAC,kDAAkD,CAAC,CAAC;wBACzE,MAAM;oBACR,CAAC;oBACD,KAAK,oBAAoB,CAAC,CAAC,CAAC;wBAC1B,MAAM,CAAC,GAAG,KAAK,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;wBAC7B,IAAI,CAAC,MAAM,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC;4BAAE,MAAM,IAAI,KAAK,CAAC,+BAA+B,CAAC,CAAC;wBAC7F,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC;wBAAC,MAAM;oBAC9B,CAAC;oBACD,KAAK,eAAe;wBAClB,IAAI,OAAO,KAAK,SAAS;4BAAE,MAAM,IAAI,KAAK,CAAC,uBAAuB,CAAC,CAAC;wBACpE,MAAM,CAAC,MAAM,CAAC,OAAO,EAAE,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC,CAAC;wBAC5C,IAAI,KAAK,CAAC,KAAK,KAAK,SAAS;4BAAE,OAAO,CAAC,KAAK,GAAG,EAAE,GAAG,MAAM,CAAC,OAAO,CAAC,KAAK,IAAI,EAAE,CAAC,EAAE,GAAG,MAAM,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC;wBAC1G,MAAM;oBACR,KAAK,cAAc;wBAAE,OAAO,GAAG,IAAI,CAAC;wBAAC,MAAM;gBAC7C,CAAC;YACH,CAAC;YACD,MAAM;gBACJ,IAAI,OAAO,KAAK,SAAS,IAAI,CAAC,OAAO;oBAAE,MAAM,IAAI,KAAK,CAAC,6BAA6B,CAAC,CAAC;gBACtF,MAAM,OAAO,GAAG,CAAC,GAAG,MAAM,CAAC,CAAC,IAAI,CAAC,CAAC,CAAC,CAAC,CAAC,EAAE,CAAC,CAAC,CAAC,EAAE,EAAE,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,EAAE,KAAK,CAAC,EAAE,EAAE;oBACvE,IAAI,CAAC,aAAa,CAAC,GAAG,CAAC,CAAC,CAAC;wBAAE,MAAM,IAAI,KAAK,CAAC,0BAA0B,CAAC,CAAC;oBACvE,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;wBAC9B,IAAI,OAAQ,CAAC,WAAW,KAAK,UAAU;4BAAE,MAAM,IAAI,KAAK,CAAC,sCAAsC,CAAC,CAAC;wBACjG,MAAM,IAAI,GAAG,IAAI,CAAC,GAAG,CAAC,CAAC,CAAC,IAAI,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,IAAI,EAAE,CAAC,CAAC;wBAC9D,YAAY,CAAC,KAAK,CAAC,EAAE,EAAE,KAAK,CAAC,IAAI,EAAE,IAAI,CAAC,CAAC;wBACzC,KAAK,CAAC,KAAK,GAAG,IAAI,CAAC,KAAK,CAAC,IAAI,CAAY,CAAC;oBAC5C,CAAC;oBACD,OAAO,KAAK,CAAC;gBACf,CAAC,CAAC,CAAC;gBACH,MAAM,GAAG,GAAG,OAAO,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,IAAI,KAAK,UAAU,CAAC,CAAC,GAAG,CAAC,KAAK,CAAC,EAAE,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;gBACtF,IAAI,IAAI,GAAG,CAAC,GAAG,CAAC,CAAC,IAAI,KAAK,GAAG,CAAC,MAAM;oBAAE,MAAM,IAAI,KAAK,CAAC,iCAAiC,CAAC,CAAC;gBACzF,OAAO,EAAE,GAAG,OAAO,EAAE,OAAO,EAAE,CAAC;YACjC,CAAC;SACF,CAAC;IACJ,CAAC;IACD,CAAC,MAAM,CAAC,IAAI;QACV,MAAM,QAAQ,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;QAC9B,IAAI,CAAC,KAAK,CAAC,OAAO,CAAC,QAAQ,CAAC,OAAO,CAAC;YAAE,MAAM,IAAI,KAAK,CAAC,yBAAyB,CAAC,CAAC;QACjF,MAAM,KAAK,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,OAAO,EAAE,EAAE,GAAG,QAAQ,EAAE,OAAO,EAAE,EAAE,EAAE,WAAW,EAAE,IAAI,EAAE,aAAa,EAAE,IAAI,EAAE,EAAE,EAAE,eAAe,CAAC,CAAC;QACvI,KAAK,MAAM,CAAC,CAAC,EAAE,IAAI,CAAC,IAAI,QAAQ,CAAC,OAAO,CAAC,OAAO,EAAE,EAAE,CAAC;YACnD,MAAM,KAAK,GAAG,MAAM,CAAC,IAAI,CAAC,CAAC;YAC3B,MAAM,KAAK,GAAG,EAAE,GAAG,KAAK,EAAE,CAAC;YAC3B,MAAM,MAAM,GAAkB,EAAE,CAAC;YACjC,IAAI,KAAK,CAAC,IAAI,KAAK,MAAM,EAAE,CAAC;gBAAC,KAAK,CAAC,IAAI,GAAG,EAAE,CAAC;gBAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,YAAY,EAAE,IAAI,EAAE,KAAK,CAAC,IAAI,EAAE,CAAC,CAAC;YAAC,CAAC;YACtG,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAAC,KAAK,CAAC,KAAK,GAAG,EAAE,CAAC;gBAAC,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,kBAAkB,EAAE,YAAY,EAAE,IAAI,CAAC,SAAS,CAAC,KAAK,CAAC,KAAK,CAAC,EAAE,CAAC,CAAC;YAAC,CAAC;YAC1I,IAAI,KAAK,CAAC,IAAI,KAAK,UAAU,EAAE,CAAC;gBAC9B,KAAK,CAAC,QAAQ,GAAG,EAAE,CAAC;gBAAC,KAAK,CAAC,SAAS,GAAG,EAAE,CAAC;gBAC1C,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,gBAAgB,EAAE,QAAQ,EAAE,KAAK,CAAC,QAAQ,EAAE,CAAC,CAAC;gBAClE,IAAI,OAAO,KAAK,CAAC,SAAS,KAAK,QAAQ;oBAAE,MAAM,CAAC,IAAI,CAAC,EAAE,IAAI,EAAE,iBAAiB,EAAE,SAAS,EAAE,KAAK,CAAC,SAAS,EAAE,CAAC,CAAC;YAChH,CAAC;YACD,MAAM,KAAK,CAAC,EAAE,IAAI,EAAE,qBAAqB,EAAE,KAAK,EAAE,CAAC,EAAE,aAAa,EAAE,KAAK,EAAE,EAAE,qBAAqB,CAAC,CAAC;YACpG,KAAK,MAAM,KAAK,IAAI,MAAM;gBAAE,MAAM,KAAK,CAAC,EAAE,IAAI,EAAE,qBAAqB,EAAE,KAAK,EAAE,CAAC,EAAE,KAAK,EAAE,EAAE,qBAAqB,CAAC,CAAC;YACjH,MAAM,KAAK,CAAC,EAAE,IAAI,EAAE,oBAAoB,EAAE,KAAK,EAAE,CAAC,EAAE,EAAE,oBAAoB,CAAC,CAAC;QAC9E,CAAC;QACD,MAAM,KAAK,CAAC,EAAE,IAAI,EAAE,eAAe,EAAE,KAAK,EAAE,EAAE,WAAW,EAAE,QAAQ,CAAC,WAAW,EAAE,aAAa,EAAE,QAAQ,CAAC,aAAa,EAAE,EAAE,KAAK,EAAE,QAAQ,CAAC,KAAK,IAAI,EAAE,EAAE,EAAE,eAAe,CAAC,CAAC;QAC1K,MAAM,KAAK,CAAC,EAAE,IAAI,EAAE,cAAc,EAAE,EAAE,cAAc,CAAC,CAAC;IACxD,CAAC;CACF,CAAC"}

package/dist/proxy/providers/types.d.ts

@@ -0,0 +1,109 @@
+import type { StreamingToolLoopAdapter } from './streaming-tool-loop.js';
+/**
+ * LLM Content Provider Interface
+ * Abstracts LLM-specific request/response formats for the proxy middleware.
+ */
+/**
+ * Structured tool call information extracted from a non-streaming response.
+ * Used by the tool execution loop to invoke user callbacks.
+ */
+export interface ToolCallInfo {
+    /** Tool call ID (e.g., "call_abc123" for OpenAI, "toolu_01A..." for Anthropic) */
+    id: string;
+    /** Function/tool name */
+    name: string;
+    /** Raw arguments as JSON string (in anonymized form from LLM) */
+    arguments: string;
+}
+/**
+ * A tool result to be sent back to the LLM in the next round.
+ */
+export interface ToolResultMessage {
+    /** The tool call ID this result corresponds to */
+    toolCallId: string;
+    /** The tool result content (JSON-stringified, anonymized) */
+    content: string;
+}
+/**
+ * A tool call argument fragment from an SSE chunk.
+ * Keyed by index to support interleaved multi-tool streaming.
+ */
+export interface ToolCallDelta {
+    /** Numeric index identifying which tool call this fragment belongs to */
+    index: number;
+    /** The partial argument string fragment */
+    arguments: string;
+}
+/**
+ * Interface for extracting and rebuilding text content from LLM API requests/responses.
+ * Each provider handles a specific LLM API format (OpenAI, Anthropic, etc.).
+ */
+export interface LLMContentProvider {
+    /** Provider name (e.g., "openai", "anthropic") */
+    readonly name: string;
+    /** Optional SSE conversion for server-side tool execution loops. */
+    readonly streamingToolLoop?: StreamingToolLoopAdapter;
+    /** Detect if a request matches this provider based on URL/headers */
+    matchesRequest(url: string, headers: Headers): boolean;
+    /** Extract text strings from a request body for anonymization */
+    extractRequestText(body: unknown): string[];
+    /** Rebuild the request body with anonymized text (same order as extractRequestText) */
+    rebuildRequestBody(body: unknown, anonymizedTexts: string[]): unknown;
+    /** Extract text strings from a non-streaming response body for rehydration */
+    extractResponseText(body: unknown): string[];
+    /** Rebuild the response body with rehydrated text (same order as extractResponseText) */
+    rebuildResponseBody(body: unknown, rehydratedTexts: string[]): unknown;
+    /** Extract text delta from a parsed SSE data payload (returns null if no text content) */
+    extractSSEDelta(data: unknown): string | null;
+    /** Rebuild an SSE data payload with rehydrated text */
+    rebuildSSEDelta(data: unknown, rehydratedText: string): unknown;
+    /** Check if the request body indicates a streaming response is expected */
+    isStreamingRequest(body: unknown): boolean;
+    /** Extract tool call argument strings from a non-streaming response body for rehydration */
+    extractResponseToolCalls?(body: unknown): string[];
+    /** Rebuild the response body with rehydrated tool call arguments (same order as extractResponseToolCalls) */
+    rebuildResponseToolCalls?(body: unknown, rehydratedArgs: string[]): unknown;
+    /** Extract tool call argument deltas from a parsed SSE data payload */
+    extractSSEToolCallDeltas?(data: unknown): ToolCallDelta[] | null;
+    /** Rebuild an SSE data payload with rehydrated tool call arguments */
+    rebuildSSEToolCallDeltas?(data: unknown, rehydratedArgs: Map<number, string>): unknown;
+    /**
+     * Detect a "tool call block finished" SSE event and return its block index.
+     * Used to flush per-index tag buffers before the stop event is forwarded.
+     * Returns null if this event is not a tool call stop signal.
+     */
+    extractSSEToolCallStop?(data: unknown): number | null;
+    /**
+     * Check if a non-streaming response body contains tool calls that need execution.
+     * Returns true if the response has tool_calls (OpenAI) or tool_use blocks (Anthropic).
+     */
+    hasResponseToolCalls?(body: unknown): boolean;
+    /**
+     * Extract structured tool call information from a non-streaming response.
+     * Arguments are returned as raw JSON strings (in anonymized form from the LLM).
+     */
+    extractResponseToolCallInfo?(body: unknown): ToolCallInfo[];
+    /**
+     * Extract the messages array from a request body.
+     */
+    extractMessages?(body: unknown): unknown[];
+    /**
+     * Build the next-round request body for a tool execution loop.
+     * Appends the assistant's tool-call response and tool result messages
+     * to the current message array without re-anonymizing existing messages.
+     *
+     * @param originalBody - The original request body (used as template for non-message fields)
+     * @param currentMessages - The current (anonymized) messages array
+     * @param assistantResponse - The raw LLM response body containing tool calls
+     * @param toolResults - Tool results to append (already anonymized)
+     * @returns A new request body with updated messages and stream: false
+     */
+    buildToolLoopBody?(originalBody: unknown, currentMessages: unknown[], assistantResponse: unknown, toolResults: ToolResultMessage[]): unknown;
+    /**
+     * Inject a system-level instruction into the request body.
+     * Used to tell the model how to handle PII placeholders in tool calls.
+     * Returns a new body with the instruction prepended to the system prompt.
+     */
+    injectSystemInstruction?(body: unknown, instruction: string): unknown;
+}
+//# sourceMappingURL=types.d.ts.map

package/dist/proxy/providers/types.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.d.ts","sourceRoot":"","sources":["../../../src/proxy/providers/types.ts"],"names":[],"mappings":"AAAA,OAAO,KAAK,EAAE,wBAAwB,EAAE,MAAM,0BAA0B,CAAC;AACzE;;;GAGG;AAEH;;;GAGG;AACH,MAAM,WAAW,YAAY;IAC3B,kFAAkF;IAClF,EAAE,EAAE,MAAM,CAAC;IACX,yBAAyB;IACzB,IAAI,EAAE,MAAM,CAAC;IACb,iEAAiE;IACjE,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;GAEG;AACH,MAAM,WAAW,iBAAiB;IAChC,kDAAkD;IAClD,UAAU,EAAE,MAAM,CAAC;IACnB,6DAA6D;IAC7D,OAAO,EAAE,MAAM,CAAC;CACjB;AAED;;;GAGG;AACH,MAAM,WAAW,aAAa;IAC5B,yEAAyE;IACzE,KAAK,EAAE,MAAM,CAAC;IACd,2CAA2C;IAC3C,SAAS,EAAE,MAAM,CAAC;CACnB;AAED;;;GAGG;AACH,MAAM,WAAW,kBAAkB;IACjC,kDAAkD;IAClD,QAAQ,CAAC,IAAI,EAAE,MAAM,CAAC;IAEtB,oEAAoE;IACpE,QAAQ,CAAC,iBAAiB,CAAC,EAAE,wBAAwB,CAAC;IAEtD,qEAAqE;IACrE,cAAc,CAAC,GAAG,EAAE,MAAM,EAAE,OAAO,EAAE,OAAO,GAAG,OAAO,CAAC;IAEvD,iEAAiE;IACjE,kBAAkB,CAAC,IAAI,EAAE,OAAO,GAAG,MAAM,EAAE,CAAC;IAE5C,uFAAuF;IACvF,kBAAkB,CAAC,IAAI,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;IAEtE,8EAA8E;IAC9E,mBAAmB,CAAC,IAAI,EAAE,OAAO,GAAG,MAAM,EAAE,CAAC;IAE7C,yFAAyF;IACzF,mBAAmB,CAAC,IAAI,EAAE,OAAO,EAAE,eAAe,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;IAEvE,0FAA0F;IAC1F,eAAe,CAAC,IAAI,EAAE,OAAO,GAAG,MAAM,GAAG,IAAI,CAAC;IAE9C,uDAAuD;IACvD,eAAe,CAAC,IAAI,EAAE,OAAO,EAAE,cAAc,EAAE,MAAM,GAAG,OAAO,CAAC;IAEhE,2EAA2E;IAC3E,kBAAkB,CAAC,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC;IAE3C,4FAA4F;IAC5F,wBAAwB,CAAC,CAAC,IAAI,EAAE,OAAO,GAAG,MAAM,EAAE,CAAC;IAEnD,6GAA6G;IAC7G,wBAAwB,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,cAAc,EAAE,MAAM,EAAE,GAAG,OAAO,CAAC;IAE5E,uEAAuE;IACvE,wBAAwB,CAAC,CAAC,IAAI,EAAE,OAAO,GAAG,aAAa,EAAE,GAAG,IAAI,CAAC;IAEjE,sEAAsE;IACtE,wBAAwB,CAAC,CACvB,IAAI,EAAE,OAAO,EACb,cAAc,EAAE,GAAG,CAAC,MAAM,EAAE,MAAM,CAAC,GAClC,OAAO,CAAC;IAEX;;;;OAIG;IACH,sBAAsB,CAAC,CAAC,IAAI,EAAE,OAAO,GAAG,MAAM,GAAG,IAAI,CAAC;IAItD;;;OAGG;IACH,oBAAoB,CAAC,CAAC,IAAI,EAAE,OAAO,GAAG,OAAO,CAAC;IAE9C;;;OAGG;IACH,2BAA2B,CAAC,CAAC,IAAI,EAAE,OAAO,GAAG,YAAY,EAAE,CAAC;IAE5D;;OAEG;IACH,eAAe,CAAC,CAAC,IAAI,EAAE,OAAO,GAAG,OAAO,EAAE,CAAC;IAE3C;;;;;;;;;;OAUG;IACH,iBAAiB,CAAC,CAChB,YAAY,EAAE,OAAO,EACrB,eAAe,EAAE,OAAO,EAAE,EAC1B,iBAAiB,EAAE,OAAO,EAC1B,WAAW,EAAE,iBAAiB,EAAE,GAC/B,OAAO,CAAC;IAEX;;;;OAIG;IACH,uBAAuB,CAAC,CAAC,IAAI,EAAE,OAAO,EAAE,WAAW,EAAE,MAAM,GAAG,OAAO,CAAC;CACvE"}

package/dist/proxy/providers/types.js

@@ -0,0 +1,2 @@
+export {};
+//# sourceMappingURL=types.js.map

package/dist/proxy/providers/types.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"types.js","sourceRoot":"","sources":["../../../src/proxy/providers/types.ts"],"names":[],"mappings":""}

package/dist/proxy/proxy-server.d.ts

@@ -0,0 +1,61 @@
+/**
+ * Rehydra Proxy Server
+ * Standalone HTTP server that proxies LLM API requests with
+ * automatic PII anonymization and rehydration.
+ */
+import { type Server, type IncomingMessage, type ServerResponse } from "node:http";
+import type { RehydraProxyConfig } from "./types.js";
+/**
+ * Configuration for the standalone proxy server
+ */
+export interface RehydraProxyServerConfig extends RehydraProxyConfig {
+    /** Port to listen on */
+    port: number;
+    /** Host to bind to (default: "127.0.0.1") */
+    host?: string;
+}
+/**
+ * A running Rehydra proxy server
+ */
+export interface RehydraProxyServer {
+    /** The underlying HTTP server */
+    server: Server;
+    /** The port the server is listening on */
+    port: number;
+    /** The host the server is bound to */
+    host: string;
+    /** Gracefully close the server */
+    close(): Promise<void>;
+}
+/**
+ * Creates and starts a standalone HTTP proxy server that anonymizes
+ * LLM API requests and rehydrates responses.
+ *
+ * @example
+ * ```typescript
+ * const proxy = await createRehydraProxyServer({
+ *   port: 8080,
+ *   upstream: 'https://api.openai.com',
+ *   keyProvider: new ConfigKeyProvider(process.env.PII_KEY!),
+ *   piiStorageProvider: new SQLitePIIStorageProvider('proxy.db'),
+ * });
+ *
+ * console.log(`Proxy running on http://${proxy.host}:${proxy.port}`);
+ *
+ * // Point your OpenAI client at the proxy:
+ * const openai = new OpenAI({ baseURL: `http://localhost:${proxy.port}/v1` });
+ *
+ * // Stop the server
+ * await proxy.close();
+ * ```
+ */
+export declare function createRehydraProxyServer(config: RehydraProxyServerConfig): Promise<RehydraProxyServer>;
+/**
+ * Convert a Node.js IncomingMessage to a Web API Request
+ */
+export declare function incomingMessageToRequest(req: IncomingMessage, host: string, port: number): Request;
+/**
+ * Write a Web API Response to a Node.js ServerResponse
+ */
+export declare function writeResponse(res: ServerResponse, webResponse: Response): Promise<void>;
+//# sourceMappingURL=proxy-server.d.ts.map

package/dist/proxy/proxy-server.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"proxy-server.d.ts","sourceRoot":"","sources":["../../src/proxy/proxy-server.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAgB,KAAK,MAAM,EAAE,KAAK,eAAe,EAAE,KAAK,cAAc,EAAE,MAAM,WAAW,CAAC;AAEjG,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAErD;;GAEG;AACH,MAAM,WAAW,wBAAyB,SAAQ,kBAAkB;IAClE,wBAAwB;IACxB,IAAI,EAAE,MAAM,CAAC;IACb,6CAA6C;IAC7C,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED;;GAEG;AACH,MAAM,WAAW,kBAAkB;IACjC,iCAAiC;IACjC,MAAM,EAAE,MAAM,CAAC;IACf,0CAA0C;IAC1C,IAAI,EAAE,MAAM,CAAC;IACb,sCAAsC;IACtC,IAAI,EAAE,MAAM,CAAC;IACb,kCAAkC;IAClC,KAAK,IAAI,OAAO,CAAC,IAAI,CAAC,CAAC;CACxB;AAED;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,wBAAsB,wBAAwB,CAC5C,MAAM,EAAE,wBAAwB,GAC/B,OAAO,CAAC,kBAAkB,CAAC,CAwC7B;AAED;;GAEG;AACH,wBAAgB,wBAAwB,CACtC,GAAG,EAAE,eAAe,EACpB,IAAI,EAAE,MAAM,EACZ,IAAI,EAAE,MAAM,GACX,OAAO,CA0BT;AAqBD;;GAEG;AACH,wBAAsB,aAAa,CACjC,GAAG,EAAE,cAAc,EACnB,WAAW,EAAE,QAAQ,GACpB,OAAO,CAAC,IAAI,CAAC,CA0Bf"}

package/dist/proxy/proxy-server.js

@@ -0,0 +1,146 @@
+/**
+ * Rehydra Proxy Server
+ * Standalone HTTP server that proxies LLM API requests with
+ * automatic PII anonymization and rehydration.
+ */
+import { createServer } from "node:http";
+import { createRehydraProxy } from "./rehydra-proxy.js";
+/**
+ * Creates and starts a standalone HTTP proxy server that anonymizes
+ * LLM API requests and rehydrates responses.
+ *
+ * @example
+ * ```typescript
+ * const proxy = await createRehydraProxyServer({
+ *   port: 8080,
+ *   upstream: 'https://api.openai.com',
+ *   keyProvider: new ConfigKeyProvider(process.env.PII_KEY!),
+ *   piiStorageProvider: new SQLitePIIStorageProvider('proxy.db'),
+ * });
+ *
+ * console.log(`Proxy running on http://${proxy.host}:${proxy.port}`);
+ *
+ * // Point your OpenAI client at the proxy:
+ * const openai = new OpenAI({ baseURL: `http://localhost:${proxy.port}/v1` });
+ *
+ * // Stop the server
+ * await proxy.close();
+ * ```
+ */
+export async function createRehydraProxyServer(config) {
+    const host = config.host ?? "127.0.0.1";
+    const proxy = createRehydraProxy(config);
+    const server = createServer((req, res) => {
+        void (async () => {
+            try {
+                const webRequest = incomingMessageToRequest(req, host, config.port);
+                const webResponse = await proxy(webRequest);
+                await writeResponse(res, webResponse);
+            }
+            catch (error) {
+                res.writeHead(502, { "Content-Type": "application/json" });
+                res.end(JSON.stringify({
+                    error: "proxy_error",
+                    message: error instanceof Error ? error.message : "Unknown proxy error",
+                }));
+            }
+        })();
+    });
+    await new Promise((resolve, reject) => {
+        server.on("error", reject);
+        server.listen(config.port, host, () => {
+            resolve();
+        });
+    });
+    return {
+        server,
+        port: config.port,
+        host,
+        async close() {
+            return new Promise((resolve, reject) => {
+                server.close((err) => {
+                    if (err !== undefined)
+                        reject(err);
+                    else
+                        resolve();
+                });
+            });
+        },
+    };
+}
+/**
+ * Convert a Node.js IncomingMessage to a Web API Request
+ */
+export function incomingMessageToRequest(req, host, port) {
+    const url = `http://${host}:${port}${req.url ?? "/"}`;
+    const headers = new Headers();
+    for (const [key, value] of Object.entries(req.headers)) {
+        if (value !== undefined) {
+            if (Array.isArray(value)) {
+                for (const v of value) {
+                    headers.append(key, v);
+                }
+            }
+            else {
+                headers.set(key, value);
+            }
+        }
+    }
+    const method = req.method ?? "GET";
+    const hasBody = method !== "GET" && method !== "HEAD";
+    return new Request(url, {
+        method,
+        headers,
+        body: hasBody ? nodeStreamToReadableStream(req) : undefined,
+        // @ts-expect-error - duplex is needed for streaming request bodies
+        duplex: hasBody ? "half" : undefined,
+    });
+}
+/**
+ * Convert a Node.js Readable stream to a Web API ReadableStream
+ */
+function nodeStreamToReadableStream(nodeStream) {
+    return new ReadableStream({
+        start(controller) {
+            nodeStream.on("data", (chunk) => {
+                controller.enqueue(new Uint8Array(chunk));
+            });
+            nodeStream.on("end", () => {
+                controller.close();
+            });
+            nodeStream.on("error", (err) => {
+                controller.error(err);
+            });
+        },
+    });
+}
+/**
+ * Write a Web API Response to a Node.js ServerResponse
+ */
+export async function writeResponse(res, webResponse) {
+    // Copy status and headers
+    const headers = {};
+    webResponse.headers.forEach((value, key) => {
+        headers[key] = value;
+    });
+    res.writeHead(webResponse.status, headers);
+    if (webResponse.body === null) {
+        res.end();
+        return;
+    }
+    // Stream the response body
+    const reader = webResponse.body.getReader();
+    try {
+        for (;;) {
+            const { done, value } = await reader.read();
+            if (done)
+                break;
+            res.write(value);
+        }
+    }
+    finally {
+        reader.releaseLock();
+        res.end();
+    }
+}
+//# sourceMappingURL=proxy-server.js.map

package/dist/proxy/proxy-server.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"proxy-server.js","sourceRoot":"","sources":["../../src/proxy/proxy-server.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAEH,OAAO,EAAE,YAAY,EAA0D,MAAM,WAAW,CAAC;AACjG,OAAO,EAAE,kBAAkB,EAAE,MAAM,oBAAoB,CAAC;AA2BxD;;;;;;;;;;;;;;;;;;;;;GAqBG;AACH,MAAM,CAAC,KAAK,UAAU,wBAAwB,CAC5C,MAAgC;IAEhC,MAAM,IAAI,GAAG,MAAM,CAAC,IAAI,IAAI,WAAW,CAAC;IACxC,MAAM,KAAK,GAAG,kBAAkB,CAAC,MAAM,CAAC,CAAC;IAEzC,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,GAAoB,EAAE,GAAmB,EAAE,EAAE;QACxE,KAAK,CAAC,KAAK,IAAmB,EAAE;YAC9B,IAAI,CAAC;gBACH,MAAM,UAAU,GAAG,wBAAwB,CAAC,GAAG,EAAE,IAAI,EAAE,MAAM,CAAC,IAAI,CAAC,CAAC;gBACpE,MAAM,WAAW,GAAG,MAAM,KAAK,CAAC,UAAU,CAAC,CAAC;gBAC5C,MAAM,aAAa,CAAC,GAAG,EAAE,WAAW,CAAC,CAAC;YACxC,CAAC;YAAC,OAAO,KAAK,EAAE,CAAC;gBACf,GAAG,CAAC,SAAS,CAAC,GAAG,EAAE,EAAE,cAAc,EAAE,kBAAkB,EAAE,CAAC,CAAC;gBAC3D,GAAG,CAAC,GAAG,CAAC,IAAI,CAAC,SAAS,CAAC;oBACrB,KAAK,EAAE,aAAa;oBACpB,OAAO,EAAE,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,qBAAqB;iBACxE,CAAC,CAAC,CAAC;YACN,CAAC;QACH,CAAC,CAAC,EAAE,CAAC;IACP,CAAC,CAAC,CAAC;IAEH,MAAM,IAAI,OAAO,CAAO,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;QAC1C,MAAM,CAAC,EAAE,CAAC,OAAO,EAAE,MAAM,CAAC,CAAC;QAC3B,MAAM,CAAC,MAAM,CAAC,MAAM,CAAC,IAAI,EAAE,IAAI,EAAE,GAAG,EAAE;YACpC,OAAO,EAAE,CAAC;QACZ,CAAC,CAAC,CAAC;IACL,CAAC,CAAC,CAAC;IAEH,OAAO;QACL,MAAM;QACN,IAAI,EAAE,MAAM,CAAC,IAAI;QACjB,IAAI;QACJ,KAAK,CAAC,KAAK;YACT,OAAO,IAAI,OAAO,CAAC,CAAC,OAAO,EAAE,MAAM,EAAE,EAAE;gBACrC,MAAM,CAAC,KAAK,CAAC,CAAC,GAAG,EAAE,EAAE;oBACnB,IAAI,GAAG,KAAK,SAAS;wBAAE,MAAM,CAAC,GAAG,CAAC,CAAC;;wBAC9B,OAAO,EAAE,CAAC;gBACjB,CAAC,CAAC,CAAC;YACL,CAAC,CAAC,CAAC;QACL,CAAC;KACF,CAAC;AACJ,CAAC;AAED;;GAEG;AACH,MAAM,UAAU,wBAAwB,CACtC,GAAoB,EACpB,IAAY,EACZ,IAAY;IAEZ,MAAM,GAAG,GAAG,UAAU,IAAI,IAAI,IAAI,GAAG,GAAG,CAAC,GAAG,IAAI,GAAG,EAAE,CAAC;IACtD,MAAM,OAAO,GAAG,IAAI,OAAO,EAAE,CAAC;IAE9B,KAAK,MAAM,CAAC,GAAG,EAAE,KAAK,CAAC,IAAI,MAAM,CAAC,OAAO,CAAC,GAAG,CAAC,OAAO,CAAC,EAAE,CAAC;QACvD,IAAI,KAAK,KAAK,SAAS,EAAE,CAAC;YACxB,IAAI,KAAK,CAAC,OAAO,CAAC,KAAK,CAAC,EAAE,CAAC;gBACzB,KAAK,MAAM,CAAC,IAAI,KAAK,EAAE,CAAC;oBACtB,OAAO,CAAC,MAAM,CAAC,GAAG,EAAE,CAAC,CAAC,CAAC;gBACzB,CAAC;YACH,CAAC;iBAAM,CAAC;gBACN,OAAO,CAAC,GAAG,CAAC,GAAG,EAAE,KAAK,CAAC,CAAC;YAC1B,CAAC;QACH,CAAC;IACH,CAAC;IAED,MAAM,MAAM,GAAG,GAAG,CAAC,MAAM,IAAI,KAAK,CAAC;IACnC,MAAM,OAAO,GAAG,MAAM,KAAK,KAAK,IAAI,MAAM,KAAK,MAAM,CAAC;IAEtD,OAAO,IAAI,OAAO,CAAC,GAAG,EAAE;QACtB,MAAM;QACN,OAAO;QACP,IAAI,EAAE,OAAO,CAAC,CAAC,CAAC,0BAA0B,CAAC,GAAG,CAAC,CAAC,CAAC,CAAC,SAAS;QAC3D,mEAAmE;QACnE,MAAM,EAAE,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,CAAC,CAAC,SAAS;KACrC,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,SAAS,0BAA0B,CAAC,UAA2B;IAC7D,OAAO,IAAI,cAAc,CAAC;QACxB,KAAK,CAAC,UAAU;YACd,UAAU,CAAC,EAAE,CAAC,MAAM,EAAE,CAAC,KAAa,EAAE,EAAE;gBACtC,UAAU,CAAC,OAAO,CAAC,IAAI,UAAU,CAAC,KAAK,CAAC,CAAC,CAAC;YAC5C,CAAC,CAAC,CAAC;YACH,UAAU,CAAC,EAAE,CAAC,KAAK,EAAE,GAAG,EAAE;gBACxB,UAAU,CAAC,KAAK,EAAE,CAAC;YACrB,CAAC,CAAC,CAAC;YACH,UAAU,CAAC,EAAE,CAAC,OAAO,EAAE,CAAC,GAAU,EAAE,EAAE;gBACpC,UAAU,CAAC,KAAK,CAAC,GAAG,CAAC,CAAC;YACxB,CAAC,CAAC,CAAC;QACL,CAAC;KACF,CAAC,CAAC;AACL,CAAC;AAED;;GAEG;AACH,MAAM,CAAC,KAAK,UAAU,aAAa,CACjC,GAAmB,EACnB,WAAqB;IAErB,0BAA0B;IAC1B,MAAM,OAAO,GAA2B,EAAE,CAAC;IAC3C,WAAW,CAAC,OAAO,CAAC,OAAO,CAAC,CAAC,KAAK,EAAE,GAAG,EAAE,EAAE;QACzC,OAAO,CAAC,GAAG,CAAC,GAAG,KAAK,CAAC;IACvB,CAAC,CAAC,CAAC;IAEH,GAAG,CAAC,SAAS,CAAC,WAAW,CAAC,MAAM,EAAE,OAAO,CAAC,CAAC;IAE3C,IAAI,WAAW,CAAC,IAAI,KAAK,IAAI,EAAE,CAAC;QAC9B,GAAG,CAAC,GAAG,EAAE,CAAC;QACV,OAAO;IACT,CAAC;IAED,2BAA2B;IAC3B,MAAM,MAAM,GAAG,WAAW,CAAC,IAAI,CAAC,SAAS,EAAE,CAAC;IAC5C,IAAI,CAAC;QACH,SAAS,CAAC;YACR,MAAM,EAAE,IAAI,EAAE,KAAK,EAAE,GAAG,MAAM,MAAM,CAAC,IAAI,EAAE,CAAC;YAC5C,IAAI,IAAI;gBAAE,MAAM;YAChB,GAAG,CAAC,KAAK,CAAC,KAAK,CAAC,CAAC;QACnB,CAAC;IACH,CAAC;YAAS,CAAC;QACT,MAAM,CAAC,WAAW,EAAE,CAAC;QACrB,GAAG,CAAC,GAAG,EAAE,CAAC;IACZ,CAAC;AACH,CAAC"}

package/dist/proxy/rehydra-fetch.d.ts

@@ -0,0 +1,35 @@
+/**
+ * Rehydra Fetch Wrapper
+ * Wraps the native fetch to anonymize outgoing LLM requests
+ * and rehydrate incoming LLM responses.
+ */
+import type { RehydraFetchConfig } from "./types.js";
+/**
+ * Creates a fetch-compatible function that automatically:
+ * 1. Anonymizes text in outgoing LLM API requests
+ * 2. Rehydrates text in incoming LLM API responses
+ *
+ * PII never leaves your infrastructure.
+ *
+ * @example
+ * ```typescript
+ * import OpenAI from 'openai';
+ * import { createRehydraFetch, InMemoryKeyProvider, InMemoryPIIStorageProvider } from 'rehydra';
+ *
+ * const rehydraFetch = createRehydraFetch({
+ *   keyProvider: new InMemoryKeyProvider(),
+ *   piiStorageProvider: new InMemoryPIIStorageProvider(),
+ * });
+ *
+ * const openai = new OpenAI({ fetch: rehydraFetch });
+ *
+ * // PII is automatically anonymized before being sent to OpenAI
+ * const response = await openai.chat.completions.create({
+ *   model: 'gpt-4',
+ *   messages: [{ role: 'user', content: 'Email john@example.com about the meeting' }],
+ * });
+ * // Response is automatically rehydrated — contains original PII
+ * ```
+ */
+export declare function createRehydraFetch(config: RehydraFetchConfig): typeof globalThis.fetch;
+//# sourceMappingURL=rehydra-fetch.d.ts.map

package/dist/proxy/rehydra-fetch.d.ts.map

@@ -0,0 +1 @@
+{"version":3,"file":"rehydra-fetch.d.ts","sourceRoot":"","sources":["../../src/proxy/rehydra-fetch.ts"],"names":[],"mappings":"AAAA;;;;GAIG;AAUH,OAAO,KAAK,EAAE,kBAAkB,EAAE,MAAM,YAAY,CAAC;AAgDrD;;;;;;;;;;;;;;;;;;;;;;;;;;GA0BG;AACH,wBAAgB,kBAAkB,CAChC,MAAM,EAAE,kBAAkB,GACzB,OAAO,UAAU,CAAC,KAAK,CAkMzB"}