notioncode 0.1.0 → 0.1.2

package/src/shared/modelRegistry.js

@@ -0,0 +1,163 @@
+/**
+ * Shared model registry used by both the frontend workspace UI and the
+ * agent-runtime sidecar.
+ */
+
+/** @typedef {'claude' | 'cursor' | 'codex' | 'gemini'} AgentProvider */
+
+/**
+ * @typedef {{ value: string, label: string }} ModelOption
+ */
+
+export const CLAUDE_MODELS = {
+  OPTIONS: [
+    { value: 'sonnet', label: 'Sonnet' },
+    { value: 'opus', label: 'Opus' },
+    { value: 'haiku', label: 'Haiku' },
+    { value: 'opusplan', label: 'Opus Plan' },
+    { value: 'sonnet[1m]', label: 'Sonnet [1M]' },
+  ],
+  DEFAULT: 'sonnet',
+};
+
+export const CURSOR_MODELS = {
+  OPTIONS: [
+    { value: 'opus-4.6-thinking', label: 'Claude 4.6 Opus (Thinking)' },
+    { value: 'gpt-5.3-codex', label: 'GPT-5.3' },
+    { value: 'gpt-5.2-high', label: 'GPT-5.2 High' },
+    { value: 'gemini-3-pro', label: 'Gemini 3 Pro' },
+    { value: 'opus-4.5-thinking', label: 'Claude 4.5 Opus (Thinking)' },
+    { value: 'gpt-5.2', label: 'GPT-5.2' },
+    { value: 'gpt-5.1', label: 'GPT-5.1' },
+    { value: 'gpt-5.1-high', label: 'GPT-5.1 High' },
+    { value: 'composer-1', label: 'Composer 1' },
+    { value: 'auto', label: 'Auto' },
+    { value: 'sonnet-4.5', label: 'Claude 4.5 Sonnet' },
+    { value: 'sonnet-4.5-thinking', label: 'Claude 4.5 Sonnet (Thinking)' },
+    { value: 'opus-4.5', label: 'Claude 4.5 Opus' },
+    { value: 'gpt-5.1-codex', label: 'GPT-5.1 Codex' },
+    { value: 'gpt-5.1-codex-high', label: 'GPT-5.1 Codex High' },
+    { value: 'gpt-5.1-codex-max', label: 'GPT-5.1 Codex Max' },
+    { value: 'gpt-5.1-codex-max-high', label: 'GPT-5.1 Codex Max High' },
+    { value: 'opus-4.1', label: 'Claude 4.1 Opus' },
+    { value: 'grok', label: 'Grok' },
+  ],
+  DEFAULT: 'gpt-5-3-codex',
+};
+
+export const CODEX_MODELS = {
+  OPTIONS: [
+    { value: 'gpt-5.4', label: 'GPT-5.4' },
+    { value: 'gpt-5.3-codex', label: 'GPT-5.3 Codex' },
+    { value: 'gpt-5.2-codex', label: 'GPT-5.2 Codex' },
+    { value: 'gpt-5.2', label: 'GPT-5.2' },
+    { value: 'gpt-5.1-codex-max', label: 'GPT-5.1 Codex Max' },
+    { value: 'o3', label: 'O3' },
+    { value: 'o4-mini', label: 'O4-mini' },
+  ],
+  DEFAULT: 'gpt-5.4',
+};
+
+export const GEMINI_MODELS = {
+  OPTIONS: [
+    { value: 'gemini-3.1-pro-preview', label: 'Gemini 3.1 Pro Preview' },
+    { value: 'gemini-3-pro-preview', label: 'Gemini 3 Pro Preview' },
+    { value: 'gemini-3-flash-preview', label: 'Gemini 3 Flash Preview' },
+    { value: 'gemini-2.5-flash', label: 'Gemini 2.5 Flash' },
+    { value: 'gemini-2.5-pro', label: 'Gemini 2.5 Pro' },
+    { value: 'gemini-2.0-flash-lite', label: 'Gemini 2.0 Flash Lite' },
+    { value: 'gemini-2.0-flash', label: 'Gemini 2.0 Flash' },
+    { value: 'gemini-2.0-pro-exp', label: 'Gemini 2.0 Pro Experimental' },
+    { value: 'gemini-2.0-flash-thinking-exp', label: 'Gemini 2.0 Flash Thinking' },
+  ],
+  DEFAULT: 'gemini-2.5-flash',
+};
+
+export const MODEL_CONFIGS = {
+  claude: CLAUDE_MODELS,
+  cursor: CURSOR_MODELS,
+  codex: CODEX_MODELS,
+  gemini: GEMINI_MODELS,
+};
+
+const LEGACY_MODEL_MIGRATIONS = {
+  claude: {
+    sonnet: 'sonnet',
+    opus: 'opus',
+    haiku: 'haiku',
+    'claude-3-5-sonnet-20241022': 'sonnet',
+    'claude-3-7-sonnet-20250219': 'sonnet',
+    'claude-3-opus-20240229': 'opus',
+    'claude-3-5-haiku-20241022': 'haiku',
+  },
+  cursor: {},
+  codex: {},
+  gemini: {},
+};
+
+/**
+ * @param {AgentProvider} provider
+ * @returns {ModelOption[]}
+ */
+export function getModelOptions(provider) {
+  return MODEL_CONFIGS[provider]?.OPTIONS ?? [];
+}
+
+/**
+ * @param {AgentProvider} provider
+ * @returns {string}
+ */
+export function getDefaultModel(provider) {
+  return MODEL_CONFIGS[provider]?.DEFAULT ?? CODEX_MODELS.DEFAULT;
+}
+
+/**
+ * @param {AgentProvider} provider
+ * @param {string | null | undefined} value
+ * @returns {boolean}
+ */
+export function isSupportedModel(provider, value) {
+  if (!value) {
+    return false;
+  }
+
+  return getModelOptions(provider).some((option) => option.value === value);
+}
+
+/**
+ * @param {AgentProvider} provider
+ * @param {string | null | undefined} value
+ * @returns {string | null}
+ */
+export function migrateLegacyModel(provider, value) {
+  if (!value) {
+    return null;
+  }
+
+  const directMigration = LEGACY_MODEL_MIGRATIONS[provider]?.[value];
+  if (directMigration) {
+    return directMigration;
+  }
+
+  if (provider === 'claude' && value.startsWith('claude-')) {
+    return getDefaultModel('claude');
+  }
+
+  return value;
+}
+
+/**
+ * Normalize a model selection to a canonical supported value for a provider.
+ *
+ * @param {AgentProvider} provider
+ * @param {string | null | undefined} value
+ * @returns {string}
+ */
+export function normalizeModel(provider, value) {
+  const migrated = migrateLegacyModel(provider, value);
+  if (migrated && isSupportedModel(provider, migrated)) {
+    return migrated;
+  }
+
+  return getDefaultModel(provider);
+}