cloudassist-ai-provider 0.0.1 → 0.0.2

package/dist/google-error.d.ts

@@ -1,12 +0,0 @@
-import { type InferValidator } from '@ai-sdk/provider-utils';
-declare const googleErrorDataSchema: import("@ai-sdk/provider-utils").LazySchema<{
-    error: {
-        code: number | null;
-        message: string;
-        status: string;
-    };
-}>;
-export type GoogleErrorData = InferValidator<typeof googleErrorDataSchema>;
-export declare const googleFailedResponseHandler: import("@ai-sdk/provider-utils").ResponseHandler<import("@ai-sdk/provider").APICallError>;
-export {};
-//# sourceMappingURL=google-error.d.ts.map

package/dist/google-error.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"google-error.d.ts","sourceRoot":"","sources":["../src/google-error.ts"],"names":[],"mappings":"AAAA,OAAO,EAEL,KAAK,cAAc,EAGpB,MAAM,wBAAwB,CAAC;AAGhC,QAAA,MAAM,qBAAqB;;;;;;EAU1B,CAAC;AAEF,MAAM,MAAM,eAAe,GAAG,cAAc,CAAC,OAAO,qBAAqB,CAAC,CAAC;AAE3E,eAAO,MAAM,2BAA2B,2FAGtC,CAAC"}

package/dist/google-error.js

@@ -1,13 +0,0 @@
-import { createJsonErrorResponseHandler, lazySchema, zodSchema, } from '@ai-sdk/provider-utils';
-import { z } from 'zod/v4';
-const googleErrorDataSchema = lazySchema(() => zodSchema(z.object({
-    error: z.object({
-        code: z.number().nullable(),
-        message: z.string(),
-        status: z.string(),
-    }),
-})));
-export const googleFailedResponseHandler = createJsonErrorResponseHandler({
-    errorSchema: googleErrorDataSchema,
-    errorToMessage: data => data.error.message,
-});

package/dist/google-generative-ai-embedding-model.d.ts

@@ -1,21 +0,0 @@
-import { EmbeddingModelV2 } from '@ai-sdk/provider';
-import { FetchFunction } from '@ai-sdk/provider-utils';
-import { GoogleGenerativeAIEmbeddingModelId } from './google-generative-ai-embedding-options';
-type GoogleGenerativeAIEmbeddingConfig = {
-    provider: string;
-    baseURL: string;
-    headers: () => Record<string, string | undefined>;
-    fetch?: FetchFunction;
-};
-export declare class GoogleGenerativeAIEmbeddingModel implements EmbeddingModelV2<string> {
-    readonly specificationVersion = "v2";
-    readonly modelId: GoogleGenerativeAIEmbeddingModelId;
-    readonly maxEmbeddingsPerCall = 2048;
-    readonly supportsParallelCalls = true;
-    private readonly config;
-    get provider(): string;
-    constructor(modelId: GoogleGenerativeAIEmbeddingModelId, config: GoogleGenerativeAIEmbeddingConfig);
-    doEmbed({ values, headers, abortSignal, providerOptions, }: Parameters<EmbeddingModelV2<string>['doEmbed']>[0]): Promise<Awaited<ReturnType<EmbeddingModelV2<string>['doEmbed']>>>;
-}
-export {};
-//# sourceMappingURL=google-generative-ai-embedding-model.d.ts.map

