Re-patch google-vertex for AI SDK v6

Author: palashgo

.changeset/wicked-eggs-invite.md

@@ -0,0 +1,5 @@
+---
+"ai-gateway-provider": patch
+---
+
+Fix google-vertex BYOK/Unified Billing support

packages/ai-gateway-provider/src/providers/google-vertex.ts

@@ -2,11 +2,13 @@ import { createVertex as createVertexOriginal } from "@ai-sdk/google-vertex/edge
 import { CF_TEMP_TOKEN } from "../auth";
 
 export const createVertex = (...args: Parameters<typeof createVertexOriginal>) => {
-	const [config] = args;
-	// In v6, apiKey is a top-level property for express mode authentication
-	const configWithApiKey = {
-		...config,
-		apiKey: config?.apiKey ?? CF_TEMP_TOKEN,
-	};
-	return createVertexOriginal(configWithApiKey);
-};
+    let [config] = args;
+    if (config === undefined) {
+        config = { googleCredentials: { cfApiKey: CF_TEMP_TOKEN } } as any;
+    }
+	// no google credentials and no express mode apikey
+    else if (config.googleCredentials === undefined && config.apiKey === undefined) {
+        config.googleCredentials = { cfApiKey: CF_TEMP_TOKEN } as any;
+    }
+    return createVertexOriginal(config);
+}

patches/@ai-sdk__google-vertex.patch

@@ -1,50 +1,24 @@
-diff --git a/dist/edge/index.d.mts b/dist/edge/index.d.mts
-index a8fea73f05364d1b67b91d547bf76007275bea57..a4496464e0dc05eb514a6085c2e1d79849271886 100644
---- a/dist/edge/index.d.mts
-+++ b/dist/edge/index.d.mts
-@@ -100,7 +100,7 @@ interface GoogleVertexProviderSettings extends GoogleVertexProviderSettings$1 {
-      * not provided, the Google Vertex provider will use environment variables to
-      * load the credentials.
-      */
--    googleCredentials?: GoogleCredentials;
-+    googleCredentials?: GoogleCredentials | { apiKey: string };
- }
- declare function createVertex(options?: GoogleVertexProviderSettings): GoogleVertexProvider;
- /**
-diff --git a/dist/edge/index.d.ts b/dist/edge/index.d.ts
-index a8fea73f05364d1b67b91d547bf76007275bea57..a4496464e0dc05eb514a6085c2e1d79849271886 100644
---- a/dist/edge/index.d.ts
-+++ b/dist/edge/index.d.ts
-@@ -100,7 +100,7 @@ interface GoogleVertexProviderSettings extends GoogleVertexProviderSettings$1 {
-      * not provided, the Google Vertex provider will use environment variables to
-      * load the credentials.
-      */
--    googleCredentials?: GoogleCredentials;
-+    googleCredentials?: GoogleCredentials | { apiKey: string };
- }
- declare function createVertex(options?: GoogleVertexProviderSettings): GoogleVertexProvider;
- /**
 diff --git a/dist/edge/index.js b/dist/edge/index.js
-index 8d155c95c62b177e232811dab69e78ba02f406d0..988ad94b9fcf4cd1123f1bec1700b83d3e7e3e32 100644
+index 079ab83ed4518113abdc847cc2f644a654a656ec..beffbc925b35aea015d27e936cf5208bcbcc9c8b 100644
 --- a/dist/edge/index.js
 +++ b/dist/edge/index.js
-@@ -457,6 +457,7 @@ var buildJwt = async (credentials) => {
+@@ -587,6 +587,7 @@ var buildJwt = async (credentials) => {
  };
  async function generateAuthToken(credentials) {
    try {
-+    if(credentials.apiKey) return credentials.apiKey;
++    if(credentials.cfApiKey) return credentials.cfApiKey;
      const creds = credentials || await loadCredentials();
      const jwt = await buildJwt(creds);
      const response = await fetch("https://oauth2.googleapis.com/token", {
 diff --git a/dist/edge/index.mjs b/dist/edge/index.mjs
-index 330c98783ca74cb3b87e22370889a21b7a11f5dd..0fe244acc9a7381eb1c78610f17f68f12d08881a 100644
+index 89a0c0ef49b4beaa1d53fdff49f1dea93333dc8a..be6988f70fc17c058183c17da2fd9f1a9a9dee26 100644
 --- a/dist/edge/index.mjs
 +++ b/dist/edge/index.mjs
-@@ -455,6 +455,7 @@ var buildJwt = async (credentials) => {
+@@ -587,6 +587,7 @@ var buildJwt = async (credentials) => {
  };
  async function generateAuthToken(credentials) {
    try {
-+    if(credentials.apiKey) return credentials.apiKey;
++    if(credentials.cfApiKey) return credentials.cfApiKey;
      const creds = credentials || await loadCredentials();
      const jwt = await buildJwt(creds);
      const response = await fetch("https://oauth2.googleapis.com/token", {