move steps to kwargs (#384)

Author: zainhas

src/together/resources/images.py

@@ -21,7 +21,6 @@ def generate(
         *,
         prompt: str,
         model: str,
-        steps: int | None = 20,
         seed: int | None = None,
         n: int | None = 1,
         height: int | None = 1024,
@@ -37,8 +36,6 @@ def generate(
 
             model (str, optional): The model to use for image generation.
 
-            steps (int, optional): Number of generation steps. Defaults to 20
-
             seed (int, optional): Seed used for generation. Can be used to reproduce image generations.
                 Defaults to None.
 
@@ -51,7 +48,8 @@ def generate(
             negative_prompt (str, optional): The prompt or prompts not to guide the image generation.
                 Defaults to None
 
-            image_base64: (str, optional): Reference image used for generation. Defaults to None.
+            **kwargs: Additional parameters like steps (int, optional): Number of generation steps,
+                image_base64 (str, optional): Reference image used for generation, etc.
 
         Returns:
             ImageResponse: Object containing image data
@@ -64,7 +62,6 @@ def generate(
         parameter_payload = ImageRequest(
             prompt=prompt,
             model=model,
-            steps=steps,
             seed=seed,
             n=n,
             height=height,
@@ -96,7 +93,6 @@ async def generate(
         *,
         prompt: str,
         model: str,
-        steps: int | None = 20,
         seed: int | None = None,
         n: int | None = 1,
         height: int | None = 1024,
@@ -112,8 +108,6 @@ async def generate(
 
             model (str, optional): The model to use for image generation.
 
-            steps (int, optional): Number of generation steps. Defaults to 20
-
             seed (int, optional): Seed used for generation. Can be used to reproduce image generations.
                 Defaults to None.
 
@@ -126,7 +120,8 @@ async def generate(
             negative_prompt (str, optional): The prompt or prompts not to guide the image generation.
                 Defaults to None
 
-            image_base64: (str, optional): Reference image used for generation. Defaults to None.
+            **kwargs: Additional parameters like steps (int, optional): Number of generation steps,
+                image_base64 (str, optional): Reference image used for generation, etc.
 
         Returns:
             ImageResponse: Object containing image data
@@ -139,7 +134,6 @@ async def generate(
         parameter_payload = ImageRequest(
             prompt=prompt,
             model=model,
-            steps=steps,
             seed=seed,
             n=n,
             height=height,

src/together/types/images.py

@@ -10,8 +10,6 @@ class ImageRequest(BaseModel):
     prompt: str
     # model to query
     model: str
-    # num generation steps
-    steps: int | None = 20
     # seed
     seed: int | None = None
     # number of results to return