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);