/**
 * @license
 * Copyright 2025 Google LLC
 * SPDX-License-Identifier: Apache-2.0
 */
import { GoogleGenAI, } from '@google/genai';
/**
 * Wrapper around GoogleGenAI models interface to implement ContentGenerator
 * This wrapper ensures that user_prompt_id is NOT passed to the Google GenAI API
 */
export class GoogleGenAIWrapper {
    models;
    constructor(config, httpOptions) {
        const googleGenAI = new GoogleGenAI({
            apiKey: config.apiKey === '' ? undefined : config.apiKey,
            vertexai: config.vertexai,
            httpOptions,
        });
        this.models = googleGenAI.models;
    }
    async generateContent(request) {
        // The Google GenAI SDK doesn't accept user_prompt_id, so we just pass the request as-is
        return this.models.generateContent(request);
    }
    async generateContentStream(request) {
        // The Google GenAI SDK doesn't accept user_prompt_id, so we just pass the request as-is
        return this.models.generateContentStream(request);
    }
    async countTokens(request) {
        return this.models.countTokens(request);
    }
    async embedContent(request) {
        return this.models.embedContent(request);
    }
}
//# sourceMappingURL=googleGenAIWrapper.js.map