@just-every/ensemble 0.1.0

package/dist/utils/stream_converter.js

@@ -0,0 +1,172 @@
+export async function convertStreamToMessages(stream, initialMessages = [], options = {}) {
+    const messages = [...initialMessages];
+    let fullResponse = '';
+    const collectedToolCalls = [];
+    const toolResults = [];
+    for await (const event of stream) {
+        switch (event.type) {
+            case 'message_complete': {
+                const messageEvent = event;
+                if (messageEvent.thinking_content) {
+                    const thinkingMessage = {
+                        type: 'thinking',
+                        role: 'assistant',
+                        content: messageEvent.thinking_content &&
+                            messageEvent.thinking_content !== '{empty}'
+                            ? messageEvent.thinking_content
+                            : '',
+                        signature: messageEvent.thinking_signature || '',
+                        thinking_id: messageEvent.message_id || '',
+                        status: 'completed',
+                        model: options.model || messageEvent.model || 'unknown',
+                    };
+                    messages.push(thinkingMessage);
+                    if (options.onThinking) {
+                        await options.onThinking(thinkingMessage);
+                    }
+                }
+                if (messageEvent.content) {
+                    fullResponse = messageEvent.content;
+                    const contentMessage = {
+                        id: messageEvent.message_id,
+                        type: 'message',
+                        role: 'assistant',
+                        content: messageEvent.content,
+                        status: 'completed',
+                        model: options.model || messageEvent.model || 'unknown',
+                    };
+                    messages.push(contentMessage);
+                    if (options.onResponse) {
+                        await options.onResponse(contentMessage);
+                    }
+                }
+                break;
+            }
+            case 'tool_start': {
+                const toolEvent = event;
+                if (!toolEvent.tool_calls || toolEvent.tool_calls.length === 0) {
+                    continue;
+                }
+                collectedToolCalls.push(...toolEvent.tool_calls);
+                if (options.processToolCall) {
+                    try {
+                        const results = await options.processToolCall(toolEvent.tool_calls);
+                        const resultsArray = Array.isArray(results)
+                            ? results
+                            : [results];
+                        for (let i = 0; i < toolEvent.tool_calls.length; i++) {
+                            const toolCall = toolEvent.tool_calls[i];
+                            const result = resultsArray[i] || resultsArray[0];
+                            toolResults.push({
+                                id: toolCall.id,
+                                call_id: toolCall.call_id || toolCall.id,
+                                output: typeof result === 'string'
+                                    ? result
+                                    : JSON.stringify(result),
+                            });
+                        }
+                    }
+                    catch (error) {
+                        const errorResult = {
+                            error: `Tool execution failed: ${error instanceof Error ? error.message : String(error)}`,
+                        };
+                        for (const toolCall of toolEvent.tool_calls) {
+                            toolResults.push({
+                                id: toolCall.id,
+                                call_id: toolCall.call_id || toolCall.id,
+                                output: JSON.stringify(errorResult),
+                            });
+                        }
+                    }
+                }
+                break;
+            }
+            case 'tool_done': {
+                for (const toolCall of collectedToolCalls) {
+                    const functionCall = {
+                        type: 'function_call',
+                        id: toolCall.id,
+                        call_id: toolCall.call_id || toolCall.id,
+                        name: toolCall.function.name,
+                        arguments: toolCall.function.arguments,
+                        model: options.model || 'unknown',
+                    };
+                    messages.push(functionCall);
+                    const result = toolResults.find(r => r.call_id === (toolCall.call_id || toolCall.id));
+                    if (result) {
+                        const functionOutput = {
+                            type: 'function_call_output',
+                            id: toolCall.id,
+                            call_id: toolCall.call_id || toolCall.id,
+                            name: toolCall.function.name,
+                            output: result.output,
+                            model: options.model || 'unknown',
+                        };
+                        messages.push(functionOutput);
+                    }
+                }
+                collectedToolCalls.length = 0;
+                toolResults.length = 0;
+                break;
+            }
+            case 'error': {
+                console.error('[Stream Converter] Error event:', event.error);
+                break;
+            }
+        }
+    }
+    if (collectedToolCalls.length > 0) {
+        for (const toolCall of collectedToolCalls) {
+            const functionCall = {
+                type: 'function_call',
+                id: toolCall.id,
+                call_id: toolCall.call_id || toolCall.id,
+                name: toolCall.function.name,
+                arguments: toolCall.function.arguments,
+                model: options.model || 'unknown',
+            };
+            messages.push(functionCall);
+            const result = toolResults.find(r => r.call_id === (toolCall.call_id || toolCall.id));
+            if (result) {
+                const functionOutput = {
+                    type: 'function_call_output',
+                    id: toolCall.id,
+                    call_id: toolCall.call_id || toolCall.id,
+                    name: toolCall.function.name,
+                    output: result.output,
+                    model: options.model || 'unknown',
+                };
+                messages.push(functionOutput);
+            }
+        }
+    }
+    return {
+        messages,
+        fullResponse,
+        toolCalls: collectedToolCalls,
+    };
+}
+export async function chainRequests(requests, initialMessages = []) {
+    let currentMessages = initialMessages;
+    let lastResult = {
+        messages: currentMessages,
+        fullResponse: '',
+        toolCalls: [],
+    };
+    const { request } = await import('../index.js');
+    for (const req of requests) {
+        const messages = [...currentMessages];
+        if (req.systemPrompt) {
+            messages.push({
+                type: 'message',
+                role: 'system',
+                content: req.systemPrompt,
+            });
+        }
+        const stream = request(req.model, messages);
+        lastResult = await convertStreamToMessages(stream, [], req.options);
+        currentMessages = [...currentMessages, ...lastResult.messages];
+    }
+    return lastResult;
+}
+//# sourceMappingURL=stream_converter.js.map

