init

2026-04-25 06:45:36 +09:00
commit e77acee8ba
1903 changed files with 513282 additions and 0 deletions
@@ -0,0 +1,50 @@
+import memoize from 'lodash-es/memoize.js'
+import { getAPIProvider } from './providers.js'
+
+export type ModelCapabilityOverride =
+  | 'effort'
+  | 'max_effort'
+  | 'thinking'
+  | 'adaptive_thinking'
+  | 'interleaved_thinking'
+
+const TIERS = [
+  {
+    modelEnvVar: 'ANTHROPIC_DEFAULT_OPUS_MODEL',
+    capabilitiesEnvVar: 'ANTHROPIC_DEFAULT_OPUS_MODEL_SUPPORTED_CAPABILITIES',
+  },
+  {
+    modelEnvVar: 'ANTHROPIC_DEFAULT_SONNET_MODEL',
+    capabilitiesEnvVar: 'ANTHROPIC_DEFAULT_SONNET_MODEL_SUPPORTED_CAPABILITIES',
+  },
+  {
+    modelEnvVar: 'ANTHROPIC_DEFAULT_HAIKU_MODEL',
+    capabilitiesEnvVar: 'ANTHROPIC_DEFAULT_HAIKU_MODEL_SUPPORTED_CAPABILITIES',
+  },
+] as const
+
+/**
+ * Check whether a 3p model capability override is set for a model that matches one of
+ * the pinned ANTHROPIC_DEFAULT_*_MODEL env vars.
+ */
+export const get3PModelCapabilityOverride = memoize(
+  (model: string, capability: ModelCapabilityOverride): boolean | undefined => {
+    if (getAPIProvider() === 'firstParty') {
+      return undefined
+    }
+    const m = model.toLowerCase()
+    for (const tier of TIERS) {
+      const pinned = process.env[tier.modelEnvVar]
+      const capabilities = process.env[tier.capabilitiesEnvVar]
+      if (!pinned || capabilities === undefined) continue
+      if (m !== pinned.toLowerCase()) continue
+      return capabilities
+        .toLowerCase()
+        .split(',')
+        .map(s => s.trim())
+        .includes(capability)
+    }
+    return undefined
+  },
+  (model, capability) => `${model.toLowerCase()}:${capability}`,
+)