update image gen params

Author: winzamark123

jigsawstack/image_generation.py

@@ -6,34 +6,51 @@
 from typing import List, Union
 from ._config import ClientConfig
 
+class AdvanceConfig(TypedDict):
+    negative_prompt: NotRequired[str]
+    guidance: NotRequired[int]
+    seed: NotRequired[int]
+
 class ImageGenerationParams(TypedDict):
     prompt: Required[str]
-    """"
-    The text to generate the image from."
+    """
+    The text to generate the image from.
     """
     aspect_ratio: NotRequired[Literal["1:1", "16:9", "21:9", "3:2", "2:3", "4:5", "5:4", "3:4", "4:3", "9:16", "9:21"]]
     """
     The aspect ratio of the image. The default is 1:1.
     """
     width: NotRequired[int]
-    """"
-    The width of the image. The default is 512."
+    """
+    The width of the image. The default is 512.
     """
     height: NotRequired[int]
     """
-    The height of the image. The default is 512."
+    The height of the image. The default is 512.
     """
     steps: NotRequired[int]
-    """"
-    The number of steps to generate the image.""
     """
-    advance_config: NotRequired[Dict[str, Union[int, str]]]
+    The number of steps to generate the image.
+    """
+    output_format: NotRequired[Literal["png", "svg"]]
+    """
+    The output format of the generated image. Can be 'png' or 'svg'.
+    """
+    advance_config: NotRequired[AdvanceConfig]
     """
     The advance configuration for the image generation. The default is None.
     You can pass the following:
-    - `seed`: The seed for the image generation. The default is None.
-    - `guidance`: The guidance for the image generation. The default is None.
-    - `negative_prompt`: The negative prompt for the image generation. The default is None.
+    - `negative_prompt`: The negative prompt for the image generation
+    - `guidance`: The guidance scale for the image generation
+    - `seed`: The seed for reproducible generation
+    """
+    url: NotRequired[str]
+    """
+    URL to use as image input.
+    """
+    file_store_key: NotRequired[str]
+    """
+    File store key to use as image input.
     """
 
 class ImageGenerationResponse(TypedDict):