alexion
4def357d09
Imported as a linear vendor commit so forge rebases do not replay the upstream root tree. git-subtree-dir: third_party/pi git-subtree-split: d981de1229ef899957bbe968bc8dcda02a21f477
16 lines
575 B
TypeScript
16 lines
575 B
TypeScript
import { openAICompletionsApi } from "../api/openai-completions.lazy.ts";
|
|
import { envApiKeyAuth } from "../auth/helpers.ts";
|
|
import { createProvider, type Provider } from "../models.ts";
|
|
import { CEREBRAS_MODELS } from "./cerebras.models.ts";
|
|
|
|
export function cerebrasProvider(): Provider<"openai-completions"> {
|
|
return createProvider({
|
|
id: "cerebras",
|
|
name: "Cerebras",
|
|
baseUrl: "https://api.cerebras.ai/v1",
|
|
auth: { apiKey: envApiKeyAuth("Cerebras API key", ["CEREBRAS_API_KEY"]) },
|
|
models: Object.values(CEREBRAS_MODELS),
|
|
api: openAICompletionsApi(),
|
|
});
|
|
}
|