index.ts

import { resolveOAuthApiKeyMarker } from "openclaw/plugin-sdk/provider-auth";
import { defineSingleProviderPluginEntry } from "openclaw/plugin-sdk/provider-entry";
import { buildProviderReplayFamilyHooks } from "openclaw/plugin-sdk/provider-model-shared";
import { resolveRuntimeConfig } from "./src/config.js";
import { PROVIDER_ID } from "./src/constants.js";
import {
  buildFallbackModel,
  buildProviderConfig,
  normalizeZenMuxModelId,
  refreshModels,
} from "./src/model-catalog.js";
import { loginZenMux, refreshZenMuxOAuthCredential } from "./src/oauth.js";

const replayHooks = buildProviderReplayFamilyHooks({
  family: "hybrid-anthropic-openai",
  anthropicModelDropThinkingBlocks: true,
});
export default defineSingleProviderPluginEntry({
  id: PROVIDER_ID,
  name: "ZenMux Provider",
  description: "ZenMux OAuth provider with dynamic multi-protocol model discovery",
  provider: {
    label: "ZenMux",
    docsPath: "https://zenmux.ai/docs",
    envVars: [
      "ZENMUX_OAUTH_CLIENT_ID",
      "ZENMUX_OAUTH_ORIGIN",
      "ZENMUX_API_BASE_URL",
      "ZENMUX_ANTHROPIC_BASE_URL",
      "ZENMUX_MODEL_CATALOG_URL",
    ],
    extraAuth: [
      {
        id: "oauth",
        label: "ZenMux OAuth",
        hint: "Browser sign-in with PKCE",
        kind: "oauth",
        wizard: {
          choiceId: "zenmux-oauth",
          choiceLabel: "ZenMux OAuth",
          choiceHint: "Browser sign-in with PKCE",
          groupId: PROVIDER_ID,
          groupLabel: "ZenMux",
          groupHint: "OAuth access to ZenMux models",
          methodId: "oauth",
        },
        run: loginZenMux,
      },
    ],
    catalog: {
      order: "profile",
      run: async (ctx) => {
        const config = resolveRuntimeConfig(ctx.env);
        const { apiKey } = ctx.resolveProviderAuth(PROVIDER_ID, {
          oauthMarker: resolveOAuthApiKeyMarker(PROVIDER_ID),
        });
        if (!apiKey) return null;
        const models = await refreshModels({ config });
        return { provider: buildProviderConfig({ config, models, apiKey }) };
      },
      staticRun: async (ctx) => {
        const config = resolveRuntimeConfig(ctx.env);
        return {
          provider: buildProviderConfig({ config, models: [buildFallbackModel(config)] }),
        };
      },
    },
    normalizeModelId: ({ modelId }) => normalizeZenMuxModelId(modelId),
    refreshOAuth: refreshZenMuxOAuthCredential,
    ...replayHooks,
  },
});