GPTRouter 0.2.0__tar.gz → 0.2.2__tar.gz

{GPTRouter-0.2.0 → GPTRouter-0.2.2}/GPTRouter.egg-info/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.1
 Name: GPTRouter
-Version: 0.2.0
+Version: 0.2.2
 Summary: A Python package for working with GPTRouter APIs
-Author-email: sirjanpreet.ext@writesonic.com
+Author-email: sirjanpreet@writesonic.com
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Programming Language :: Python :: 3

GPTRouter-0.2.2/GPTRouter.egg-info/requires.txt

@@ -0,0 +1,3 @@
+httpx>=0.23.3
+pydantic>=1.10.13
+pyhumps>=3.8.0

{GPTRouter-0.2.0 → GPTRouter-0.2.2}/PKG-INFO

@@ -1,8 +1,8 @@
 Metadata-Version: 2.1
 Name: GPTRouter
-Version: 0.2.0
+Version: 0.2.2
 Summary: A Python package for working with GPTRouter APIs
-Author-email: sirjanpreet.ext@writesonic.com
+Author-email: sirjanpreet@writesonic.com
 Classifier: Development Status :: 3 - Alpha
 Classifier: Intended Audience :: Developers
 Classifier: Programming Language :: Python :: 3

{GPTRouter-0.2.0 → GPTRouter-0.2.2}/gpt_router/client.py

@@ -6,6 +6,8 @@ import httpx
 from typing import Generator, List, Union
 
 from gpt_router.models import (
+    ImageGenerationRequest,
+    ImageGenerationResponse,
     ModelGenerationRequest,
     GenerationResponse,
     ChunkedGenerationResponse,
@@ -226,3 +228,24 @@ class GPTRouterClient:
             payload=api_payload,
         )
         return parse_obj_as(GenerationResponse, result)
+
+    async def agenerate_images(
+        self, *, image_generation_request: ImageGenerationRequest
+    ) -> List[ImageGenerationResponse]:
+        api_path = "/v1/generate/generate-image"
+        api_method = "POST"
+        api_payload = image_generation_request.dict()
+
+        api_response = await self._async_api_call(
+            path=api_path,
+            method=api_method,
+            payload=api_payload,
+        )
+        generated_images = api_response.get("response", [])
+        if isinstance(generated_images, dict):
+            generated_images = generated_images.get('artifacts', [])
+        
+        return [
+            parse_obj_as(ImageGenerationResponse, generated_img)
+            for generated_img in generated_images
+        ]

{GPTRouter-0.2.0 → GPTRouter-0.2.2}/gpt_router/enums.py

@@ -6,6 +6,8 @@ class ProvidersEnum(Enum):
     ANTHROPIC = "anthropic"
     AZURE_CHAT_OPENAI = "azure_chat_openai"
     COHERE = "cohere"
+    DALLE = "dall-e"
+    STABLE_DIFFUSION = "stable-diffusion-v1-6"
 
 class ModelsEnum(Enum):
     OPENAI_GPT_35_TURBO_16K = "gpt-3.5-turbo-16k"
@@ -29,6 +31,11 @@ class ModelsEnum(Enum):
 
     CLAUDE_INSTANT_12 = "claude-instant-1.2"
 
+    DALLE_3 = "dall-e-3"
+    
+    STABLE_DIFFUSION_XL = "stable-diffusion-xl-1024-v1-0"
+    STABLE_DIFFUSION_V6 = "stable-diffusion-v1-6"
+
 class StreamingEventType(Enum):
     UPDATE = "update"
     META = "meta"

{GPTRouter-0.2.0 → GPTRouter-0.2.2}/gpt_router/models.py

@@ -66,3 +66,32 @@ class GenerationResponse(BaseModel):
 class ChunkedGenerationResponse(BaseModel):
     event: str
     data: dict
+
+
+class ImageGenerationRequest(BaseModel):
+    model_name: str = Field(alias="model")
+    provider_name: str = Field(alias="imageVendor")
+    prompt: str
+    num_images: int = Field(alias="numImages", default=1)
+    width: Optional[int] = Field(None, alias="width")
+    height: Optional[int] = Field(None, alias="height")
+
+    class Config:
+        allow_population_by_field_name = True
+        protected_namespaces = ()
+
+    @property
+    def size(self) -> Optional[str]:
+        return f"{self.width}x{self.height}" if self.width and self.height else None
+
+    def dict(self, *args, **kwargs):
+        base_dict = super().dict(*args, **kwargs, exclude_none=True, by_alias=True)
+        if self.size is not None:
+            base_dict["size"] = self.size
+        return base_dict
+
+
+class ImageGenerationResponse(BaseModel):
+    url: str = None
+    base64: str = None
+    finish_reason: Optional[str] = Field(default="SUCCESS", alias='finishReason')

GPTRouter-0.2.2/setup.py

@@ -0,0 +1,22 @@
+from setuptools import setup
+
+with open("README.md", "r") as fh:
+    long_description = fh.read()
+
+setup(
+    name="GPTRouter",
+    version="0.2.2",
+    description="A Python package for working with GPTRouter APIs",
+    long_description=long_description,
+    long_description_content_type="text/markdown",
+    author_email="sirjanpreet@writesonic.com",
+    packages=["gpt_router"],
+    install_requires=["httpx>=0.23.3", "pydantic>=1.10.13", "pyhumps>=3.8.0"],
+    classifiers=[
+        "Development Status :: 3 - Alpha",
+        "Intended Audience :: Developers",
+        "Programming Language :: Python :: 3",
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
+    ],
+)

GPTRouter-0.2.0/GPTRouter.egg-info/requires.txt

@@ -1,2 +0,0 @@
-httpx>=0.25.2
-pydantic==1.10.14

GPTRouter-0.2.0/setup.py

@@ -1,25 +0,0 @@
-from setuptools import setup
-
-with open("README.md", "r") as fh:
-    long_description = fh.read()
-
-setup(
-    name='GPTRouter',
-    version='0.2.0',
-    description='A Python package for working with GPTRouter APIs',
-    long_description=long_description,
-    long_description_content_type="text/markdown",
-    author_email='sirjanpreet.ext@writesonic.com',
-    packages=['gpt_router'],
-    install_requires=[
-        'httpx>=0.25.2',
-        "pydantic==1.10.14"
-    ],
-    classifiers=[
-        'Development Status :: 3 - Alpha',
-        'Intended Audience :: Developers',
-        'Programming Language :: Python :: 3',
-        'Programming Language :: Python :: 3.10',
-        'Programming Language :: Python :: 3.11',
-    ],
-)