package/dist/google-generative-ai-embedding-model.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"google-generative-ai-embedding-model.d.ts","sourceRoot":"","sources":["../src/google-generative-ai-embedding-model.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,gBAAgB,EAEjB,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EAGL,aAAa,EAMd,MAAM,wBAAwB,CAAC;AAGhC,OAAO,EACL,kCAAkC,EAEnC,MAAM,0CAA0C,CAAC;AAElD,KAAK,iCAAiC,GAAG;IACvC,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,EAAE,MAAM,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;IAClD,KAAK,CAAC,EAAE,aAAa,CAAC;CACvB,CAAC;AAEF,qBAAa,gCACX,YAAW,gBAAgB,CAAC,MAAM,CAAC;IAEnC,QAAQ,CAAC,oBAAoB,QAAQ;IACrC,QAAQ,CAAC,OAAO,EAAE,kCAAkC,CAAC;IACrD,QAAQ,CAAC,oBAAoB,QAAQ;IACrC,QAAQ,CAAC,qBAAqB,QAAQ;IAEtC,OAAO,CAAC,QAAQ,CAAC,MAAM,CAAoC;IAE3D,IAAI,QAAQ,IAAI,MAAM,CAErB;gBAEC,OAAO,EAAE,kCAAkC,EAC3C,MAAM,EAAE,iCAAiC;IAMrC,OAAO,CAAC,EACZ,MAAM,EACN,OAAO,EACP,WAAW,EACX,eAAe,GAChB,EAAE,UAAU,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CAAC,CAAC,GAAG,OAAO,CAC7D,OAAO,CAAC,UAAU,CAAC,gBAAgB,CAAC,MAAM,CAAC,CAAC,SAAS,CAAC,CAAC,CAAC,CACzD;CAmFF"}

package/dist/google-generative-ai-embedding-model.js

@@ -1,88 +0,0 @@
-import { TooManyEmbeddingValuesForCallError, } from '@ai-sdk/provider';
-import { combineHeaders, createJsonResponseHandler, lazySchema, parseProviderOptions, postJsonToApi, resolve, zodSchema, } from '@ai-sdk/provider-utils';
-import { z } from 'zod/v4';
-import { googleFailedResponseHandler } from './google-error';
-import { googleGenerativeAIEmbeddingProviderOptions, } from './google-generative-ai-embedding-options';
-export class GoogleGenerativeAIEmbeddingModel {
-    get provider() {
-        return this.config.provider;
-    }
-    constructor(modelId, config) {
-        this.specificationVersion = 'v2';
-        this.maxEmbeddingsPerCall = 2048;
-        this.supportsParallelCalls = true;
-        this.modelId = modelId;
-        this.config = config;
-    }
-    async doEmbed({ values, headers, abortSignal, providerOptions, }) {
-        // Parse provider options
-        const googleOptions = await parseProviderOptions({
-            provider: 'google',
-            providerOptions,
-            schema: googleGenerativeAIEmbeddingProviderOptions,
-        });
-        if (values.length > this.maxEmbeddingsPerCall) {
-            throw new TooManyEmbeddingValuesForCallError({
-                provider: this.provider,
-                modelId: this.modelId,
-                maxEmbeddingsPerCall: this.maxEmbeddingsPerCall,
-                values,
-            });
-        }
-        const mergedHeaders = combineHeaders(await resolve(this.config.headers), headers);
-        // For single embeddings, use the single endpoint (ratelimits, etc.)
-        if (values.length === 1) {
-            const { responseHeaders, value: response, rawValue, } = await postJsonToApi({
-                url: `${this.config.baseURL}/models/${this.modelId}:embedContent`,
-                headers: mergedHeaders,
-                body: {
-                    model: `models/${this.modelId}`,
-                    content: {
-                        parts: [{ text: values[0] }],
-                    },
-                    outputDimensionality: googleOptions === null || googleOptions === void 0 ? void 0 : googleOptions.outputDimensionality,
-                    taskType: googleOptions === null || googleOptions === void 0 ? void 0 : googleOptions.taskType,
-                },
-                failedResponseHandler: googleFailedResponseHandler,
-                successfulResponseHandler: createJsonResponseHandler(googleGenerativeAISingleEmbeddingResponseSchema),
-                abortSignal,
-                fetch: this.config.fetch,
-            });
-            return {
-                embeddings: [response.embedding.values],
-                usage: undefined,
-                response: { headers: responseHeaders, body: rawValue },
-            };
-        }
-        const { responseHeaders, value: response, rawValue, } = await postJsonToApi({
-            url: `${this.config.baseURL}/models/${this.modelId}:batchEmbedContents`,
-            headers: mergedHeaders,
-            body: {
-                requests: values.map(value => ({
-                    model: `models/${this.modelId}`,
-                    content: { role: 'user', parts: [{ text: value }] },
-                    outputDimensionality: googleOptions === null || googleOptions === void 0 ? void 0 : googleOptions.outputDimensionality,
-                    taskType: googleOptions === null || googleOptions === void 0 ? void 0 : googleOptions.taskType,
-                })),
-            },
-            failedResponseHandler: googleFailedResponseHandler,
-            successfulResponseHandler: createJsonResponseHandler(googleGenerativeAITextEmbeddingResponseSchema),
-            abortSignal,
-            fetch: this.config.fetch,
-        });
-        return {
-            embeddings: response.embeddings.map(item => item.values),
-            usage: undefined,
-            response: { headers: responseHeaders, body: rawValue },
-        };
-    }
-}
-// minimal version of the schema, focussed on what is needed for the implementation
-// this approach limits breakages when the API changes and increases efficiency
-const googleGenerativeAITextEmbeddingResponseSchema = lazySchema(() => zodSchema(z.object({
-    embeddings: z.array(z.object({ values: z.array(z.number()) })),
-})));
-// Schema for single embedding response
-const googleGenerativeAISingleEmbeddingResponseSchema = lazySchema(() => zodSchema(z.object({
-    embedding: z.object({ values: z.array(z.number()) }),
-})));