package/dist/utils/stream_converter.js.map

@@ -0,0 +1 @@
+{"version":3,"file":"stream_converter.js","sourceRoot":"","sources":["../../utils/stream_converter.ts"],"names":[],"mappings":"AAmCA,MAAM,CAAC,KAAK,UAAU,uBAAuB,CACzC,MAA2C,EAC3C,kBAAiC,EAAE,EACnC,UAA6B,EAAE;IAE/B,MAAM,QAAQ,GAAkB,CAAC,GAAG,eAAe,CAAC,CAAC;IACrD,IAAI,YAAY,GAAG,EAAE,CAAC;IACtB,MAAM,kBAAkB,GAAe,EAAE,CAAC;IAC1C,MAAM,WAAW,GAA2D,EAAE,CAAC;IAE/E,IAAI,KAAK,EAAE,MAAM,KAAK,IAAI,MAAM,EAAE,CAAC;QAC/B,QAAQ,KAAK,CAAC,IAAI,EAAE,CAAC;YACjB,KAAK,kBAAkB,CAAC,CAAC,CAAC;gBACtB,MAAM,YAAY,GAAG,KAAY,CAAC;gBAGlC,IAAI,YAAY,CAAC,gBAAgB,EAAE,CAAC;oBAChC,MAAM,eAAe,GAA4B;wBAC7C,IAAI,EAAE,UAAU;wBAChB,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,YAAY,CAAC,gBAAgB;4BAC9B,YAAY,CAAC,gBAAgB,KAAK,SAAS;4BAC/C,CAAC,CAAC,YAAY,CAAC,gBAAgB;4BAC/B,CAAC,CAAC,EAAE;wBACR,SAAS,EAAE,YAAY,CAAC,kBAAkB,IAAI,EAAE;wBAChD,WAAW,EAAE,YAAY,CAAC,UAAU,IAAI,EAAE;wBAC1C,MAAM,EAAE,WAAW;wBACnB,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,YAAY,CAAC,KAAK,IAAI,SAAS;qBAC1D,CAAC;oBACF,QAAQ,CAAC,IAAI,CAAC,eAAe,CAAC,CAAC;oBAE/B,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;wBACrB,MAAM,OAAO,CAAC,UAAU,CAAC,eAAe,CAAC,CAAC;oBAC9C,CAAC;gBACL,CAAC;gBAGD,IAAI,YAAY,CAAC,OAAO,EAAE,CAAC;oBACvB,YAAY,GAAG,YAAY,CAAC,OAAO,CAAC;oBACpC,MAAM,cAAc,GAA0B;wBAC1C,EAAE,EAAE,YAAY,CAAC,UAAU;wBAC3B,IAAI,EAAE,SAAS;wBACf,IAAI,EAAE,WAAW;wBACjB,OAAO,EAAE,YAAY,CAAC,OAAO;wBAC7B,MAAM,EAAE,WAAW;wBACnB,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,YAAY,CAAC,KAAK,IAAI,SAAS;qBAC1D,CAAC;oBACF,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBAE9B,IAAI,OAAO,CAAC,UAAU,EAAE,CAAC;wBACrB,MAAM,OAAO,CAAC,UAAU,CAAC,cAAc,CAAC,CAAC;oBAC7C,CAAC;gBACL,CAAC;gBACD,MAAM;YACV,CAAC;YAED,KAAK,YAAY,CAAC,CAAC,CAAC;gBAChB,MAAM,SAAS,GAAG,KAAY,CAAC;gBAC/B,IAAI,CAAC,SAAS,CAAC,UAAU,IAAI,SAAS,CAAC,UAAU,CAAC,MAAM,KAAK,CAAC,EAAE,CAAC;oBAC7D,SAAS;gBACb,CAAC;gBAGD,kBAAkB,CAAC,IAAI,CAAC,GAAG,SAAS,CAAC,UAAU,CAAC,CAAC;gBAGjD,IAAI,OAAO,CAAC,eAAe,EAAE,CAAC;oBAC1B,IAAI,CAAC;wBACD,MAAM,OAAO,GAAG,MAAM,OAAO,CAAC,eAAe,CAAC,SAAS,CAAC,UAAU,CAAC,CAAC;wBAGpE,MAAM,YAAY,GAAG,KAAK,CAAC,OAAO,CAAC,OAAO,CAAC;4BACvC,CAAC,CAAC,OAAO;4BACT,CAAC,CAAC,CAAC,OAAO,CAAC,CAAC;wBAGhB,KAAK,IAAI,CAAC,GAAG,CAAC,EAAE,CAAC,GAAG,SAAS,CAAC,UAAU,CAAC,MAAM,EAAE,CAAC,EAAE,EAAE,CAAC;4BACnD,MAAM,QAAQ,GAAG,SAAS,CAAC,UAAU,CAAC,CAAC,CAAC,CAAC;4BACzC,MAAM,MAAM,GAAG,YAAY,CAAC,CAAC,CAAC,IAAI,YAAY,CAAC,CAAC,CAAC,CAAC;4BAElD,WAAW,CAAC,IAAI,CAAC;gCACb,EAAE,EAAE,QAAQ,CAAC,EAAE;gCACf,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE;gCACxC,MAAM,EAAE,OAAO,MAAM,KAAK,QAAQ;oCAC9B,CAAC,CAAC,MAAM;oCACR,CAAC,CAAC,IAAI,CAAC,SAAS,CAAC,MAAM,CAAC;6BAC/B,CAAC,CAAC;wBACP,CAAC;oBACL,CAAC;oBAAC,OAAO,KAAK,EAAE,CAAC;wBAEb,MAAM,WAAW,GAAG;4BAChB,KAAK,EAAE,0BAA0B,KAAK,YAAY,KAAK,CAAC,CAAC,CAAC,KAAK,CAAC,OAAO,CAAC,CAAC,CAAC,MAAM,CAAC,KAAK,CAAC,EAAE;yBAC5F,CAAC;wBAEF,KAAK,MAAM,QAAQ,IAAI,SAAS,CAAC,UAAU,EAAE,CAAC;4BAC1C,WAAW,CAAC,IAAI,CAAC;gCACb,EAAE,EAAE,QAAQ,CAAC,EAAE;gCACf,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE;gCACxC,MAAM,EAAE,IAAI,CAAC,SAAS,CAAC,WAAW,CAAC;6BACtC,CAAC,CAAC;wBACP,CAAC;oBACL,CAAC;gBACL,CAAC;gBACD,MAAM;YACV,CAAC;YAED,KAAK,WAAW,CAAC,CAAC,CAAC;gBAEf,KAAK,MAAM,QAAQ,IAAI,kBAAkB,EAAE,CAAC;oBAExC,MAAM,YAAY,GAA8B;wBAC5C,IAAI,EAAE,eAAe;wBACrB,EAAE,EAAE,QAAQ,CAAC,EAAE;wBACf,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE;wBACxC,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;wBAC5B,SAAS,EAAE,QAAQ,CAAC,QAAQ,CAAC,SAAS;wBACtC,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,SAAS;qBACpC,CAAC;oBACF,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;oBAG5B,MAAM,MAAM,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAChC,CAAC,CAAC,OAAO,KAAK,CAAC,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE,CAAC,CAClD,CAAC;oBAEF,IAAI,MAAM,EAAE,CAAC;wBACT,MAAM,cAAc,GAAoC;4BACpD,IAAI,EAAE,sBAAsB;4BAC5B,EAAE,EAAE,QAAQ,CAAC,EAAE;4BACf,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE;4BACxC,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;4BAC5B,MAAM,EAAE,MAAM,CAAC,MAAM;4BACrB,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,SAAS;yBACpC,CAAC;wBACF,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;oBAClC,CAAC;gBACL,CAAC;gBAGD,kBAAkB,CAAC,MAAM,GAAG,CAAC,CAAC;gBAC9B,WAAW,CAAC,MAAM,GAAG,CAAC,CAAC;gBACvB,MAAM;YACV,CAAC;YAED,KAAK,OAAO,CAAC,CAAC,CAAC;gBAEX,OAAO,CAAC,KAAK,CAAC,iCAAiC,EAAG,KAAa,CAAC,KAAK,CAAC,CAAC;gBACvE,MAAM;YACV,CAAC;QACL,CAAC;IACL,CAAC;IAGD,IAAI,kBAAkB,CAAC,MAAM,GAAG,CAAC,EAAE,CAAC;QAChC,KAAK,MAAM,QAAQ,IAAI,kBAAkB,EAAE,CAAC;YAExC,MAAM,YAAY,GAA8B;gBAC5C,IAAI,EAAE,eAAe;gBACrB,EAAE,EAAE,QAAQ,CAAC,EAAE;gBACf,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE;gBACxC,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;gBAC5B,SAAS,EAAE,QAAQ,CAAC,QAAQ,CAAC,SAAS;gBACtC,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,SAAS;aACpC,CAAC;YACF,QAAQ,CAAC,IAAI,CAAC,YAAY,CAAC,CAAC;YAG5B,MAAM,MAAM,GAAG,WAAW,CAAC,IAAI,CAAC,CAAC,CAAC,EAAE,CAChC,CAAC,CAAC,OAAO,KAAK,CAAC,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE,CAAC,CAClD,CAAC;YAEF,IAAI,MAAM,EAAE,CAAC;gBACT,MAAM,cAAc,GAAoC;oBACpD,IAAI,EAAE,sBAAsB;oBAC5B,EAAE,EAAE,QAAQ,CAAC,EAAE;oBACf,OAAO,EAAE,QAAQ,CAAC,OAAO,IAAI,QAAQ,CAAC,EAAE;oBACxC,IAAI,EAAE,QAAQ,CAAC,QAAQ,CAAC,IAAI;oBAC5B,MAAM,EAAE,MAAM,CAAC,MAAM;oBACrB,KAAK,EAAE,OAAO,CAAC,KAAK,IAAI,SAAS;iBACpC,CAAC;gBACF,QAAQ,CAAC,IAAI,CAAC,cAAc,CAAC,CAAC;YAClC,CAAC;QACL,CAAC;IACL,CAAC;IAED,OAAO;QACH,QAAQ;QACR,YAAY;QACZ,SAAS,EAAE,kBAAkB;KAChC,CAAC;AACN,CAAC;AAMD,MAAM,CAAC,KAAK,UAAU,aAAa,CAC/B,QAIE,EACF,kBAAiC,EAAE;IAEnC,IAAI,eAAe,GAAG,eAAe,CAAC;IACtC,IAAI,UAAU,GAAqB;QAC/B,QAAQ,EAAE,eAAe;QACzB,YAAY,EAAE,EAAE;QAChB,SAAS,EAAE,EAAE;KAChB,CAAC;IAGF,MAAM,EAAE,OAAO,EAAE,GAAG,MAAM,MAAM,CAAC,aAAa,CAAC,CAAC;IAEhD,KAAK,MAAM,GAAG,IAAI,QAAQ,EAAE,CAAC;QAEzB,MAAM,QAAQ,GAAG,CAAC,GAAG,eAAe,CAAC,CAAC;QACtC,IAAI,GAAG,CAAC,YAAY,EAAE,CAAC;YACnB,QAAQ,CAAC,IAAI,CAAC;gBACV,IAAI,EAAE,SAAS;gBACf,IAAI,EAAE,QAAQ;gBACd,OAAO,EAAE,GAAG,CAAC,YAAY;aAC5B,CAAC,CAAC;QACP,CAAC;QAGD,MAAM,MAAM,GAAG,OAAO,CAAC,GAAG,CAAC,KAAK,EAAE,QAAQ,CAAC,CAAC;QAG5C,UAAU,GAAG,MAAM,uBAAuB,CACtC,MAAM,EACN,EAAE,EACF,GAAG,CAAC,OAAO,CACd,CAAC;QAGF,eAAe,GAAG,CAAC,GAAG,eAAe,EAAE,GAAG,UAAU,CAAC,QAAQ,CAAC,CAAC;IACnE,CAAC;IAED,OAAO,UAAU,CAAC;AACtB,CAAC"}