enh: rework getLargeWhitelistedModel (#8159)

Co-authored-by: Henry Fontanier <[email protected]>
dust-tt · Oct 22, 2024 · cea37f6 · cea37f6
1 parent 0c9835d
commit cea37f6
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;