package/dist/google-generative-ai-embedding-model.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=google-generative-ai-embedding-model.test.d.ts.map

package/dist/google-generative-ai-embedding-model.test.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"google-generative-ai-embedding-model.test.d.ts","sourceRoot":"","sources":["../src/google-generative-ai-embedding-model.test.ts"],"names":[],"mappings":""}

package/dist/google-generative-ai-embedding-model.test.js

@@ -1,148 +0,0 @@
-import { createTestServer } from '@ai-sdk/test-server/with-vitest';
-import { GoogleGenerativeAIEmbeddingModel } from './google-generative-ai-embedding-model';
-import { createGoogleGenerativeAI } from './google-provider';
-import { describe, it, expect, vi } from 'vitest';
-vi.mock('./version', () => ({
-    VERSION: '0.0.0-test',
-}));
-const dummyEmbeddings = [
-    [0.1, 0.2, 0.3, 0.4, 0.5],
-    [0.6, 0.7, 0.8, 0.9, 1.0],
-];
-const testValues = ['sunny day at the beach', 'rainy day in the city'];
-const provider = createGoogleGenerativeAI({ apiKey: 'test-api-key' });
-const model = provider.textEmbeddingModel('gemini-embedding-001');
-const BATCH_URL = 'https://cloudcode-pa.googleapis.com/models/gemini-embedding-001\\:batchEmbedContents';
-const SINGLE_URL = 'https://cloudcode-pa.googleapis.com/models/gemini-embedding-001\\:embedContent';
-const server = createTestServer({
-    [SINGLE_URL]: {},
-    [BATCH_URL]: {},
-});
-describe('GoogleGenerativeAIEmbeddingModel', () => {
-    function prepareBatchJsonResponse({ embeddings = dummyEmbeddings, headers, } = {}) {
-        server.urls[BATCH_URL].response = {
-            type: 'json-value',
-            headers,
-            body: {
-                embeddings: embeddings.map(embedding => ({ values: embedding })),
-            },
-        };
-    }
-    function prepareSingleJsonResponse({ embeddings = dummyEmbeddings, headers, } = {}) {
-        server.urls[SINGLE_URL].response = {
-            type: 'json-value',
-            headers,
-            body: {
-                embedding: { values: embeddings[0] },
-            },
-        };
-    }
-    it('should extract embedding', async () => {
-        prepareBatchJsonResponse();
-        const { embeddings } = await model.doEmbed({ values: testValues });
-        expect(embeddings).toStrictEqual(dummyEmbeddings);
-    });
-    it('should expose the raw response', async () => {
-        prepareBatchJsonResponse({
-            headers: {
-                'test-header': 'test-value',
-            },
-        });
-        const { response } = await model.doEmbed({ values: testValues });
-        expect(response === null || response === void 0 ? void 0 : response.headers).toStrictEqual({
-            // default headers:
-            'content-length': '80',
-            'content-type': 'application/json',
-            // custom header
-            'test-header': 'test-value',
-        });
-        expect(response).toMatchSnapshot();
-    });
-    it('should pass the model and the values', async () => {
-        prepareBatchJsonResponse();
-        await model.doEmbed({ values: testValues });
-        expect(await server.calls[0].requestBodyJson).toStrictEqual({
-            requests: testValues.map(value => ({
-                model: 'models/gemini-embedding-001',
-                content: { role: 'user', parts: [{ text: value }] },
-            })),
-        });
-    });
-    it('should pass the outputDimensionality setting', async () => {
-        prepareBatchJsonResponse();
-        await provider.embedding('gemini-embedding-001').doEmbed({
-            values: testValues,
-            providerOptions: {
-                google: { outputDimensionality: 64 },
-            },
-        });
-        expect(await server.calls[0].requestBodyJson).toStrictEqual({
-            requests: testValues.map(value => ({
-                model: 'models/gemini-embedding-001',
-                content: { role: 'user', parts: [{ text: value }] },
-                outputDimensionality: 64,
-            })),
-        });
-    });
-    it('should pass the taskType setting', async () => {
-        prepareBatchJsonResponse();
-        await provider.embedding('gemini-embedding-001').doEmbed({
-            values: testValues,
-            providerOptions: { google: { taskType: 'SEMANTIC_SIMILARITY' } },
-        });
-        expect(await server.calls[0].requestBodyJson).toStrictEqual({
-            requests: testValues.map(value => ({
-                model: 'models/gemini-embedding-001',
-                content: { role: 'user', parts: [{ text: value }] },
-                taskType: 'SEMANTIC_SIMILARITY',
-            })),
-        });
-    });
-    it('should pass headers', async () => {
-        prepareBatchJsonResponse();
-        const provider = createGoogleGenerativeAI({
-            apiKey: 'test-api-key',
-            headers: {
-                'Custom-Provider-Header': 'provider-header-value',
-            },
-        });
-        await provider.embedding('gemini-embedding-001').doEmbed({
-            values: testValues,
-            headers: {
-                'Custom-Request-Header': 'request-header-value',
-            },
-        });
-        expect(server.calls[0].requestHeaders).toStrictEqual({
-            'authorization': 'Bearer test-api-key',
-            'content-type': 'application/json',
-            'custom-provider-header': 'provider-header-value',
-            'custom-request-header': 'request-header-value',
-        });
-        expect(server.calls[0].requestUserAgent).toContain(`ai-sdk/google-cloudassist/0.0.0-test`);
-    });
-    it('should throw an error if too many values are provided', async () => {
-        const model = new GoogleGenerativeAIEmbeddingModel('gemini-embedding-001', {
-            provider: 'google.generative-ai',
-            baseURL: 'https://generativelanguage.googleapis.com/v1beta',
-            headers: () => ({}),
-        });
-        const tooManyValues = Array(2049).fill('test');
-        await expect(model.doEmbed({ values: tooManyValues })).rejects.toThrow('Too many values for a single embedding call. The google.generative-ai model "gemini-embedding-001" can only embed up to 2048 values per call, but 2049 values were provided.');
-    });
-    it('should use the batch embeddings endpoint', async () => {
-        prepareBatchJsonResponse();
-        const model = provider.textEmbeddingModel('gemini-embedding-001');
-        await model.doEmbed({
-            values: testValues,
-        });
-        expect(server.calls[0].requestUrl).toBe('https://cloudcode-pa.googleapis.com/models/gemini-embedding-001:batchEmbedContents');
-    });
-    it('should use the single embeddings endpoint', async () => {
-        prepareSingleJsonResponse();
-        const model = provider.textEmbeddingModel('gemini-embedding-001');
-        await model.doEmbed({
-            values: [testValues[0]],
-        });
-        expect(server.calls[0].requestUrl).toBe('https://cloudcode-pa.googleapis.com/models/gemini-embedding-001:embedContent');
-    });
-});

