GPTRouter 0.3.7__tar.gz → 0.3.8__tar.gz

{GPTRouter-0.3.7 → GPTRouter-0.3.8}/GPTRouter.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: GPTRouter
-Version: 0.3.7
+Version: 0.3.8
 Summary: A Python package for working with GPTRouter APIs
 Author-email: sirjanpreet@writesonic.com
 Classifier: Development Status :: 3 - Alpha

{GPTRouter-0.3.7 → GPTRouter-0.3.8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: GPTRouter
-Version: 0.3.7
+Version: 0.3.8
 Summary: A Python package for working with GPTRouter APIs
 Author-email: sirjanpreet@writesonic.com
 Classifier: Development Status :: 3 - Alpha

{GPTRouter-0.3.7 → GPTRouter-0.3.8}/gpt_router/client.py

@@ -391,9 +391,18 @@ class GPTRouterClient:
         )
         generated_images = api_response.get("response", [])
         if isinstance(generated_images, dict):
-            generated_images = generated_images.get("artifacts", [])
+            generated_images = self._extract_images_from_dict(generated_images)
 
         return [
             parse_obj_as(ImageGenerationResponse, generated_img)
             for generated_img in generated_images
         ]
+
+    def _extract_images_from_dict(self, image_dict: dict) -> List[dict]:
+        if artifacts := image_dict.get("artifacts"):
+            return artifacts
+        
+        if image_urls := image_dict.get("imageUrls"):
+            return [{"url": url} for url in image_urls]
+
+        return []

{GPTRouter-0.3.7 → GPTRouter-0.3.8}/gpt_router/enums.py

@@ -15,6 +15,7 @@ class ProvidersEnum(Enum):
     STABLE_DIFFUSION = "stable-diffusion"
     DALL_E = "dall-e"
     OPENROUTER = "openrouter"
+    REPLICATE = "replicate"
 
 class ModelsEnum(Enum):
     OPENAI_GPT_35_TURBO_16K = "gpt-3.5-turbo-16k"
@@ -43,6 +44,8 @@ class ModelsEnum(Enum):
     STABLE_DIFFUSION_XL = "stable-diffusion-xl-1024-v1-0"
     STABLE_DIFFUSION_V6 = "stable-diffusion-v1-6"
 
+    REPLICATE_FLUX = "black-forest-labs/flux-dev"
+
 class StreamingEventType(Enum):
     UPDATE = "update"
     META = "meta"
@@ -150,6 +153,9 @@ class GPTRouterEngines(Enum):
         ProvidersEnum.STABLE_DIFFUSION.value, "stable-diffusion-xl-1024-v1-0"
     )
     DALLE_3 = ModelDetails(ProvidersEnum.DALL_E.value, "dall-e-3")
+    REPLICATE_FLUX = ModelDetails(
+        ProvidersEnum.REPLICATE.value, "black-forest-labs/flux-dev"
+    )
 
     LLAMA_3_70B_NITRO = ModelDetails(
         ProvidersEnum.OPENROUTER, "meta-llama/llama-3-70b-instruct:nitro"

{GPTRouter-0.3.7 → GPTRouter-0.3.8}/setup.py

@@ -5,7 +5,7 @@ with open("README.md", "r") as fh:
 
 setup(
     name="GPTRouter",
-    version="0.3.7",
+    version="0.3.8",
     description="A Python package for working with GPTRouter APIs",
     long_description=long_description,
     long_description_content_type="text/markdown",