@ai-sdk/google-vertex 5.0.98 → 5.0.100

package/src/google-vertex-provider-base.ts

@@ -6,16 +6,18 @@ import {
   type GoogleInteractionsModelId,
   type GoogleInteractionsModelInput,
 } from '@ai-sdk/google/internal';
-import type {
-  Experimental_VideoModelV4,
-  ImageModelV4,
-  LanguageModelV4,
-  ProviderV4,
-  SpeechModelV4,
-  TranscriptionModelV4,
+import {
+  InvalidArgumentError,
+  type Experimental_VideoModelV4,
+  type ImageModelV4,
+  type LanguageModelV4,
+  type ProviderV4,
+  type SpeechModelV4,
+  type TranscriptionModelV4,
 } from '@ai-sdk/provider';
 import {
   generateId,
+  isValidHostnamePart,
   loadOptionalSetting,
   loadSetting,
   normalizeHeaders,
@@ -225,13 +227,22 @@ export function createGoogleVertex(
       description: 'Google Vertex project',
     });
 
-  const loadGoogleVertexLocation = () =>
-    loadSetting({
+  const loadGoogleVertexLocation = () => {
+    const location = loadSetting({
       settingValue: options.location,
       settingName: 'location',
       environmentVariableName: 'GOOGLE_VERTEX_LOCATION',
       description: 'Google Vertex location',
     });
+    if (!isValidHostnamePart(location)) {
+      throw new InvalidArgumentError({
+        argument: 'location',
+        message:
+          'Invalid Google Vertex location. Expected a single DNS label (letters, digits, and hyphens). Use `baseURL` for custom endpoints.',
+      });
+    }
+    return location;
+  };
 
   // Tuned models are addressed via their deployed endpoint
   // `.../locations/{region}/endpoints/{id}` instead of the base-model
@@ -242,6 +253,11 @@ export function createGoogleVertex(
       return withoutTrailingSlash(options.baseURL) ?? EXPRESS_MODE_BASE_URL;
     }
 
+    const baseURL = withoutTrailingSlash(options.baseURL);
+    if (baseURL != null) {
+      return baseURL;
+    }
+
     const region = loadGoogleVertexLocation();
     const project = loadGoogleVertexProject();
 
@@ -255,12 +271,9 @@ export function createGoogleVertex(
       }
     };
 
-    return (
-      withoutTrailingSlash(options.baseURL) ??
-      `https://${getHost()}/v1beta1/projects/${project}/locations/${region}${
-        endpoint ? '' : '/publishers/google'
-      }`
-    );
+    return `https://${getHost()}/v1beta1/projects/${project}/locations/${region}${
+      endpoint ? '' : '/publishers/google'
+    }`;
   };
 
   const createConfig = (

package/src/google-vertex-transcription-model-options.ts

@@ -32,7 +32,8 @@ export const googleVertexTranscriptionProviderOptionsSchema = z.object({
 
   /**
    * The Cloud Speech-to-Text region for the request (e.g. `'us'`, `'eu'`,
-   * `'us-central1'`). Defaults to the provider `location`.
+   * `'us-central1'`). Defaults to the provider `location`. Must be a single
+   * DNS label (letters, digits, and hyphens).
    *
    * Note: Speech-to-Text regions differ from Vertex AI regions. Chirp is only
    * available in specific Speech-to-Text regions and is not available in the

package/src/google-vertex-transcription-model.ts

@@ -1,8 +1,13 @@
-import type { SharedV4Warning, TranscriptionModelV4 } from '@ai-sdk/provider';
+import {
+  InvalidArgumentError,
+  type SharedV4Warning,
+  type TranscriptionModelV4,
+} from '@ai-sdk/provider';
 import {
   combineHeaders,
   convertUint8ArrayToBase64,
   createJsonResponseHandler,
+  isValidHostnamePart,
   parseProviderOptions,
   postJsonToApi,
   resolve,
@@ -106,6 +111,13 @@ export class GoogleVertexTranscriptionModel implements TranscriptionModelV4 {
     }
 
     const region = googleOptions?.region ?? this.config.location;
+    if (!isValidHostnamePart(region)) {
+      throw new InvalidArgumentError({
+        argument: googleOptions?.region != null ? 'region' : 'location',
+        message:
+          'Invalid Google Cloud Speech-to-Text region. Expected a single DNS label (letters, digits, and hyphens).',
+      });
+    }
     const languageCodes = googleOptions?.languageCodes ?? ['auto'];
 
     // The recognize API takes base64-encoded audio in the `content` field. A

package/src/maas/google-vertex-maas-provider.ts

@@ -1,8 +1,10 @@
+import { InvalidArgumentError } from '@ai-sdk/provider';
 import {
   createOpenAICompatible,
   type OpenAICompatibleProvider,
 } from '@ai-sdk/openai-compatible';
 import {
+  isValidHostnamePart,
   loadOptionalSetting,
   loadSetting,
   withoutTrailingSlash,
@@ -74,11 +76,21 @@ export function createGoogleVertexMaas(
   options: GoogleVertexMaasProviderSettings = {},
 ): GoogleVertexMaasProvider {
   // Lazy-load settings to support loading from environment variables at runtime
-  const loadLocation = () =>
-    loadOptionalSetting({
-      settingValue: options.location,
-      environmentVariableName: 'GOOGLE_VERTEX_LOCATION',
-    });
+  const loadLocation = () => {
+    const location =
+      loadOptionalSetting({
+        settingValue: options.location,
+        environmentVariableName: 'GOOGLE_VERTEX_LOCATION',
+      }) ?? 'global';
+    if (!isValidHostnamePart(location)) {
+      throw new InvalidArgumentError({
+        argument: 'location',
+        message:
+          'Invalid Google Vertex location. Expected a single DNS label (letters, digits, and hyphens). Use `baseURL` for custom endpoints.',
+      });
+    }
+    return location;
+  };
 
   const loadProject = () =>
     loadSetting({
@@ -100,7 +112,7 @@ export function createGoogleVertexMaas(
 
   const constructBaseURL = () => {
     const projectId = loadProject();
-    const location = loadLocation() ?? 'global';
+    const location = loadLocation();
 
     return `https://${getHost(location)}/v1/projects/${projectId}/locations/${location}/endpoints/openapi`;
   };