GPTRouter 0.5.2__tar.gz → 0.5.3__tar.gz

{GPTRouter-0.5.2 → GPTRouter-0.5.3}/GPTRouter.egg-info/PKG-INFO

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

{GPTRouter-0.5.2 → GPTRouter-0.5.3}/PKG-INFO

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

{GPTRouter-0.5.2 → GPTRouter-0.5.3}/gpt_router/client.py

@@ -10,6 +10,7 @@ from gpt_router.models import (
     GPTRouterMetadata,
     ImageGenerationRequest,
     ImageGenerationResponse,
+    ImageEditRequest,
     ModelGenerationRequest,
     GenerationResponse,
     ChunkedGenerationResponse,
@@ -67,6 +68,8 @@ class GPTRouterClient:
         additional_metadata: Optional[GPTRouterMetadata] = None,
     ):
         self.base_url = base_url
+        if api_key is None:
+            raise ValueError("API key cannot be None. Please provide a valid API key.")
         self.api_key = api_key
         self.request_timeout = request_timeout
         self.additional_metadata = additional_metadata
@@ -463,6 +466,55 @@ class GPTRouterClient:
             for generated_img in generated_images
         ]
 
+    def edit_image(self, *, image_edit_request: ImageEditRequest) -> List[ImageGenerationResponse]:
+        """
+        Edit an image based on the provided ImageEditRequest.
+        This is a synchronous wrapper around the async method.
+        
+        Args:
+            image_edit_request: The request containing the image and editing instructions
+            
+        Returns:
+            List of image generation responses
+        """
+        import asyncio
+        
+        async def _async_call():
+            return await self.aedit_image(image_edit_request=image_edit_request)
+            
+        return asyncio.run(_async_call())
+    
+    async def aedit_image(self, *, image_edit_request: ImageEditRequest) -> List[ImageGenerationResponse]:
+        """
+        Edit an image based on the provided ImageEditRequest.
+        
+        Args:
+            image_edit_request: The request containing the image and editing instructions
+            
+        Returns:
+            List of image generation responses
+        """
+        api_path = "/v1/generate/edit-image"
+        api_method = "POST"
+        api_payload = image_edit_request.dict()
+        
+        api_response = await self._async_api_call(
+            path=api_path,
+            method=api_method,
+            payload=api_payload,
+        )
+        
+        generated_images = api_response.get("response", {}).get("data", [])
+        
+        return [
+            parse_obj_as(ImageGenerationResponse, {
+                "url": img.get("url", None),
+                "base64": img.get("b64_json", None),
+                "finish_reason": "SUCCESS"
+            })
+            for img in generated_images
+        ]
+
     def _extract_images_from_dict(self, image_dict: dict) -> List[dict]:
         if artifacts := image_dict.get("artifacts"):
             return artifacts

{GPTRouter-0.5.2 → GPTRouter-0.5.3}/gpt_router/models.py

@@ -1,8 +1,7 @@
 from uuid import UUID
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, root_validator, validator
 from enum import Enum
-
-from typing import List, Optional, Dict, Any, Union
+from typing import Dict, List, Optional, Any, Union
 from humps import camelize
 
 
@@ -142,6 +141,26 @@ class ImageGenerationResponse(BaseModel):
     finish_reason: Optional[str] = Field(default="SUCCESS", alias="finishReason")
 
 
+class ImageEditRequest(BaseModel):
+    model_name: str = Field(alias="model", default="gpt-image-1")
+    image: Union[str, List[str]] = Field(...)
+    prompt: str = Field(...)
+    mask: Optional[str] = Field(default=None)
+    num_images: Optional[int] = Field(alias="n", default=1)
+    quality: Optional[str] = Field(default="high")
+    response_format: Optional[str] = Field(alias="responseFormat", default=None)
+    size: Optional[str] = Field(default="1024x1024")
+    user: Optional[str] = Field(default=None)
+    
+    class Config:
+        allow_population_by_field_name = True
+        protected_namespaces = ()
+        
+    def dict(self, *args, **kwargs):
+        base_dict = super().dict(*args, **kwargs, exclude_none=True, by_alias=True)
+        return base_dict
+
+
 class ModelRouterGenerationRequest(BaseModel):
     model_name: str
     model_provider: str

{GPTRouter-0.5.2 → GPTRouter-0.5.3}/setup.py

@@ -3,9 +3,10 @@ from setuptools import setup
 with open("README.md", "r") as fh:
     long_description = fh.read()
 
+
 setup(
     name="GPTRouter",
-    version="0.5.2",
+    version="0.5.3",
     description="A Python package for working with GPTRouter APIs",
     long_description=long_description,
     long_description_content_type="text/markdown",