cloudassist-ai-provider 0.0.1

package/dist/google-prepare-tools.test.js

@@ -0,0 +1,447 @@
+import { expect, it } from 'vitest';
+import { prepareTools } from './google-prepare-tools';
+it('should return undefined tools and tool_choice when tools are null', () => {
+    const result = prepareTools({
+        tools: undefined,
+        modelId: 'gemini-2.5-flash',
+    });
+    expect(result).toEqual({
+        tools: undefined,
+        tool_choice: undefined,
+        toolWarnings: [],
+    });
+});
+it('should return undefined tools and tool_choice when tools are empty', () => {
+    const result = prepareTools({ tools: [], modelId: 'gemini-2.5-flash' });
+    expect(result).toEqual({
+        tools: undefined,
+        tool_choice: undefined,
+        toolWarnings: [],
+    });
+});
+it('should correctly prepare function tools', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'function',
+                name: 'testFunction',
+                description: 'A test function',
+                inputSchema: { type: 'object', properties: {} },
+            },
+        ],
+        modelId: 'gemini-2.5-flash',
+    });
+    expect(result.tools).toEqual([
+        {
+            functionDeclarations: [
+                {
+                    name: 'testFunction',
+                    description: 'A test function',
+                    parameters: undefined,
+                },
+            ],
+        },
+    ]);
+    expect(result.toolConfig).toBeUndefined();
+    expect(result.toolWarnings).toEqual([]);
+});
+it('should correctly prepare provider-defined tools as array', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'provider-defined',
+                id: 'google.google_search',
+                name: 'google_search',
+                args: {},
+            },
+            {
+                type: 'provider-defined',
+                id: 'google.url_context',
+                name: 'url_context',
+                args: {},
+            },
+            {
+                type: 'provider-defined',
+                id: 'google.file_search',
+                name: 'file_search',
+                args: { fileSearchStoreNames: ['projects/foo/fileSearchStores/bar'] },
+            },
+        ],
+        modelId: 'gemini-2.5-flash',
+    });
+    expect(result.tools).toEqual([
+        { googleSearch: {} },
+        { urlContext: {} },
+        {
+            fileSearch: {
+                fileSearchStoreNames: ['projects/foo/fileSearchStores/bar'],
+            },
+        },
+    ]);
+    expect(result.toolConfig).toBeUndefined();
+    expect(result.toolWarnings).toEqual([]);
+});
+it('should correctly prepare single provider-defined tool', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'provider-defined',
+                id: 'google.google_search',
+                name: 'google_search',
+                args: {},
+            },
+        ],
+        modelId: 'gemini-2.5-flash',
+    });
+    expect(result.tools).toEqual([{ googleSearch: {} }]);
+    expect(result.toolConfig).toBeUndefined();
+    expect(result.toolWarnings).toEqual([]);
+});
+it('should add warnings for unsupported tools', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'provider-defined',
+                id: 'unsupported.tool',
+                name: 'unsupported_tool',
+                args: {},
+            },
+        ],
+        modelId: 'gemini-2.5-flash',
+    });
+    expect(result.tools).toBeUndefined();
+    expect(result.toolConfig).toBeUndefined();
+    expect(result.toolWarnings).toMatchInlineSnapshot(`
+    [
+      {
+        "tool": {
+          "args": {},
+          "id": "unsupported.tool",
+          "name": "unsupported_tool",
+          "type": "provider-defined",
+        },
+        "type": "unsupported-tool",
+      },
+    ]
+  `);
+});
+it('should add warnings for file search on unsupported models', () => {
+    const tool = {
+        type: 'provider-defined',
+        id: 'google.file_search',
+        name: 'file_search',
+        args: { fileSearchStoreNames: ['projects/foo/fileSearchStores/bar'] },
+    };
+    const result = prepareTools({
+        tools: [tool],
+        modelId: 'gemini-1.5-flash-8b',
+    });
+    expect(result.tools).toBeUndefined();
+    expect(result.toolWarnings).toMatchInlineSnapshot(`
+    [
+      {
+        "details": "The file search tool is only supported with Gemini 2.5 models.",
+        "tool": {
+          "args": {
+            "fileSearchStoreNames": [
+              "projects/foo/fileSearchStores/bar",
+            ],
+          },
+          "id": "google.file_search",
+          "name": "file_search",
+          "type": "provider-defined",
+        },
+        "type": "unsupported-tool",
+      },
+    ]
+  `);
+});
+it('should correctly prepare file search tool', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'provider-defined',
+                id: 'google.file_search',
+                name: 'file_search',
+                args: {
+                    fileSearchStoreNames: ['projects/foo/fileSearchStores/bar'],
+                    metadataFilter: 'author=Robert Graves',
+                    topK: 5,
+                },
+            },
+        ],
+        modelId: 'gemini-2.5-pro',
+    });
+    expect(result.tools).toEqual([
+        {
+            fileSearch: {
+                fileSearchStoreNames: ['projects/foo/fileSearchStores/bar'],
+                metadataFilter: 'author=Robert Graves',
+                topK: 5,
+            },
+        },
+    ]);
+    expect(result.toolWarnings).toEqual([]);
+});
+it('should handle tool choice "auto"', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'function',
+                name: 'testFunction',
+                description: 'Test',
+                inputSchema: {},
+            },
+        ],
+        toolChoice: { type: 'auto' },
+        modelId: 'gemini-2.5-flash',
+    });
+    expect(result.toolConfig).toEqual({
+        functionCallingConfig: { mode: 'AUTO' },
+    });
+});
+it('should handle tool choice "required"', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'function',
+                name: 'testFunction',
+                description: 'Test',
+                inputSchema: {},
+            },
+        ],
+        toolChoice: { type: 'required' },
+        modelId: 'gemini-2.5-flash',
+    });
+    expect(result.toolConfig).toEqual({
+        functionCallingConfig: { mode: 'ANY' },
+    });
+});
+it('should handle tool choice "none"', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'function',
+                name: 'testFunction',
+                description: 'Test',
+                inputSchema: {},
+            },
+        ],
+        toolChoice: { type: 'none' },
+        modelId: 'gemini-2.5-flash',
+    });
+    expect(result.tools).toEqual([
+        {
+            functionDeclarations: [
+                {
+                    name: 'testFunction',
+                    description: 'Test',
+                    parameters: {},
+                },
+            ],
+        },
+    ]);
+    expect(result.toolConfig).toEqual({
+        functionCallingConfig: { mode: 'NONE' },
+    });
+});
+it('should handle tool choice "tool"', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'function',
+                name: 'testFunction',
+                description: 'Test',
+                inputSchema: {},
+            },
+        ],
+        toolChoice: { type: 'tool', toolName: 'testFunction' },
+        modelId: 'gemini-2.5-flash',
+    });
+    expect(result.toolConfig).toEqual({
+        functionCallingConfig: {
+            mode: 'ANY',
+            allowedFunctionNames: ['testFunction'],
+        },
+    });
+});
+it('should warn when mixing function and provider-defined tools', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'function',
+                name: 'testFunction',
+                description: 'A test function',
+                inputSchema: { type: 'object', properties: {} },
+            },
+            {
+                type: 'provider-defined',
+                id: 'google.google_search',
+                name: 'google_search',
+                args: {},
+            },
+        ],
+        modelId: 'gemini-2.5-flash',
+    });
+    // Should only include provider-defined tools as array
+    expect(result.tools).toEqual([{ googleSearch: {} }]);
+    // Should have warning about mixed tool types
+    expect(result.toolWarnings).toEqual([
+        {
+            type: 'unsupported-tool',
+            tool: {
+                type: 'function',
+                name: 'testFunction',
+                description: 'A test function',
+                inputSchema: { type: 'object', properties: {} },
+            },
+            details: 'Cannot mix function tools with provider-defined tools in the same request. Falling back to provider-defined tools only. The following function tools will be ignored: testFunction. Please use either function tools or provider-defined tools, but not both.',
+        },
+    ]);
+    expect(result.toolConfig).toBeUndefined();
+});
+it('should handle tool choice with mixed tools (provider-defined tools only)', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'function',
+                name: 'testFunction',
+                description: 'A test function',
+                inputSchema: { type: 'object', properties: {} },
+            },
+            {
+                type: 'provider-defined',
+                id: 'google.google_search',
+                name: 'google_search',
+                args: {},
+            },
+        ],
+        toolChoice: { type: 'auto' },
+        modelId: 'gemini-2.5-flash',
+    });
+    // Should only include provider-defined tools as array
+    expect(result.tools).toEqual([{ googleSearch: {} }]);
+    // Should apply tool choice to provider-defined tools
+    expect(result.toolConfig).toEqual(undefined);
+    // Should have warning about mixed tool types
+    expect(result.toolWarnings).toEqual([
+        {
+            type: 'unsupported-tool',
+            tool: {
+                type: 'function',
+                name: 'testFunction',
+                description: 'A test function',
+                inputSchema: { type: 'object', properties: {} },
+            },
+            details: 'Cannot mix function tools with provider-defined tools in the same request. Falling back to provider-defined tools only. The following function tools will be ignored: testFunction. Please use either function tools or provider-defined tools, but not both.',
+        },
+    ]);
+});
+it('should handle latest modelId for provider-defined tools correctly', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'provider-defined',
+                id: 'google.google_search',
+                name: 'google_search',
+                args: {},
+            },
+        ],
+        modelId: 'gemini-flash-latest',
+    });
+    expect(result.tools).toEqual([{ googleSearch: {} }]);
+    expect(result.toolConfig).toBeUndefined();
+    expect(result.toolWarnings).toEqual([]);
+});
+it('should handle gemini-3 modelId for provider-defined tools correctly', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'provider-defined',
+                id: 'google.google_search',
+                name: 'google_search',
+                args: {},
+            },
+        ],
+        modelId: 'gemini-3-pro-preview',
+    });
+    expect(result.tools).toEqual([{ googleSearch: {} }]);
+    expect(result.toolConfig).toBeUndefined();
+    expect(result.toolWarnings).toEqual([]);
+});
+it('should handle code execution tool', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'provider-defined',
+                id: 'google.code_execution',
+                name: 'code_execution',
+                args: {},
+            },
+        ],
+        modelId: 'gemini-2.5-flash',
+    });
+    expect(result.tools).toEqual([{ codeExecution: {} }]);
+    expect(result.toolConfig).toBeUndefined();
+    expect(result.toolWarnings).toEqual([]);
+});
+it('should handle url context tool alone', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'provider-defined',
+                id: 'google.url_context',
+                name: 'url_context',
+                args: {},
+            },
+        ],
+        modelId: 'gemini-2.5-flash',
+    });
+    expect(result.tools).toEqual([{ urlContext: {} }]);
+    expect(result.toolConfig).toBeUndefined();
+    expect(result.toolWarnings).toEqual([]);
+});
+it('should handle google maps tool', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'provider-defined',
+                id: 'google.google_maps',
+                name: 'google_maps',
+                args: {},
+            },
+        ],
+        modelId: 'gemini-2.5-flash',
+    });
+    expect(result.tools).toEqual([{ googleMaps: {} }]);
+    expect(result.toolConfig).toBeUndefined();
+    expect(result.toolWarnings).toEqual([]);
+});
+it('should add warnings for google maps on unsupported models', () => {
+    const result = prepareTools({
+        tools: [
+            {
+                type: 'provider-defined',
+                id: 'google.google_maps',
+                name: 'google_maps',
+                args: {},
+            },
+        ],
+        modelId: 'gemini-1.5-flash',
+    });
+    expect(result.tools).toBeUndefined();
+    expect(result.toolWarnings).toMatchInlineSnapshot(`
+    [
+      {
+        "details": "The Google Maps grounding tool is not supported with Gemini models other than Gemini 2 or newer.",
+        "tool": {
+          "args": {},
+          "id": "google.google_maps",
+          "name": "google_maps",
+          "type": "provider-defined",
+        },
+        "type": "unsupported-tool",
+      },
+    ]
+  `);
+});

