@ai-sdk/google-vertex 1.0.4 → 2.0.1

package/dist/index.d.ts

@@ -1,30 +1,10 @@
 import { ProviderV1, LanguageModelV1 } from '@ai-sdk/provider';
-import { VertexInit, VertexAI } from '@google-cloud/vertexai';
+import { Resolvable, FetchFunction } from '@ai-sdk/provider-utils';
+import { InternalGoogleGenerativeAISettings } from '@ai-sdk/google/internal';
+import { GoogleAuthOptions } from 'google-auth-library';
 
-type GoogleVertexModelId = 'gemini-1.5-flash' | 'gemini-1.5-flash-001' | 'gemini-1.5-flash-002' | 'gemini-1.5-pro' | 'gemini-1.5-pro-001' | 'gemini-1.5-pro-002' | 'gemini-1.0-pro' | 'gemini-1.0-pro-001' | 'gemini-1.0-pro-vision' | 'gemini-1.0-pro-vision-001' | (string & {});
-interface GoogleVertexSettings {
-    /**
-     * Optional. Enable structured output. Default is true.
-     *
-     * This is useful when the JSON Schema contains elements that are
-     * not supported by the OpenAPI schema version that
-     * Google Generative AI uses. You can use this to disable
-     * structured outputs if you need to.
-     */
-    structuredOutputs?: boolean;
-    /**
-  Optional. A list of unique safety settings for blocking unsafe content.
-     */
-    safetySettings?: Array<{
-        category: 'HARM_CATEGORY_UNSPECIFIED' | 'HARM_CATEGORY_HATE_SPEECH' | 'HARM_CATEGORY_DANGEROUS_CONTENT' | 'HARM_CATEGORY_HARASSMENT' | 'HARM_CATEGORY_SEXUALLY_EXPLICIT';
-        threshold: 'HARM_BLOCK_THRESHOLD_UNSPECIFIED' | 'BLOCK_LOW_AND_ABOVE' | 'BLOCK_MEDIUM_AND_ABOVE' | 'BLOCK_ONLY_HIGH' | 'BLOCK_NONE';
-    }>;
-    /**
-  Optional. When enabled, the model will use Google search to ground the response.
-  
-  @see https://cloud.google.com/vertex-ai/generative-ai/docs/grounding/overview
-     */
-    useSearchGrounding?: boolean;
+type GoogleVertexModelId = 'gemini-1.5-flash' | 'gemini-1.5-flash-001' | 'gemini-1.5-flash-002' | 'gemini-1.5-pro' | 'gemini-1.5-pro-001' | 'gemini-1.5-pro-002' | 'gemini-1.0-pro-001' | 'gemini-1.0-pro-vision-001' | 'gemini-1.0-pro' | 'gemini-1.0-pro-001' | 'gemini-1.0-pro-002' | (string & {});
+interface GoogleVertexSettings extends InternalGoogleGenerativeAISettings {
 }
 
 interface GoogleVertexProvider extends ProviderV1 {
@@ -34,7 +14,7 @@ interface GoogleVertexProvider extends ProviderV1 {
     (modelId: GoogleVertexModelId, settings?: GoogleVertexSettings): LanguageModelV1;
     languageModel: (modelId: GoogleVertexModelId, settings?: GoogleVertexSettings) => LanguageModelV1;
 }
-interface GoogleVertexProviderSettings {
+interface GoogleVertexProviderSettings$1 {
     /**
   Your Google Vertex location. Defaults to the environment variable `GOOGLE_VERTEX_LOCATION`.
      */
@@ -43,22 +23,32 @@ interface GoogleVertexProviderSettings {
   Your Google Vertex project. Defaults to the environment variable `GOOGLE_VERTEX_PROJECT`.
     */
     project?: string;
+    /**
+     * Headers to use for requests. Can be:
+     * - A headers object
+     * - A Promise that resolves to a headers object
+     * - A function that returns a headers object
+     * - A function that returns a Promise of a headers object
+     */
+    headers?: Resolvable<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;
+    generateId?: () => string;
+}
+
+interface GoogleVertexProviderSettings extends GoogleVertexProviderSettings$1 {
     /**
    Optional. The Authentication options provided by google-auth-library.
   Complete list of authentication options is documented in the
   GoogleAuthOptions interface:
   https://github.com/googleapis/google-auth-library-nodejs/blob/main/src/auth/googleauth.ts.
      */
-    googleAuthOptions?: VertexInit['googleAuthOptions'];
-    generateId?: () => string;
-    createVertexAI?: ({ project, location, }: {
-        project: string;
-        location: string;
-    }) => VertexAI;
+    googleAuthOptions?: GoogleAuthOptions;
 }
-/**
-Create a Google Vertex AI provider instance.
- */
+
 declare function createVertex(options?: GoogleVertexProviderSettings): GoogleVertexProvider;
 /**
 Default Google Vertex AI provider instance.