elizaos--eliza/plugins/plugin-openrouter/utils/config.ts

/**
 * Setting resolution for the plugin. `getSetting` reads character settings first
 * (`runtime.getSetting`), then `process.env`, then a caller default. The
 * `get*Model` helpers layer the priority the README documents: `OPENROUTER_*`
 * variant, then the generic (`SMALL_MODEL`, …) fallback, then a hard default —
 * with nano/medium/mega/response-handler/action-planner cascading onto their
 * base tier when unset. Also holds the default model constants and the
 * base-URL/dimension/cleanup accessors.
 */
import type { IAgentRuntime } from "@elizaos/core";

export const DEFAULT_BASE_URL = "https://openrouter.ai/api/v1";
export const DEFAULT_SMALL_MODEL = "google/gemini-2.5-flash-lite";
export const DEFAULT_LARGE_MODEL = "google/gemini-2.5-flash";
export const DEFAULT_IMAGE_MODEL = "x-ai/grok-2-vision-1212";
export const DEFAULT_IMAGE_GENERATION_MODEL = "google/gemini-2.5-flash-image-preview";
export const DEFAULT_EMBEDDING_MODEL = "openai/text-embedding-3-small";
export const DEFAULT_TRANSCRIPTION_MODEL = "openai/whisper-large-v3";
export const DEFAULT_EMBEDDING_DIMENSIONS = 1536;

function getEnvValue(key: string): string | undefined {
  if (typeof process === "undefined" || !process.env) {
    return undefined;
  }
  const value = process.env[key];
  return value === undefined ? undefined : String(value);
}

export function getSetting(
  runtime: IAgentRuntime,
  key: string,
  defaultValue?: string
): string | undefined {
  const value = runtime.getSetting(key);
  if (value !== undefined && value !== null) {
    return String(value);
  }
  return getEnvValue(key) ?? defaultValue;
}

export function getBaseURL(runtime: IAgentRuntime): string {
  const browserURL = getSetting(runtime, "OPENROUTER_BROWSER_BASE_URL");
  if (
    typeof globalThis !== "undefined" &&
    (globalThis as Record<string, unknown>).document &&
    browserURL
  ) {
    return browserURL;
  }
  return getSetting(runtime, "OPENROUTER_BASE_URL", DEFAULT_BASE_URL) || DEFAULT_BASE_URL;
}

export function getApiKey(runtime: IAgentRuntime): string | undefined {
  return getSetting(runtime, "OPENROUTER_API_KEY");
}

export function getSmallModel(runtime: IAgentRuntime): string {
  return (
    getSetting(runtime, "OPENROUTER_SMALL_MODEL") ??
    getSetting(runtime, "SMALL_MODEL", DEFAULT_SMALL_MODEL) ??
    DEFAULT_SMALL_MODEL
  );
}

export function getNanoModel(runtime: IAgentRuntime): string {
  return (
    getSetting(runtime, "OPENROUTER_NANO_MODEL") ??
    getSetting(runtime, "NANO_MODEL") ??
    getSmallModel(runtime)
  );
}

export function getMediumModel(runtime: IAgentRuntime): string {
  return (
    getSetting(runtime, "OPENROUTER_MEDIUM_MODEL") ??
    getSetting(runtime, "MEDIUM_MODEL") ??
    getSmallModel(runtime)
  );
}

export function getLargeModel(runtime: IAgentRuntime): string {
  return (
    getSetting(runtime, "OPENROUTER_LARGE_MODEL") ??
    getSetting(runtime, "LARGE_MODEL", DEFAULT_LARGE_MODEL) ??
    DEFAULT_LARGE_MODEL
  );
}

export function getMegaModel(runtime: IAgentRuntime): string {
  return (
    getSetting(runtime, "OPENROUTER_MEGA_MODEL") ??
    getSetting(runtime, "MEGA_MODEL") ??
    getLargeModel(runtime)
  );
}

export function getResponseHandlerModel(runtime: IAgentRuntime): string {
  return (
    getSetting(runtime, "OPENROUTER_RESPONSE_HANDLER_MODEL") ??
    getSetting(runtime, "OPENROUTER_SHOULD_RESPOND_MODEL") ??
    getSetting(runtime, "RESPONSE_HANDLER_MODEL") ??
    getSetting(runtime, "SHOULD_RESPOND_MODEL") ??
    getNanoModel(runtime)
  );
}

export function getActionPlannerModel(runtime: IAgentRuntime): string {
  return (
    getSetting(runtime, "OPENROUTER_ACTION_PLANNER_MODEL") ??
    getSetting(runtime, "OPENROUTER_PLANNER_MODEL") ??
    getSetting(runtime, "ACTION_PLANNER_MODEL") ??
    getSetting(runtime, "PLANNER_MODEL") ??
    getMediumModel(runtime)
  );
}

export function getImageModel(runtime: IAgentRuntime): string {
  return (
    getSetting(runtime, "OPENROUTER_IMAGE_MODEL") ??
    getSetting(runtime, "IMAGE_MODEL", DEFAULT_IMAGE_MODEL) ??
    DEFAULT_IMAGE_MODEL
  );
}

export function getImageGenerationModel(runtime: IAgentRuntime): string {
  return (
    getSetting(runtime, "OPENROUTER_IMAGE_GENERATION_MODEL") ??
    getSetting(runtime, "IMAGE_GENERATION_MODEL", DEFAULT_IMAGE_GENERATION_MODEL) ??
    DEFAULT_IMAGE_GENERATION_MODEL
  );
}

export function getEmbeddingModel(runtime: IAgentRuntime): string {
  return (
    getSetting(runtime, "OPENROUTER_EMBEDDING_MODEL") ??
    getSetting(runtime, "EMBEDDING_MODEL", DEFAULT_EMBEDDING_MODEL) ??
    DEFAULT_EMBEDDING_MODEL
  );
}

export function getTranscriptionModel(runtime: IAgentRuntime): string {
  return (
    getSetting(runtime, "OPENROUTER_TRANSCRIPTION_MODEL") ??
    getSetting(runtime, "TRANSCRIPTION_MODEL", DEFAULT_TRANSCRIPTION_MODEL) ??
    DEFAULT_TRANSCRIPTION_MODEL
  );
}

export function getEmbeddingDimensions(runtime: IAgentRuntime): number {
  const setting =
    getSetting(runtime, "OPENROUTER_EMBEDDING_DIMENSIONS") ??
    getSetting(runtime, "EMBEDDING_DIMENSIONS");
  return setting ? parseInt(setting, 10) : DEFAULT_EMBEDDING_DIMENSIONS;
}

export function shouldAutoCleanupImages(runtime: IAgentRuntime): boolean {
  const setting = getSetting(runtime, "OPENROUTER_AUTO_CLEANUP_IMAGES", "false");
  return setting?.toLowerCase() === "true";
}