fix: image model handling in sqs handler

jackmcguire1 · Nov 10, 2024 · e5d67a9 · e5d67a9
1 parent 72a1e92
commit e5d67a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/cmd/sqs/main.go b/cmd/sqs/main.go
@@ -34,7 +34,7 @@ func (handler *SqsHandler) ProcessChatGPTRequest(ctx context.Context, req *chatm
 
 	if req.ImageModel != nil {
 		switch *req.ImageModel {
-		case chatmodels.IMAGE_MODEL_STABLE_DIFFUSION, chatmodels.IMAGE_MODEL_DALL_E_2:
+		case chatmodels.IMAGE_MODEL_STABLE_DIFFUSION, chatmodels.IMAGE_MODEL_DALL_E_2, chatmodels.IMAGE_MODEL_DALL_E_3:
 			imageBody, err := handler.ChatModelSvc.GenerateImage(ctx, req.Prompt, *req.ImageModel)
 			if err != nil {
 				handler.Logger.