agent-relay-sdk 0.1.0 → 0.2.1

package/src/provider-catalog.ts

@@ -0,0 +1,174 @@
+import type { SpawnProvider } from "./types.js";
+
+export type ProviderEffort = "low" | "medium" | "high" | "xhigh" | "max";
+export type ProviderCatalogValueSource = "catalog" | "provider" | "runtime" | "override";
+export type ProviderCatalogConfidence = "declared" | "verified" | "estimated" | "unknown";
+
+export interface ProviderCatalogValue<T> {
+  value: T;
+  source: ProviderCatalogValueSource;
+  confidence: ProviderCatalogConfidence;
+  lastUpdatedAt?: number;
+}
+
+export interface ProviderModelLimits {
+  contextWindowTokens?: ProviderCatalogValue<number>;
+  maxOutputTokens?: ProviderCatalogValue<number>;
+}
+
+export interface ProviderModelModalities {
+  input: {
+    text: boolean;
+    image?: boolean;
+    audio?: boolean;
+    video?: boolean;
+    pdf?: boolean;
+  };
+  output: {
+    text: boolean;
+    image?: boolean;
+    audio?: boolean;
+    video?: boolean;
+  };
+}
+
+export interface ProviderModelToolCapabilities {
+  code?: boolean;
+  review?: boolean;
+  debug?: boolean;
+  refactor?: boolean;
+  shell?: boolean;
+  fileRead?: boolean;
+  fileWrite?: boolean;
+  webSearch?: boolean;
+  imageGeneration?: boolean;
+  imageEditing?: boolean;
+}
+
+export interface ProviderModelCapabilities {
+  modalities: ProviderModelModalities;
+  tools?: ProviderModelToolCapabilities;
+  source: ProviderCatalogValueSource;
+  confidence: ProviderCatalogConfidence;
+  lastUpdatedAt?: number;
+}
+
+export interface ProviderModelCatalogEntry {
+  alias: string;
+  label: string;
+  providerModel: string;
+  efforts: ProviderEffort[];
+  defaultEffort?: ProviderEffort;
+  limits?: ProviderModelLimits;
+  capabilities?: ProviderModelCapabilities;
+}
+
+export interface ProviderCatalogEntry {
+  provider: SpawnProvider;
+  label: string;
+  defaultModel?: string;
+  models: ProviderModelCatalogEntry[];
+}
+
+export interface ProviderSelection {
+  provider: SpawnProvider;
+  model?: string;
+  effort?: ProviderEffort;
+}
+
+export interface ResolvedProviderSelection {
+  provider: SpawnProvider;
+  modelAlias?: string;
+  providerModel?: string;
+  effort?: ProviderEffort;
+  model?: ProviderModelCatalogEntry;
+}
+
+const CLAUDE_LOW_TO_MAX: ProviderEffort[] = ["low", "medium", "high", "max"];
+const CLAUDE_LOW_TO_XHIGH_MAX: ProviderEffort[] = ["low", "medium", "high", "xhigh", "max"];
+const CODEX_REASONING: ProviderEffort[] = ["low", "medium", "high", "xhigh"];
+const CONTEXT_200K: ProviderCatalogValue<number> = { value: 200_000, source: "catalog", confidence: "declared" };
+const CONTEXT_1M: ProviderCatalogValue<number> = { value: 1_000_000, source: "catalog", confidence: "declared" };
+const CODE_MODEL_CAPABILITIES: ProviderModelCapabilities = {
+  modalities: {
+    input: { text: true, image: true },
+    output: { text: true },
+  },
+  tools: {
+    code: true,
+    review: true,
+    debug: true,
+    refactor: true,
+  },
+  source: "catalog",
+  confidence: "declared",
+};
+
+function codeModel(limits?: ProviderModelLimits): Pick<ProviderModelCatalogEntry, "limits" | "capabilities"> {
+  return {
+    ...(limits ? { limits } : {}),
+    capabilities: CODE_MODEL_CAPABILITIES,
+  };
+}
+
+export const PROVIDER_CATALOG: Record<SpawnProvider, ProviderCatalogEntry> = {
+  claude: {
+    provider: "claude",
+    label: "Claude Code",
+    defaultModel: "sonnet-4.6",
+    models: [
+      { alias: "opus-4.8", label: "Opus 4.8", providerModel: "claude-opus-4-8", efforts: CLAUDE_LOW_TO_XHIGH_MAX, defaultEffort: "medium", ...codeModel({ contextWindowTokens: CONTEXT_200K }) },
+      { alias: "opus-4.8[1m]", label: "Opus 4.8 1M", providerModel: "claude-opus-4-8[1m]", efforts: CLAUDE_LOW_TO_XHIGH_MAX, defaultEffort: "medium", ...codeModel({ contextWindowTokens: CONTEXT_1M }) },
+      { alias: "opus-4.7", label: "Opus 4.7", providerModel: "claude-opus-4-7", efforts: CLAUDE_LOW_TO_XHIGH_MAX, defaultEffort: "medium", ...codeModel({ contextWindowTokens: CONTEXT_200K }) },
+      { alias: "opus-4.7[1m]", label: "Opus 4.7 1M", providerModel: "claude-opus-4-7[1m]", efforts: CLAUDE_LOW_TO_XHIGH_MAX, defaultEffort: "medium", ...codeModel({ contextWindowTokens: CONTEXT_1M }) },
+      { alias: "opus-4.6", label: "Opus 4.6", providerModel: "claude-opus-4-6", efforts: CLAUDE_LOW_TO_MAX, defaultEffort: "medium", ...codeModel({ contextWindowTokens: CONTEXT_200K }) },
+      { alias: "opus-4.6[1m]", label: "Opus 4.6 1M", providerModel: "claude-opus-4-6[1m]", efforts: CLAUDE_LOW_TO_MAX, defaultEffort: "medium", ...codeModel({ contextWindowTokens: CONTEXT_1M }) },
+      { alias: "sonnet-4.6", label: "Sonnet 4.6", providerModel: "claude-sonnet-4-6", efforts: CLAUDE_LOW_TO_MAX, defaultEffort: "medium", ...codeModel({ contextWindowTokens: CONTEXT_200K }) },
+      { alias: "sonnet-4.6[1m]", label: "Sonnet 4.6 1M", providerModel: "claude-sonnet-4-6[1m]", efforts: CLAUDE_LOW_TO_MAX, defaultEffort: "medium", ...codeModel({ contextWindowTokens: CONTEXT_1M }) },
+      { alias: "haiku", label: "Haiku", providerModel: "haiku", efforts: [], ...codeModel() },
+    ],
+  },
+  codex: {
+    provider: "codex",
+    label: "Codex",
+    defaultModel: "gpt-5.5",
+    models: [
+      { alias: "gpt-5.5", label: "GPT-5.5", providerModel: "gpt-5.5", efforts: CODEX_REASONING, defaultEffort: "medium", ...codeModel({ contextWindowTokens: CONTEXT_200K }) },
+      { alias: "gpt-5.4", label: "GPT-5.4", providerModel: "gpt-5.4", efforts: CODEX_REASONING, defaultEffort: "medium", ...codeModel({ contextWindowTokens: CONTEXT_200K }) },
+      { alias: "gpt-5.4-mini", label: "GPT-5.4 Mini", providerModel: "gpt-5.4-mini", efforts: CODEX_REASONING, defaultEffort: "medium", ...codeModel({ contextWindowTokens: CONTEXT_200K }) },
+      { alias: "gpt-5.3-codex", label: "GPT-5.3 Codex", providerModel: "gpt-5.3-codex", efforts: CODEX_REASONING, defaultEffort: "medium", ...codeModel({ contextWindowTokens: CONTEXT_200K }) },
+      { alias: "gpt-5.3-codex-spark", label: "GPT-5.3 Codex Spark", providerModel: "gpt-5.3-codex-spark", efforts: CODEX_REASONING, defaultEffort: "high", ...codeModel({ contextWindowTokens: CONTEXT_200K }) },
+      { alias: "gpt-5.2", label: "GPT-5.2", providerModel: "gpt-5.2", efforts: CODEX_REASONING, defaultEffort: "medium", ...codeModel({ contextWindowTokens: CONTEXT_200K }) },
+    ],
+  },
+};
+
+export function providerCatalogList(): ProviderCatalogEntry[] {
+  return Object.values(PROVIDER_CATALOG);
+}
+
+export function getProviderCatalog(provider: SpawnProvider): ProviderCatalogEntry {
+  return PROVIDER_CATALOG[provider];
+}
+
+export function resolveProviderSelection(selection: ProviderSelection): ResolvedProviderSelection {
+  const catalog = getProviderCatalog(selection.provider);
+  const alias = selection.model || catalog.defaultModel;
+  const model = alias ? catalog.models.find((entry) => entry.alias === alias) : undefined;
+  if (alias && !model) {
+    throw new Error(`unknown ${selection.provider} model: ${alias}`);
+  }
+  if (selection.effort && !model) {
+    throw new Error(`effort requires a ${selection.provider} model`);
+  }
+  if (selection.effort && model && !model.efforts.includes(selection.effort)) {
+    throw new Error(`${selection.provider} model ${model.alias} does not support effort ${selection.effort}`);
+  }
+  return {
+    provider: selection.provider,
+    modelAlias: model?.alias,
+    providerModel: model?.providerModel,
+    effort: selection.effort,
+    model,
+  };
+}