@softvisio/core/lib/api/ai/gemini.js

Softisio core

import fetch from "#lib/fetch";

// DOCS: https://ai.google.dev/api

const ID = "gemini",

    // DOCS: https://ai.google.dev/gemini-api/docs/models
    MODELS = {
        "gemini-2.5-flash-preview-04-17": {},
        "gemini-2.5-pro-preview-05-06": {},
        "gemini-2.0-flash": {},
        "gemini-2.0-flash-preview-image-generation": {},
        "gemini-2.0-flash-lite": {},
        "gemini-2.0-flash-live-001": {},
        "gemini-embedding-exp-03-07": {
            "embeddingSize": 3072, // 768, 1536, 3072
        },

        // english
        "text-embedding-004": {
            "embeddingSize": 768,
        },

        // multiligual
        "text-multilingual-embedding-002": {
            "embeddingSize": 768,
        },
        "imagen-3.0-generate-002": {},
        "veo-2.0-generate-001": {},
    },
    safetySettingsOff = [
        { "category": "HARM_CATEGORY_CIVIC_INTEGRITY", "threshold": "OFF" },
        { "category": "HARM_CATEGORY_DANGEROUS_CONTENT", "threshold": "OFF" },
        { "category": "HARM_CATEGORY_HARASSMENT", "threshold": "OFF" },
        { "category": "HARM_CATEGORY_HATE_SPEECH", "threshold": "OFF" },
        { "category": "HARM_CATEGORY_SEXUALLY_EXPLICIT", "threshold": "OFF" },
    ];

export default class OllamaApi {
    #apiKey;

    constructor ( apiKey ) {
        this.#apiKey = apiKey;
    }

    // static
    static get id () {
        return ID;
    }

    static get models () {
        return MODELS;
    }

    // properties
    get id () {
        return this.constructor.id;
    }

    get models () {
        return this.constructor.models;
    }

    // public
    async getEmbedding ( model, input, { simple, ...options } = {} ) {
        const body = {
            ...options,
            "model": "models/" + model,
            "content": input,
        };

        if ( typeof input === "string" ) {
            body.content = {
                "parts": [
                    {
                        "text": input,
                    },
                ],
            };
        }

        const res = await this.#doRequest( model, "embedContent", body );

        if ( !res.ok ) return res;

        if ( simple ) {
            res.data = res.data.embedding.values;
        }

        return res;
    }

    async getCompletion ( model, prompt, { simple, ...options } = {} ) {
        const body = {
            "safetySettings": safetySettingsOff,
            ...options,
            "contents": prompt,
        };

        if ( typeof prompt === "string" ) {
            body.contents = [
                {
                    "parts": [
                        {
                            "text": prompt,
                        },
                    ],
                },
            ];
        }

        const res = await this.#doRequest( model, "generateContent", body );

        if ( simple ) {
            res.data = res.data.candidates[ 0 ].content.parts[ 0 ].text;

            if ( body.generationConfig?.responseMimeType === "application/json" ) {
                res.data = JSON.parse( res.data );
            }
        }
        else {
            if ( body.generationConfig?.responseMimeType === "application/json" ) {
                res.data.candidates[ 0 ].content.parts[ 0 ].text = JSON.parse( res.data.candidates[ 0 ].content.parts[ 0 ].text );
            }
        }

        return res;
    }

    async getModels () {
        return this.#doRequest();
    }

    async getModelInfo ( model ) {
        return this.#doRequest( model );
    }

    // private
    async #doRequest ( model, method, body ) {
        var url = "https://generativelanguage.googleapis.com/v1beta/models";

        if ( model ) {
            url += `/${ model }${ method
                ? ":" + method
                : "" }`;
        }

        url += `?key=${ this.#apiKey }`;

        const res = await fetch( url, {
            "method": body
                ? "POST"
                : "GET",
            "headers": {
                "content-type": "application/json",
            },
            "body": body
                ? JSON.stringify( body )
                : undefined,
        } );

        if ( !res.ok ) return res;

        const data = await res.json();

        return result( 200, data );
    }
}