Skip to content

Re-patch google-vertex for AI SDK v6 #353

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
palashgo merged 1 commit into from
Jan 20, 2026
Merged

Re-patch google-vertex for AI SDK v6 #353

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
5 changes: 5 additions & 0 deletions .changeset/wicked-eggs-invite.md
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
---
"ai-gateway-provider": patch
---

Fix google-vertex BYOK/Unified Billing support
18 changes: 10 additions & 8 deletions packages/ai-gateway-provider/src/providers/google-vertex.ts
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -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);
}
38 changes: 6 additions & 32 deletions patches/@ai-sdk__google-vertex.patch
View file
Open in desktop
Original file line number Diff line number Diff line change
@@ -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", {
6 changes: 3 additions & 3 deletions pnpm-lock.yaml
View file
Open in desktop

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

Loading