@n1k1t/unit-generator/lib/src/models/llm/providers/google.js

Coverage based unit tests AI generator

"use strict";
Object.defineProperty(exports, "__esModule", { value: true });
exports.LlmGoogleProvider = void 0;
const google_1 = require("@ai-sdk/google");
const model_1 = require("./model");
class LlmGoogleProvider extends model_1.LlmProvider {
    constructor() {
        super(...arguments);
        this.name = 'google';
        this.tag = (0, google_1.createGoogleGenerativeAI)({
            apiKey: this.connection.key,
            baseURL: this.connection.url,
        })(this.model);
    }
    clone() {
        const clone = new LlmGoogleProvider(this.model, {
            temperature: this.temperature,
            connection: this.connection,
            options: Object.assign({}, this.options),
            tools: Object.assign({}, this.tools),
        });
        return clone;
    }
    static build(model, parameters) {
        return new LlmGoogleProvider(model, {
            ...parameters,
            options: parameters.options ?? {
                thinkingConfig: {
                    thinkingBudget: 2048,
                    includeThoughts: true,
                },
            },
        });
    }
}
exports.LlmGoogleProvider = LlmGoogleProvider;
//# sourceMappingURL=google.js.map