package/dist/google-provider.d.ts

@@ -0,0 +1,65 @@
+import { EmbeddingModelV2, LanguageModelV2, ProviderV2, ImageModelV2 } from '@ai-sdk/provider';
+import { FetchFunction } from '@ai-sdk/provider-utils';
+import { GoogleGenerativeAIEmbeddingModelId } from './google-generative-ai-embedding-options';
+import { GoogleGenerativeAIModelId } from './google-generative-ai-options';
+import { googleTools } from './google-tools';
+import { GoogleGenerativeAIImageSettings, GoogleGenerativeAIImageModelId } from './google-generative-ai-image-settings';
+export interface GoogleGenerativeAIProvider extends ProviderV2 {
+    (modelId: GoogleGenerativeAIModelId): LanguageModelV2;
+    languageModel(modelId: GoogleGenerativeAIModelId): LanguageModelV2;
+    chat(modelId: GoogleGenerativeAIModelId): LanguageModelV2;
+    /**
+  Creates a model for image generation.
+   */
+    image(modelId: GoogleGenerativeAIImageModelId, settings?: GoogleGenerativeAIImageSettings): ImageModelV2;
+    /**
+     * @deprecated Use `chat()` instead.
+     */
+    generativeAI(modelId: GoogleGenerativeAIModelId): LanguageModelV2;
+    /**
+  @deprecated Use `textEmbedding()` instead.
+     */
+    embedding(modelId: GoogleGenerativeAIEmbeddingModelId): EmbeddingModelV2<string>;
+    textEmbedding(modelId: GoogleGenerativeAIEmbeddingModelId): EmbeddingModelV2<string>;
+    textEmbeddingModel(modelId: GoogleGenerativeAIEmbeddingModelId): EmbeddingModelV2<string>;
+    tools: typeof googleTools;
+}
+export interface GoogleGenerativeAIProviderSettings {
+    /**
+  Use a different URL prefix for API calls, e.g. to use proxy servers.
+  The default prefix is `https://cloudcode-pa.googleapis.com`.
+     */
+    baseURL?: string;
+    /**
+  Google Cloud access token used for Bearer authentication.
+  It defaults to the `GOOGLE_CLOUD_ACCESS_TOKEN` environment variable.
+     */
+    apiKey?: string;
+    /**
+  Custom headers to include in the requests.
+       */
+    headers?: Record<string, string | undefined>;
+    /**
+  Custom fetch implementation. You can use it as a middleware to intercept requests,
+  or to provide a custom fetch implementation for e.g. testing.
+      */
+    fetch?: FetchFunction;
+    /**
+  Optional function to generate a unique ID for each request.
+       */
+    generateId?: () => string;
+    /**
+     * Custom provider name
+     * Defaults to 'google.generative-ai'.
+     */
+    name?: string;
+}
+/**
+Create a Google Generative AI provider instance.
+ */
+export declare function createGoogleGenerativeAI(options?: GoogleGenerativeAIProviderSettings): GoogleGenerativeAIProvider;
+/**
+Default Google Generative AI provider instance.
+ */
+export declare const google: GoogleGenerativeAIProvider;
+//# sourceMappingURL=google-provider.d.ts.map

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

