From 44338652fd4d936f19dc013ced49086e5accdfe4 Mon Sep 17 00:00:00 2001
From: reanon <85157-reanon@users.noreply.gitgud.io>
Date: Sat, 26 Apr 2025 04:36:12 -0800
Subject: [PATCH] Update file add-key.ts

---
 src/proxy/middleware/request/mutators/add-key.ts | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/src/proxy/middleware/request/mutators/add-key.ts b/src/proxy/middleware/request/mutators/add-key.ts
index 43bb4de..fe937a6 100644
--- a/src/proxy/middleware/request/mutators/add-key.ts
+++ b/src/proxy/middleware/request/mutators/add-key.ts
@@ -49,7 +49,9 @@ export const addKey: ProxyReqMutator = (manager) => {
         assignedKey = keyPool.get("gpt-3.5-turbo-instruct", service);
         break;
       case "openai-image":
-        assignedKey = keyPool.get("dall-e-3", service);
+        // Use the actual model from the request body instead of defaulting to dall-e-3
+        // This ensures that gpt-image-1 requests get keys that are verified for gpt-image-1
+        assignedKey = keyPool.get(body.model, service);
         break;
       case "openai-responses":
         assignedKey = keyPool.get(body.model, service);