edit : 오타 수정

main_rest/app/models.py

@@ -598,7 +598,7 @@ class BingCookieSetReq(BaseModel):
     cookie : str = Field('',description='쿠키 데이터', example='')
     
     
-class VactorImageSearchReq(BaseModel):
+class VectorImageSearchReq(BaseModel):
     """
     ### [Request] vector image search request
     """
@@ -660,7 +660,27 @@ class ImageGenerateRes(ResponseBase):
         ImageGenerateRes.error = None
         ImageGenerateRes.imageLen = img_len
         return ImageGenerateRes
+
+class ImageGenerateDataRes(ResponseBase):
+    """
+    ### image generate Data response
+    """
+    imageData : int = Field(0, description='실제 이미지 생성 갯수', example=1)
     
+    @staticmethod
+    def set_error(error,img_data=''):
+        ImageGenerateRes.result = False
+        ImageGenerateRes.error = str(error)
+        ImageGenerateRes.imageData = img_data
+        
+        return ImageGenerateRes
+    
+    @staticmethod
+    def set_message(img_data):
+        ImageGenerateRes.result = True
+        ImageGenerateRes.error = None
+        ImageGenerateRes.imageData = img_data
+        return ImageGenerateRes
     
 class BingCookieSetRes(ResponseBase):
     """