Merge pull request #183 from wandb/task/gpt-4o-mini

Default to mini
wandb · Jul 21, 2024 · 7730511 · 7730511
2 parents 42456b0 + d0d3e98
commit 7730511
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/backend/openui/server.py b/backend/openui/server.py
@@ -395,7 +395,7 @@ async def get_openai_models():
     try:
         await openai.models.list()
         # We only support 3.5 and 4 for now
-        return ["gpt-3.5-turbo", "gpt-4o", "gpt-4-turbo"]
+        return ["gpt-3.5-turbo", "gpt-4o-mini", "gpt-4o", "gpt-4-turbo"]
     except Exception:
         logger.warning("Couldn't connect to OpenAI at %s", config.OPENAI_BASE_URL)
         return []

diff --git a/frontend/src/api/openai.ts b/frontend/src/api/openai.ts
@@ -115,7 +115,7 @@ emoji: 🎉
 		if (image) {
 			// TODO: configurable
 			if (model.startsWith('gpt')) {
-				model = 'gpt-4o'
+				model = 'gpt-4o-mini'
 			}
 			const textImageRequirements = query
 				? `The following are some special requirements: \n ${query}`
@@ -156,7 +156,7 @@ emoji: 🎉
 		if (image) {
 			// TODO: configurable
 			if (model.startsWith('gpt')) {
-				model = 'gpt-4o'
+				model = 'gpt-4o-mini'
 			}
 			messages.push({
 				role: 'user',