@@ -0,0 +1 @@
+{"version":3,"file":"google-provider.d.ts","sourceRoot":"","sources":["../src/google-provider.ts"],"names":[],"mappings":"AAAA,OAAO,EACL,gBAAgB,EAChB,eAAe,EACf,UAAU,EACV,YAAY,EACb,MAAM,kBAAkB,CAAC;AAC1B,OAAO,EACL,aAAa,EAKd,MAAM,wBAAwB,CAAC;AAGhC,OAAO,EAAE,kCAAkC,EAAE,MAAM,0CAA0C,CAAC;AAE9F,OAAO,EAAE,yBAAyB,EAAE,MAAM,gCAAgC,CAAC;AAC3E,OAAO,EAAE,WAAW,EAAE,MAAM,gBAAgB,CAAC;AAE7C,OAAO,EACL,+BAA+B,EAC/B,8BAA8B,EAC/B,MAAM,uCAAuC,CAAC;AAG/C,MAAM,WAAW,0BAA2B,SAAQ,UAAU;IAC5D,CAAC,OAAO,EAAE,yBAAyB,GAAG,eAAe,CAAC;IAEtD,aAAa,CAAC,OAAO,EAAE,yBAAyB,GAAG,eAAe,CAAC;IAEnE,IAAI,CAAC,OAAO,EAAE,yBAAyB,GAAG,eAAe,CAAC;IAE1D;;KAEC;IACD,KAAK,CACH,OAAO,EAAE,8BAA8B,EACvC,QAAQ,CAAC,EAAE,+BAA+B,GACzC,YAAY,CAAC;IAEhB;;OAEG;IACH,YAAY,CAAC,OAAO,EAAE,yBAAyB,GAAG,eAAe,CAAC;IAElE;;OAEG;IACH,SAAS,CACP,OAAO,EAAE,kCAAkC,GAC1C,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAE5B,aAAa,CACX,OAAO,EAAE,kCAAkC,GAC1C,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAE5B,kBAAkB,CAChB,OAAO,EAAE,kCAAkC,GAC1C,gBAAgB,CAAC,MAAM,CAAC,CAAC;IAE5B,KAAK,EAAE,OAAO,WAAW,CAAC;CAC3B;AAED,MAAM,WAAW,kCAAkC;IACjD;;;OAGG;IACH,OAAO,CAAC,EAAE,MAAM,CAAC;IAEjB;;;OAGG;IACH,MAAM,CAAC,EAAE,MAAM,CAAC;IAEhB;;SAEK;IACL,OAAO,CAAC,EAAE,MAAM,CAAC,MAAM,EAAE,MAAM,GAAG,SAAS,CAAC,CAAC;IAE7C;;;QAGI;IACJ,KAAK,CAAC,EAAE,aAAa,CAAC;IAEtB;;SAEK;IACL,UAAU,CAAC,EAAE,MAAM,MAAM,CAAC;IAE1B;;;OAGG;IACH,IAAI,CAAC,EAAE,MAAM,CAAC;CACf;AAED;;GAEG;AACH,wBAAgB,wBAAwB,CACtC,OAAO,GAAE,kCAAuC,GAC/C,0BAA0B,CAgF5B;AAED;;GAEG;AACH,eAAO,MAAM,MAAM,4BAA6B,CAAC"}