package/dist/google-generative-ai-embedding-options.d.ts

@@ -1,8 +0,0 @@
-import { type InferValidator } from '@ai-sdk/provider-utils';
-export type GoogleGenerativeAIEmbeddingModelId = 'gemini-embedding-001' | 'text-embedding-004' | (string & {});
-export declare const googleGenerativeAIEmbeddingProviderOptions: import("@ai-sdk/provider-utils").LazySchema<{
-    outputDimensionality?: number | undefined;
-    taskType?: "SEMANTIC_SIMILARITY" | "CLASSIFICATION" | "CLUSTERING" | "RETRIEVAL_DOCUMENT" | "RETRIEVAL_QUERY" | "QUESTION_ANSWERING" | "FACT_VERIFICATION" | "CODE_RETRIEVAL_QUERY" | undefined;
-}>;
-export type GoogleGenerativeAIEmbeddingProviderOptions = InferValidator<typeof googleGenerativeAIEmbeddingProviderOptions>;
-//# sourceMappingURL=google-generative-ai-embedding-options.d.ts.map

package/dist/google-generative-ai-embedding-options.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"google-generative-ai-embedding-options.d.ts","sourceRoot":"","sources":["../src/google-generative-ai-embedding-options.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,KAAK,cAAc,EAGpB,MAAM,wBAAwB,CAAC;AAGhC,MAAM,MAAM,kCAAkC,GAC1C,sBAAsB,GACtB,oBAAoB,GACpB,CAAC,MAAM,GAAG,EAAE,CAAC,CAAC;AAElB,eAAO,MAAM,0CAA0C;;;EAmCtD,CAAC;AAEF,MAAM,MAAM,0CAA0C,GAAG,cAAc,CACrE,OAAO,0CAA0C,CAClD,CAAC"}

