enh: rework getLargeWhitelistedModel

dust-tt · Oct 22, 2024 · 26f5e37 · 26f5e37
1 parent 5b74107
commit 26f5e37
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/types/src/front/lib/assistant.ts b/types/src/front/lib/assistant.ts
@@ -63,11 +63,11 @@ export function getSmallWhitelistedModel(
 export function getLargeWhitelistedModel(
   owner: WorkspaceType
 ): ModelConfigurationType | null {
-  if (isProviderWhitelisted(owner, "openai")) {
-    return GPT_4_TURBO_MODEL_CONFIG;
-  }
   if (isProviderWhitelisted(owner, "anthropic")) {
-    return CLAUDE_3_OPUS_DEFAULT_MODEL_CONFIG;
+    return CLAUDE_3_5_SONNET_DEFAULT_MODEL_CONFIG;
+  }
+  if (isProviderWhitelisted(owner, "openai")) {
+    return GPT_4O_MODEL_CONFIG;
   }
   if (isProviderWhitelisted(owner, "google_ai_studio")) {
     return GEMINI_PRO_DEFAULT_MODEL_CONFIG;