| import type { RouteDefaultsInput } from "../route/client" |
| import { Auth } from "../route/auth" |
| import { ProviderID, type ModelID } from "../schema" |
| import * as BedrockConverse from "../protocols/bedrock-converse" |
| import type { BedrockCredentials } from "../protocols/bedrock-converse" |
|
|
| export const id = ProviderID.make("amazon-bedrock") |
|
|
| export type Config = RouteDefaultsInput & { |
| readonly apiKey?: string |
| readonly headers?: Record<string, string> |
| readonly credentials?: BedrockCredentials |
| |
| readonly region?: string |
| |
| readonly baseURL?: string |
| } |
| export const routes = [BedrockConverse.route] |
|
|
| const bedrockBaseURL = (region: string) => `https://bedrock-runtime.${region}.amazonaws.com` |
|
|
| const configuredRoute = (input: Config) => { |
| const { apiKey, credentials, region, baseURL, ...rest } = input |
| const resolvedRegion = region ?? credentials?.region ?? "us-east-1" |
| return BedrockConverse.route.with({ |
| ...rest, |
| provider: id, |
| endpoint: { baseURL: baseURL ?? bedrockBaseURL(resolvedRegion) }, |
| auth: apiKey === undefined ? BedrockConverse.sigV4Auth(credentials) : Auth.bearer(apiKey), |
| }) |
| } |
|
|
| export const configure = (input: Config = {}) => { |
| const route = configuredRoute(input) |
| return { |
| id, |
| model: (modelID: string | ModelID) => route.model({ id: modelID }), |
| configure, |
| } |
| } |
|
|
| export const provider = configure() |
| export const model = provider.model |
|
|
