koishi-plugin-chatluna-google-gemini-adapter 1.0.0-beta.8 → 1.0.0-rc.0

package/lib/utils.js

@@ -1,251 +0,0 @@
-"use strict";
-Object.defineProperty(exports, "__esModule", { value: true });
-exports.convertDeltaToMessageChunk = exports.messageTypeToGeminiRole = exports.formatToolToGeminiAITool = exports.formatToolsToGeminiAITools = exports.partAsType = exports.langchainMessageToGeminiMessage = void 0;
-/* eslint-disable @typescript-eslint/no-explicit-any */
-const messages_1 = require("@langchain/core/messages");
-const zod_to_json_schema_1 = require("zod-to-json-schema");
-async function langchainMessageToGeminiMessage(messages, model) {
-    const mappedMessage = await Promise.all(messages.map(async (rawMessage) => {
-        const role = messageTypeToGeminiRole(rawMessage._getType());
-        if (role === 'function' ||
-            rawMessage.additional_kwargs?.function_call != null) {
-            return {
-                role: 'function',
-                parts: [
-                    {
-                        functionResponse: rawMessage.additional_kwargs?.function_call !=
-                            null
-                            ? undefined
-                            : {
-                                name: rawMessage.name,
-                                response: {
-                                    name: rawMessage.name,
-                                    content: (() => {
-                                        try {
-                                            const result = JSON.parse(rawMessage.content);
-                                            if (typeof result ===
-                                                'string') {
-                                                return {
-                                                    response: result
-                                                };
-                                            }
-                                            else {
-                                                return result;
-                                            }
-                                        }
-                                        catch (e) {
-                                            return {
-                                                response: rawMessage.content
-                                            };
-                                        }
-                                    })()
-                                }
-                            },
-                        functionCall: rawMessage.additional_kwargs?.function_call !=
-                            null
-                            ? {
-                                name: rawMessage.additional_kwargs
-                                    .function_call.name,
-                                args: (() => {
-                                    try {
-                                        const result = JSON.parse(rawMessage
-                                            .additional_kwargs
-                                            .function_call
-                                            .arguments);
-                                        if (typeof result === 'string') {
-                                            return {
-                                                input: result
-                                            };
-                                        }
-                                        else {
-                                            return result;
-                                        }
-                                    }
-                                    catch (e) {
-                                        return {
-                                            input: rawMessage
-                                                .additional_kwargs
-                                                .function_call
-                                                .arguments
-                                        };
-                                    }
-                                })()
-                            }
-                            : undefined
-                    }
-                ]
-            };
-        }
-        const images = rawMessage.additional_kwargs.images;
-        const result = {
-            role,
-            parts: [
-                {
-                    text: rawMessage.content
-                }
-            ]
-        };
-        if (model.includes('vision') && images != null) {
-            for (const image of images) {
-                result.parts.push({
-                    inline_data: {
-                        // base64 image match type
-                        data: image.replace(/^data:image\/\w+;base64,/, ''),
-                        mime_type: 'image/jpeg'
-                    }
-                });
-            }
-        }
-        return result;
-    }));
-    const result = [];
-    for (let i = 0; i < mappedMessage.length; i++) {
-        const message = mappedMessage[i];
-        if (message.role !== 'system') {
-            result.push(message);
-            continue;
-        }
-        /*   if (removeSystemMessage) {
-            continue
-        } */
-        result.push({
-            role: 'user',
-            parts: message.parts
-        });
-        if (mappedMessage?.[i + 1]?.role === 'model') {
-            continue;
-        }
-        if (mappedMessage?.[i + 1]?.role === 'user') {
-            result.push({
-                role: 'model',
-                parts: [{ text: 'Okay, what do I need to do?' }]
-            });
-        }
-    }
-    if (result[result.length - 1].role === 'model') {
-        result.push({
-            role: 'user',
-            parts: [
-                {
-                    text: 'Continue what I said to you last message. Follow these instructions.'
-                }
-            ]
-        });
-    }
-    if (model.includes('vision')) {
-        // format prompts
-        const textBuffer = [];
-        const last = result.pop();
-        for (let i = 0; i < result.length; i++) {
-            const message = result[i];
-            const text = message.parts[0].text;
-            textBuffer.push(`${message.role}: ${text}`);
-        }
-        const lastParts = last.parts;
-        let lastImagesParts = lastParts.filter((part) => part.inline_data?.mime_type ===
-            'image/jpeg');
-        if (lastImagesParts.length < 1) {
-            for (let i = result.length - 1; i >= 0; i--) {
-                const message = result[i];
-                const images = message.parts.filter((part) => part.inline_data?.mime_type ===
-                    'image/jpeg');
-                if (images.length > 0) {
-                    lastImagesParts = images;
-                    break;
-                }
-            }
-        }
-        ;
-        lastParts.filter((part) => part.text !== undefined &&
-            part.text !== null).forEach((part) => {
-            textBuffer.push(`${last.role}: ${part.text}`);
-        });
-        return [
-            {
-                role: 'user',
-                parts: [
-                    {
-                        text: textBuffer.join('\n')
-                    },
-                    ...lastImagesParts
-                ]
-            }
-        ];
-    }
-    return result;
-}
-exports.langchainMessageToGeminiMessage = langchainMessageToGeminiMessage;
-function partAsType(part) {
-    return part;
-}
-exports.partAsType = partAsType;
-function formatToolsToGeminiAITools(tools) {
-    if (tools.length < 1) {
-        return undefined;
-    }
-    return tools.map(formatToolToGeminiAITool);
-}
-exports.formatToolsToGeminiAITools = formatToolsToGeminiAITools;
-function formatToolToGeminiAITool(tool) {
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any
-    const parameters = (0, zod_to_json_schema_1.zodToJsonSchema)(tool.schema);
-    // remove unsupported properties
-    delete parameters['$schema'];
-    delete parameters['additionalProperties'];
-    return {
-        name: tool.name,
-        description: tool.description,
-        // any?
-        parameters
-    };
-}
-exports.formatToolToGeminiAITool = formatToolToGeminiAITool;
-function messageTypeToGeminiRole(type) {
-    switch (type) {
-        case 'system':
-            return 'system';
-        case 'ai':
-            return 'model';
-        case 'human':
-            return 'user';
-        case 'function':
-            return 'function';
-        default:
-            throw new Error(`Unknown message type: ${type}`);
-    }
-}
-exports.messageTypeToGeminiRole = messageTypeToGeminiRole;
-function convertDeltaToMessageChunk(
-// eslint-disable-next-line @typescript-eslint/no-explicit-any
-delta, defaultRole) {
-    const role = delta.role ?? defaultRole;
-    const content = delta.content ?? '';
-    // eslint-disable-next-line @typescript-eslint/no-explicit-any, @typescript-eslint/naming-convention
-    let additional_kwargs;
-    if (delta.function_call) {
-        additional_kwargs = {
-            function_call: delta.function_call
-        };
-    }
-    else if (delta.tool_calls) {
-        additional_kwargs = {
-            tool_calls: delta.tool_calls
-        };
-    }
-    else {
-        additional_kwargs = {};
-    }
-    if (role === 'user') {
-        return new messages_1.HumanMessageChunk({ content });
-    }
-    else if (role === 'assistant') {
-        return new messages_1.AIMessageChunk({ content, additional_kwargs });
-    }
-    else if (role === 'system') {
-        return new messages_1.SystemMessageChunk({ content });
-    }
-    else {
-        return new messages_1.ChatMessageChunk({ content, role });
-    }
-}
-exports.convertDeltaToMessageChunk = convertDeltaToMessageChunk;