Merge pull request #19 from stainless-sdks/blaine/fix-image-types

Nutlope · web-flow · commit d2f23312e23a · 2025-10-23T14:34:43.000-07:00
chore: Fix Type resolution on Image API via param discrimination
diff --git a/src/resources/images.ts b/src/resources/images.ts
@@ -16,7 +16,14 @@ export class Images extends APIResource {
    * });
    * ```
    */
-  create(body: ImageCreateParams, options?: RequestOptions): APIPromise<ImageFile> {
+  create<Body extends ImageCreateParams>(
+    body: Body,
+    options?: RequestOptions,
+  ): APIPromise<
+    Omit<ImageFile, 'data'> & {
+      data: Body['response_format'] extends 'base64' ? ImageDataB64[] : ImageDataURL[];
+    }
+  > {
     return this._client.post('/images/generations', { body, ...options });
   }
 }
