Merge pull request #35 from Integration-Automation/dev

Update version and test

stable.toml → dev.toml

@@ -5,8 +5,8 @@ requires = ["setuptools>=61.0"]
 build-backend = "setuptools.build_meta"
 
 [project]
-name = "re_edge_gpt"
-version = "0.0.13"
+name = "re_edge_gpt_dev"
+version = "0.0.14"
 authors = [
     { name = "JE-Chen", email = "jechenmailman@gmail.com" },
 ]

pyproject.toml

@@ -5,8 +5,8 @@ requires = ["setuptools>=61.0"]
 build-backend = "setuptools.build_meta"
 
 [project]
-name = "re_edge_gpt_dev"
-version = "0.0.13"
+name = "re_edge_gpt"
+version = "0.0.14"
 authors = [
     { name = "JE-Chen", email = "jechenmailman@gmail.com" },
 ]

re_edge_gpt/__init__.py

@@ -1,7 +1,7 @@
-from re_edge_gpt.re_edge_gpt import Chatbot
-from re_edge_gpt.re_edge_gpt import ConversationStyle
 from re_edge_gpt.image_genearation import ImageGen
 from re_edge_gpt.image_genearation import ImageGenAsync
+from re_edge_gpt.re_edge_gpt import Chatbot
+from re_edge_gpt.re_edge_gpt import ConversationStyle
 
 __all__ = [
     "Chatbot", "ConversationStyle", "ImageGen", "ImageGenAsync"

re_edge_gpt/image_genearation.py

@@ -138,7 +138,7 @@ def get_images(self, prompt: str) -> Union[list, None]:
             url = f"{BING_URL}/images/create?q={url_encoded_prompt}&rt=3&FORM=GUH2CR"
             response = self.session.post(url, allow_redirects=False, timeout=200)
             if response.status_code != 302:
-                print("Image Creating Please Retry Later", end="", flush=True)
+                print("Image create failed pls check cookie or old image still creating", flush=True)
                 return
                 # Get redirect URL
         redirect_url = response.headers["Location"].replace("&nfy=1", "")
@@ -310,9 +310,9 @@ async def get_images(self, prompt: str) -> Union[list, None]:
                 follow_redirects=False,
                 timeout=200,
             )
-            if response.status_code != 302:
-                print("Image Creating Please Retry Later", end="", flush=True)
-                return None
+        if response.status_code != 302:
+            print("Image create failed pls check cookie or old image still creating", flush=True)
+            return None
         # Get redirect URL
         redirect_url = response.headers["Location"].replace("&nfy=1", "")
         request_id = redirect_url.split("id=")[-1]

test/unit_test/test_generate_image/test_generation.py

@@ -29,7 +29,7 @@ def test_generate_image_sync():
 
 # Generate image list async
 async def test_generate_image_async():
-    image_list = await async_gen.get_images("Big, Golden, Tree")
+    image_list = await async_gen.get_images("La Tour Eiffel")
     print(image_list)