openclaw/extensions/litellm/onboard.ts

// Litellm setup module handles plugin onboarding behavior.
import {
  createDefaultModelPresetAppliers,
  type ModelDefinitionConfig,
  type OpenClawConfig,
} from "openclaw/plugin-sdk/provider-onboard";

export const LITELLM_BASE_URL = "http://localhost:4000";
export const LITELLM_DEFAULT_MODEL_ID = "claude-opus-4-6";
export const LITELLM_DEFAULT_MODEL_REF = `litellm/${LITELLM_DEFAULT_MODEL_ID}`;
const LITELLM_DEFAULT_CONTEXT_WINDOW = 1_000_000;
const LITELLM_DEFAULT_MAX_TOKENS = 128_000;
const LITELLM_DEFAULT_COST = {
  input: 0,
  output: 0,
  cacheRead: 0,
  cacheWrite: 0,
};

export function buildLitellmModelDefinition(): ModelDefinitionConfig {
  return {
    id: LITELLM_DEFAULT_MODEL_ID,
    name: "Claude Opus 4.6",
    reasoning: true,
    input: ["text", "image"],
    cost: LITELLM_DEFAULT_COST,
    contextWindow: LITELLM_DEFAULT_CONTEXT_WINDOW,
    maxTokens: LITELLM_DEFAULT_MAX_TOKENS,
  };
}

export const { applyConfig: applyLitellmConfig, applyProviderConfig: applyLitellmProviderConfig } =
  createDefaultModelPresetAppliers<[]>({
    primaryModelRef: LITELLM_DEFAULT_MODEL_REF,
    resolveParams: (cfg: OpenClawConfig) => {
      const existingProvider = cfg.models?.providers?.litellm as { baseUrl?: unknown } | undefined;
      const resolvedBaseUrl =
        typeof existingProvider?.baseUrl === "string" ? existingProvider.baseUrl.trim() : "";

      return {
        providerId: "litellm",
        api: "openai-completions" as const,
        baseUrl: resolvedBaseUrl || LITELLM_BASE_URL,
        defaultModel: buildLitellmModelDefinition(),
        defaultModelId: LITELLM_DEFAULT_MODEL_ID,
        aliases: [{ modelRef: LITELLM_DEFAULT_MODEL_REF, alias: "LiteLLM" }],
      };
    },
  });