UNPKG

@ai-sdk/google-vertex

Version:

The **[Google Vertex provider](https://ai-sdk.dev/providers/ai-sdk-providers/google-vertex)** for the [AI SDK](https://ai-sdk.dev/docs) contains language model support for the [Google Vertex AI](https://cloud.google.com/vertex-ai) APIs.

120 lines (115 loc) 4.57 kB
"use strict"; var __defProp = Object.defineProperty; var __getOwnPropDesc = Object.getOwnPropertyDescriptor; var __getOwnPropNames = Object.getOwnPropertyNames; var __hasOwnProp = Object.prototype.hasOwnProperty; var __export = (target, all) => { for (var name in all) __defProp(target, name, { get: all[name], enumerable: true }); }; var __copyProps = (to, from, except, desc) => { if (from && typeof from === "object" || typeof from === "function") { for (let key of __getOwnPropNames(from)) if (!__hasOwnProp.call(to, key) && key !== except) __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable }); } return to; }; var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod); // src/maas/index.ts var index_exports = {}; __export(index_exports, { createVertexMaas: () => createVertexMaas2, vertexMaas: () => vertexMaas }); module.exports = __toCommonJS(index_exports); // src/maas/google-vertex-maas-provider-node.ts var import_provider_utils2 = require("@ai-sdk/provider-utils"); // src/google-vertex-auth-google-auth-library.ts var import_google_auth_library = require("google-auth-library"); function createAuthTokenGenerator(options) { const auth = new import_google_auth_library.GoogleAuth({ scopes: ["https://www.googleapis.com/auth/cloud-platform"], ...options }); return async function generateAuthToken() { var _a; const client = await auth.getClient(); const token = await client.getAccessToken(); return (_a = token == null ? void 0 : token.token) != null ? _a : null; }; } // src/maas/google-vertex-maas-provider.ts var import_openai_compatible = require("@ai-sdk/openai-compatible"); var import_provider_utils = require("@ai-sdk/provider-utils"); function createVertexMaas(options = {}) { const loadLocation = () => (0, import_provider_utils.loadOptionalSetting)({ settingValue: options.location, environmentVariableName: "GOOGLE_VERTEX_LOCATION" }); const loadProject = () => (0, import_provider_utils.loadSetting)({ settingValue: options.project, settingName: "project", environmentVariableName: "GOOGLE_VERTEX_PROJECT", description: "Google Vertex project" }); const constructBaseURL = () => { var _a; const projectId = loadProject(); const location = (_a = loadLocation()) != null ? _a : "global"; return `https://aiplatform.googleapis.com/v1/projects/${projectId}/locations/${location}/endpoints/openapi`; }; const loadBaseURL = () => { var _a; return (0, import_provider_utils.withoutTrailingSlash)((_a = options.baseURL) != null ? _a : "") || constructBaseURL(); }; let cachedProvider; const getProvider = () => cachedProvider != null ? cachedProvider : cachedProvider = (0, import_openai_compatible.createOpenAICompatible)({ name: "vertex.maas", baseURL: loadBaseURL(), fetch: options.fetch }); const provider = (modelId) => getProvider()(modelId); provider.specificationVersion = "v3"; provider.languageModel = (modelId) => getProvider().languageModel(modelId); provider.chatModel = (modelId) => getProvider().chatModel(modelId); provider.completionModel = (modelId) => getProvider().completionModel(modelId); provider.embeddingModel = (modelId) => getProvider().embeddingModel(modelId); provider.textEmbeddingModel = (modelId) => getProvider().textEmbeddingModel(modelId); provider.imageModel = (modelId) => getProvider().imageModel(modelId); return provider; } // src/maas/google-vertex-maas-provider-node.ts function createVertexMaas2(options = {}) { const generateAuthToken = createAuthTokenGenerator(options.googleAuthOptions); const customFetch = async (url, init) => { var _a; const token = await generateAuthToken(); const resolvedHeaders = await (0, import_provider_utils2.resolve)(options.headers); const authHeaders = { ...resolvedHeaders, Authorization: `Bearer ${token}` }; const fetchInit = { ...init, headers: { ...init == null ? void 0 : init.headers, ...authHeaders } }; return ((_a = options.fetch) != null ? _a : fetch)(url, fetchInit); }; return createVertexMaas({ ...options, fetch: customFetch, headers: void 0 // Don't pass headers, we handle them in fetch }); } var vertexMaas = createVertexMaas2(); // Annotate the CommonJS export names for ESM import in node: 0 && (module.exports = { createVertexMaas, vertexMaas }); //# sourceMappingURL=index.js.map