@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.
42 lines (37 loc) • 1.28 kB
text/typescript
import { resolve } from '@ai-sdk/provider-utils';
import {
generateAuthToken,
GoogleCredentials,
} from '../../edge/google-vertex-auth-edge';
import {
createVertexAnthropic as createVertexAnthropicOriginal,
GoogleVertexAnthropicProvider,
GoogleVertexAnthropicProviderSettings as GoogleVertexAnthropicProviderSettingsOriginal,
} from '../google-vertex-anthropic-provider';
export type { GoogleVertexAnthropicProvider };
export interface GoogleVertexAnthropicProviderSettings
extends GoogleVertexAnthropicProviderSettingsOriginal {
/**
* Optional. The Google credentials for the Google Cloud service account. If
* not provided, the Google Vertex provider will use environment variables to
* load the credentials.
*/
googleCredentials?: GoogleCredentials;
}
export function createVertexAnthropic(
options: GoogleVertexAnthropicProviderSettings = {},
): GoogleVertexAnthropicProvider {
return createVertexAnthropicOriginal({
...options,
headers: async () => ({
Authorization: `Bearer ${await generateAuthToken(
options.googleCredentials,
)}`,
...(await resolve(options.headers)),
}),
});
}
/**
* Default Google Vertex AI Anthropic provider instance.
*/
export const vertexAnthropic = createVertexAnthropic();