package/dist/google-generative-ai-embedding-options.js

@@ -1,33 +0,0 @@
-import { lazySchema, zodSchema, } from '@ai-sdk/provider-utils';
-import { z } from 'zod/v4';
-export const googleGenerativeAIEmbeddingProviderOptions = lazySchema(() => zodSchema(z.object({
-    /**
-     * Optional. Optional reduced dimension for the output embedding.
-     * If set, excessive values in the output embedding are truncated from the end.
-     */
-    outputDimensionality: z.number().optional(),
-    /**
-     * Optional. Specifies the task type for generating embeddings.
-     * Supported task types:
-     * - SEMANTIC_SIMILARITY: Optimized for text similarity.
-     * - CLASSIFICATION: Optimized for text classification.
-     * - CLUSTERING: Optimized for clustering texts based on similarity.
-     * - RETRIEVAL_DOCUMENT: Optimized for document retrieval.
-     * - RETRIEVAL_QUERY: Optimized for query-based retrieval.
-     * - QUESTION_ANSWERING: Optimized for answering questions.
-     * - FACT_VERIFICATION: Optimized for verifying factual information.
-     * - CODE_RETRIEVAL_QUERY: Optimized for retrieving code blocks based on natural language queries.
-     */
-    taskType: z
-        .enum([
-        'SEMANTIC_SIMILARITY',
-        'CLASSIFICATION',
-        'CLUSTERING',
-        'RETRIEVAL_DOCUMENT',
-        'RETRIEVAL_QUERY',
-        'QUESTION_ANSWERING',
-        'FACT_VERIFICATION',
-        'CODE_RETRIEVAL_QUERY',
-    ])
-        .optional(),
-})));

package/dist/google-generative-ai-image-model.d.ts

