Squashed 'third_party/pi/' content from commit d981de122

git-subtree-dir: third_party/pi
git-subtree-split: d981de1229ef899957bbe968bc8dcda02a21f477

packages/ai/src/providers/baseten.ts

@@ -0,0 +1,15 @@
+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(),
+	});
+}