Cloudflare Worker Google Vertex AI

vertex.ts

import GoogleOAuth, { GoogleKey } from 'cloudflare-workers-and-google-oauth'
import { GoogleAuth } from "google-auth-library";
import { GenerateContentRequest, GenerativeModel } from "@google-cloud/vertexai";
import authJSON from "./auth.json"; // Service account

class JsonAuth extends GoogleAuthBase {
  private oauth: GoogleOAuth
  constructor(private readonly json: GoogleKey) {
    super();
    this.oauth = new GoogleOAuth(json, ["https://www.googleapis.com/auth/cloud-platform"])
  }

  override getAccessToken(): Promise<string | null | undefined> {
    return this.oauth.getGoogleAuthToken()
  }
}
    
const model = new GenerativeModel({
  model: "gemini-1.5-flash",
  googleAuth: new JsonAuth(authJSON),
  project: authJSON.project_id,
  location: 'us-central1',
});

const request: GenerateContentRequest = {
  contents: [{role: 'user', parts: [{text: 'Hello world!'}]}],
};

const result = await model.generateContent(request);