openclaw/extensions/arcee/provider-catalog.ts

/**
 * Arcee provider catalog builders. Direct Arcee uses native ids; OpenRouter
 * catalogs expose only currently served models under OpenRouter's `arcee-ai/*`
 * namespace.
 */
import type { ModelProviderConfig } from "openclaw/plugin-sdk/provider-model-shared";
import { buildArceeModelDefinition, ARCEE_BASE_URL, ARCEE_MODEL_CATALOG } from "./models.js";

/** Canonical OpenRouter API base URL for Arcee-routed models. */
export const OPENROUTER_BASE_URL = "https://openrouter.ai/api/v1";
const OPENROUTER_LEGACY_BASE_URL = "https://openrouter.ai/v1";
const ARCEE_OPENROUTER_MODEL_IDS = new Set(["trinity-large-preview", "trinity-large-thinking"]);

function normalizeBaseUrl(baseUrl: string | undefined): string {
  return (baseUrl ?? "").trim().replace(/\/+$/, "");
}

/** Normalize OpenRouter base URLs accepted for Arcee model routing. */
export function normalizeArceeOpenRouterBaseUrl(baseUrl: string | undefined): string | undefined {
  const normalized = normalizeBaseUrl(baseUrl);
  if (!normalized) {
    return undefined;
  }
  if (normalized === OPENROUTER_BASE_URL || normalized === OPENROUTER_LEGACY_BASE_URL) {
    return OPENROUTER_BASE_URL;
  }
  return undefined;
}

/** Convert a bare or legacy Arcee model id to OpenRouter's `arcee-ai/*` id. */
export function toArceeOpenRouterModelId(modelId: string): string {
  const normalized = modelId.trim();
  if (!normalized || normalized.startsWith("arcee-ai/")) {
    return normalized;
  }
  const bareId = normalized.startsWith("arcee/") ? normalized.slice("arcee/".length) : normalized;
  return `arcee-ai/${bareId}`;
}

/** Build direct Arcee catalog models. */
export function buildArceeCatalogModels(): NonNullable<ModelProviderConfig["models"]> {
  return ARCEE_MODEL_CATALOG.map(buildArceeModelDefinition);
}

/** Build OpenRouter-routed Arcee catalog models. */
export function buildArceeOpenRouterCatalogModels(): NonNullable<ModelProviderConfig["models"]> {
  return buildArceeCatalogModels()
    .filter((model) => ARCEE_OPENROUTER_MODEL_IDS.has(model.id))
    .map((model) => Object.assign({}, model, { id: toArceeOpenRouterModelId(model.id) }));
}

/** Build the direct Arcee provider config. */
export function buildArceeProvider(): ModelProviderConfig {
  return {
    baseUrl: ARCEE_BASE_URL,
    api: "openai-completions",
    models: buildArceeCatalogModels(),
  };
}

/** Build the OpenRouter-backed Arcee provider config. */
export function buildArceeOpenRouterProvider(): ModelProviderConfig {
  return {
    baseUrl: OPENROUTER_BASE_URL,
    api: "openai-completions",
    models: buildArceeOpenRouterCatalogModels(),
  };
}