Pi Agent Provider for Berget.ai

pi-agent-berget-provider.ts

import type { ExtensionAPI } from "@mariozechner/pi-coding-agent";



export default function (pi: ExtensionAPI) {
 pi.registerProvider("berget", {
  baseUrl: "https://api.berget.ai/v1",
  apiKey: "BERGET_API_KEY",
  api: "openai-completions",
  models: [
   {
    id: "openai/gpt-oss-120b",
    name: "GPT-OSS 120B (Berget)",
    reasoning: false,
    input: ["text", "image"],
    cost: { input: 0.40, output: 0.90, cacheRead: 0, cacheWrite: 0 },
    contextWindow: 128000,
    maxTokens: 8192,
    compat: { supportsDeveloperRole: false, maxTokensField: "max_tokens" }
   },

   {
    id: "mistralai/Mistral-Small-3.2-24B-Instruct-2506",
    name: "Mistral Small 3.2 24B (Berget)",
    reasoning: false,
    input: ["text", "image"],
    cost: { input: 0.30, output: 0.30, cacheRead: 0, cacheWrite: 0 },
    contextWindow: 128000,
    maxTokens: 8192,
    compat: { supportsDeveloperRole: false, maxTokensField: "max_tokens" }
   },

   {
    id: "zai-org/GLM-4.7-FP8",
    name: "GLM 4.7 (Berget)",
    reasoning: true,
    input: ["text"],
    cost: { input: 0.70, output: 2.50, cacheRead: 0, cacheWrite: 0 },
    contextWindow: 128000,
    maxTokens: 8192,
    compat: { supportsDeveloperRole: false, maxTokensField: "max_tokens" }
   },

   {
    id: "google/gemma-4-31B-it",
    name: "Gemma 4 31B (Berget)",
    reasoning: true,
    input: ["text"],
    cost: { input: 0.25, output: 0.50, cacheRead: 0, cacheWrite: 0 },
    contextWindow: 262144,
    maxTokens: 8192,
    compat: { supportsDeveloperRole: false, maxTokensField: "max_tokens" }
   },

   {
    id: "meta-llama/Llama-3.3-70B-Instruct",
    name: "Llama 3.3 70B (Berget)",
    reasoning: false,
    input: ["text"],
    cost: { input: 0.90, output: 0.90, cacheRead: 0, cacheWrite: 0 },
    contextWindow: 128000,
    maxTokens: 8192,
    compat: { supportsDeveloperRole: false, maxTokensField: "max_tokens" }
   },

   {
    id: "meta-llama/Llama-3.1-8B-Instruct",
    name: "Llama 3.1 8B (Berget)",
    reasoning: false,
    input: ["text"],
    cost: { input: 0.20, output: 0.20, cacheRead: 0, cacheWrite: 0 },
    contextWindow: 128000,
    maxTokens: 8192,
    compat: { supportsDeveloperRole: false, maxTokensField: "max_tokens" }
   }

  ]

 });

}