@@ -1,31 +0,0 @@
-import { ImageModelV2 } from '@ai-sdk/provider';
-import { type InferValidator } from '@ai-sdk/provider-utils';
-import { GoogleGenerativeAIImageModelId, GoogleGenerativeAIImageSettings } from './google-generative-ai-image-settings';
-import { FetchFunction, Resolvable } from '@ai-sdk/provider-utils';
-interface GoogleGenerativeAIImageModelConfig {
-    provider: string;
-    baseURL: string;
-    headers?: Resolvable<Record<string, string | undefined>>;
-    fetch?: FetchFunction;
-    generateId?: () => string;
-    _internal?: {
-        currentDate?: () => Date;
-    };
-}
-export declare class GoogleGenerativeAIImageModel implements ImageModelV2 {
-    readonly modelId: GoogleGenerativeAIImageModelId;
-    private readonly settings;
-    private readonly config;
-    readonly specificationVersion = "v2";
-    get maxImagesPerCall(): number;
-    get provider(): string;
-    constructor(modelId: GoogleGenerativeAIImageModelId, settings: GoogleGenerativeAIImageSettings, config: GoogleGenerativeAIImageModelConfig);
-    doGenerate(options: Parameters<ImageModelV2['doGenerate']>[0]): Promise<Awaited<ReturnType<ImageModelV2['doGenerate']>>>;
-}
-declare const googleImageProviderOptionsSchema: import("@ai-sdk/provider-utils").LazySchema<{
-    personGeneration?: "dont_allow" | "allow_adult" | "allow_all" | null | undefined;
-    aspectRatio?: "1:1" | "3:4" | "4:3" | "9:16" | "16:9" | null | undefined;
-}>;
-export type GoogleGenerativeAIImageProviderOptions = InferValidator<typeof googleImageProviderOptionsSchema>;
-export {};
-//# sourceMappingURL=google-generative-ai-image-model.d.ts.map

package/dist/google-generative-ai-image-model.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"google-generative-ai-image-model.d.ts","sourceRoot":"","sources":["../src/google-generative-ai-image-model.ts"],"names":[],"mappings":"AAAA,OAAO,EAAE,YAAY,EAA2B,MAAM,kBAAkB,CAAC;AACzE,OAAO,EAGL,KAAK,cAAc,EAMpB,MAAM,wBAAwB,CAAC;AAGhC,OAAO,EACL,8BAA8B,EAC9B,+BAA+B,EAChC,MAAM,uCAAuC,CAAC;AAC/C,OAAO,EAAE,aAAa,EAAE,UAAU,EAAE,MAAM,wBAAwB,CAAC;AAEnE,UAAU,kCAAkC;IAC1C,QAAQ,EAAE,MAAM,CAAC;IACjB,OAAO,EAAE,MAAM,CAAC;IAChB,OAAO,CAAC,EAAE,UAAU,CAAC,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC,CAAC;IACzD,KAAK,CAAC,EAAE,aAAa,CAAC;IACtB,UAAU,CAAC,EAAE,MAAM,MAAM,CAAC;IAC1B,SAAS,CAAC,EAAE;QACV,WAAW,CAAC,EAAE,MAAM,IAAI,CAAC;KAC1B,CAAC;CACH;AAED,qBAAa,4BAA6B,YAAW,YAAY;IAa7D,QAAQ,CAAC,OAAO,EAAE,8BAA8B;IAChD,OAAO,CAAC,QAAQ,CAAC,QAAQ;IACzB,OAAO,CAAC,QAAQ,CAAC,MAAM;IAdzB,QAAQ,CAAC,oBAAoB,QAAQ;IAErC,IAAI,gBAAgB,IAAI,MAAM,CAG7B;IAED,IAAI,QAAQ,IAAI,MAAM,CAErB;gBAGU,OAAO,EAAE,8BAA8B,EAC/B,QAAQ,EAAE,+BAA+B,EACzC,MAAM,EAAE,kCAAkC;IAGvD,UAAU,CACd,OAAO,EAAE,UAAU,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC,CAAC,GACjD,OAAO,CAAC,OAAO,CAAC,UAAU,CAAC,YAAY,CAAC,YAAY,CAAC,CAAC,CAAC,CAAC;CAwF5D;AAeD,QAAA,MAAM,gCAAgC;;;EASrC,CAAC;AAEF,MAAM,MAAM,sCAAsC,GAAG,cAAc,CACjE,OAAO,gCAAgC,CACxC,CAAC"}

package/dist/google-generative-ai-image-model.js

