alexion
67b361f0e9
git-subtree-dir: third_party/pi git-subtree-split: d981de1229ef899957bbe968bc8dcda02a21f477
16 lines
572 B
TypeScript
16 lines
572 B
TypeScript
import { openAICompletionsApi } from "../api/openai-completions.lazy.ts";
|
|
import { envApiKeyAuth } from "../auth/helpers.ts";
|
|
import { createProvider, type Provider } from "../models.ts";
|
|
import { BASETEN_MODELS } from "./baseten.models.ts";
|
|
|
|
export function basetenProvider(): Provider<"openai-completions"> {
|
|
return createProvider({
|
|
id: "baseten",
|
|
name: "Baseten",
|
|
baseUrl: "https://inference.baseten.co/v1",
|
|
auth: { apiKey: envApiKeyAuth("Baseten API key", ["BASETEN_API_KEY"]) },
|
|
models: Object.values(BASETEN_MODELS),
|
|
api: openAICompletionsApi(),
|
|
});
|
|
}
|