package/dist/google-provider.js

@@ -0,0 +1,74 @@
+import { generateId, loadApiKey, withoutTrailingSlash, withUserAgentSuffix, } from '@ai-sdk/provider-utils';
+import { VERSION } from './version';
+import { GoogleGenerativeAIEmbeddingModel } from './google-generative-ai-embedding-model';
+import { GoogleGenerativeAILanguageModel } from './google-generative-ai-language-model';
+import { googleTools } from './google-tools';
+import { GoogleGenerativeAIImageModel } from './google-generative-ai-image-model';
+/**
+Create a Google Generative AI provider instance.
+ */
+export function createGoogleGenerativeAI(options = {}) {
+    var _a, _b;
+    const baseURL = (_a = withoutTrailingSlash(options.baseURL)) !== null && _a !== void 0 ? _a : 'https://cloudcode-pa.googleapis.com';
+    const providerName = (_b = options.name) !== null && _b !== void 0 ? _b : 'google.generative-ai';
+    const getHeaders = () => withUserAgentSuffix({
+        Authorization: `Bearer ${loadApiKey({
+            apiKey: options.apiKey,
+            environmentVariableName: 'GOOGLE_CLOUD_ACCESS_TOKEN',
+            description: 'Google Cloud Access Token',
+        })}`,
+        ...options.headers,
+    }, `ai-sdk/google-cloudassist/${VERSION}`);
+    const createChatModel = (modelId) => {
+        var _a;
+        return new GoogleGenerativeAILanguageModel(modelId, {
+            provider: providerName,
+            baseURL,
+            headers: getHeaders,
+            generateId: (_a = options.generateId) !== null && _a !== void 0 ? _a : generateId,
+            supportedUrls: () => ({
+                '*': [
+                    // Google Generative Language "files" endpoint
+                    // e.g. https://generativelanguage.googleapis.com/v1beta/files/...
+                    new RegExp(`^${baseURL}/files/.*$`),
+                    // YouTube URLs (public or unlisted videos)
+                    new RegExp(`^https://(?:www\\.)?youtube\\.com/watch\\?v=[\\w-]+(?:&[\\w=&.-]*)?$`),
+                    new RegExp(`^https://youtu\\.be/[\\w-]+(?:\\?[\\w=&.-]*)?$`),
+                ],
+            }),
+            fetch: options.fetch,
+        });
+    };
+    const createEmbeddingModel = (modelId) => new GoogleGenerativeAIEmbeddingModel(modelId, {
+        provider: providerName,
+        baseURL,
+        headers: getHeaders,
+        fetch: options.fetch,
+    });
+    const createImageModel = (modelId, settings = {}) => new GoogleGenerativeAIImageModel(modelId, settings, {
+        provider: providerName,
+        baseURL,
+        headers: getHeaders,
+        fetch: options.fetch,
+    });
+    const provider = function (modelId) {
+        if (new.target) {
+            throw new Error('The Google Generative AI model function cannot be called with the new keyword.');
+        }
+        return createChatModel(modelId);
+    };
+    provider.languageModel = createChatModel;
+    provider.chat = createChatModel;
+    provider.generativeAI = createChatModel;
+    provider.embedding = createEmbeddingModel;
+    provider.textEmbedding = createEmbeddingModel;
+    provider.textEmbeddingModel = createEmbeddingModel;
+    provider.image = createImageModel;
+    provider.imageModel = createImageModel;
+    provider.tools = googleTools;
+    return provider;
+}
+/**
+Default Google Generative AI provider instance.
+ */
+export const google = createGoogleGenerativeAI();

package/dist/google-provider.test.d.ts

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

package/dist/google-provider.test.d.ts.map

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