@@ -1,96 +0,0 @@
-import { combineHeaders, createJsonResponseHandler, lazySchema, parseProviderOptions, postJsonToApi, resolve, zodSchema, } from '@ai-sdk/provider-utils';
-import { z } from 'zod/v4';
-import { googleFailedResponseHandler } from './google-error';
-export class GoogleGenerativeAIImageModel {
-    get maxImagesPerCall() {
-        var _a;
-        // https://ai.google.dev/gemini-api/docs/imagen#imagen-model
-        return (_a = this.settings.maxImagesPerCall) !== null && _a !== void 0 ? _a : 4;
-    }
-    get provider() {
-        return this.config.provider;
-    }
-    constructor(modelId, settings, config) {
-        this.modelId = modelId;
-        this.settings = settings;
-        this.config = config;
-        this.specificationVersion = 'v2';
-    }
-    async doGenerate(options) {
-        var _a, _b, _c;
-        const { prompt, n = 1, size = '1024x1024', aspectRatio = '1:1', seed, providerOptions, headers, abortSignal, } = options;
-        const warnings = [];
-        if (size != null) {
-            warnings.push({
-                type: 'unsupported-setting',
-                setting: 'size',
-                details: 'This model does not support the `size` option. Use `aspectRatio` instead.',
-            });
-        }
-        if (seed != null) {
-            warnings.push({
-                type: 'unsupported-setting',
-                setting: 'seed',
-                details: 'This model does not support the `seed` option through this provider.',
-            });
-        }
-        const googleOptions = await parseProviderOptions({
-            provider: 'google',
-            providerOptions,
-            schema: googleImageProviderOptionsSchema,
-        });
-        const currentDate = (_c = (_b = (_a = this.config._internal) === null || _a === void 0 ? void 0 : _a.currentDate) === null || _b === void 0 ? void 0 : _b.call(_a)) !== null && _c !== void 0 ? _c : new Date();
-        const parameters = {
-            sampleCount: n,
-        };
-        if (aspectRatio != null) {
-            parameters.aspectRatio = aspectRatio;
-        }
-        if (googleOptions) {
-            Object.assign(parameters, googleOptions);
-        }
-        const body = {
-            instances: [{ prompt }],
-            parameters,
-        };
-        const { responseHeaders, value: response } = await postJsonToApi({
-            url: `${this.config.baseURL}/models/${this.modelId}:predict`,
-            headers: combineHeaders(await resolve(this.config.headers), headers),
-            body,
-            failedResponseHandler: googleFailedResponseHandler,
-            successfulResponseHandler: createJsonResponseHandler(googleImageResponseSchema),
-            abortSignal,
-            fetch: this.config.fetch,
-        });
-        return {
-            images: response.predictions.map((p) => p.bytesBase64Encoded),
-            warnings: warnings !== null && warnings !== void 0 ? warnings : [],
-            providerMetadata: {
-                google: {
-                    images: response.predictions.map(prediction => ({
-                    // Add any prediction-specific metadata here
-                    })),
-                },
-            },
-            response: {
-                timestamp: currentDate,
-                modelId: this.modelId,
-                headers: responseHeaders,
-            },
-        };
-    }
-}
-// minimal version of the schema
-const googleImageResponseSchema = lazySchema(() => zodSchema(z.object({
-    predictions: z
-        .array(z.object({ bytesBase64Encoded: z.string() }))
-        .default([]),
-})));
-// Note: For the initial GA launch of Imagen 3, safety filters are not configurable.
-// https://ai.google.dev/gemini-api/docs/imagen#imagen-model
-const googleImageProviderOptionsSchema = lazySchema(() => zodSchema(z.object({
-    personGeneration: z
-        .enum(['dont_allow', 'allow_adult', 'allow_all'])
-        .nullish(),
-    aspectRatio: z.enum(['1:1', '3:4', '4:3', '9:16', '16:9']).nullish(),
-})));

package/dist/google-generative-ai-image-model.test.d.ts

@@ -1,2 +0,0 @@
-export {};
-//# sourceMappingURL=google-generative-ai-image-model.test.d.ts.map

package/dist/google-generative-ai-image-model.test.d.ts.map

@@ -1 +0,0 @@
-{"version":3,"file":"google-generative-ai-image-model.test.d.ts","sourceRoot":"","sources":["../src/google-generative-ai-image-model.test.ts"],"names":[],"mappings":""}