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
568 B
TypeScript
16 lines
568 B
TypeScript
import { openAICompletionsApi } from "../api/openai-completions.lazy.ts";
|
|
import { envApiKeyAuth } from "../auth/helpers.ts";
|
|
import { createProvider, type Provider } from "../models.ts";
|
|
import { NVIDIA_MODELS } from "./nvidia.models.ts";
|
|
|
|
export function nvidiaProvider(): Provider<"openai-completions"> {
|
|
return createProvider({
|
|
id: "nvidia",
|
|
name: "NVIDIA",
|
|
baseUrl: "https://integrate.api.nvidia.com/v1",
|
|
auth: { apiKey: envApiKeyAuth("NVIDIA API key", ["NVIDIA_API_KEY"]) },
|
|
models: Object.values(NVIDIA_MODELS),
|
|
api: openAICompletionsApi(),
|
|
});
|
|
}
|