From e5d67a955cd1be393c00b80c38d0f6faf50d81e5 Mon Sep 17 00:00:00 2001
From: Jack McGuire <jackmcguire35@gmail.com>
Date: Sun, 10 Nov 2024 03:09:42 +0000
Subject: [PATCH] fix: image model handling in sqs handler

---
 cmd/sqs/main.go | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/cmd/sqs/main.go b/cmd/sqs/main.go
index 4050849..8031fe9 100644
--- 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.