@leonardo-ai/sdk 4.6.0 → 4.14.1

package/src/sdk/variation.ts

@@ -2,14 +2,13 @@
  * Code generated by Speakeasy (https://speakeasyapi.dev). DO NOT EDIT.
  */
 
-import { SDKHooks } from "../hooks";
-import { SDK_METADATA, SDKOptions, serverURLFromOptions } from "../lib/config";
-import * as enc$ from "../lib/encodings";
-import { HTTPClient } from "../lib/http";
-import * as schemas$ from "../lib/schemas";
-import { ClientSDK, RequestOptions } from "../lib/sdks";
-import * as errors from "./models/errors";
-import * as operations from "./models/operations";
+import { SDKHooks } from "../hooks/hooks.js";
+import { SDKOptions, serverURLFromOptions } from "../lib/config.js";
+import { encodeJSON as encodeJSON$, encodeSimple as encodeSimple$ } from "../lib/encodings.js";
+import { HTTPClient } from "../lib/http.js";
+import * as schemas$ from "../lib/schemas.js";
+import { ClientSDK, RequestOptions } from "../lib/sdks.js";
+import * as operations from "./models/operations/index.js";
 
 export class Variation extends ClientSDK {
     private readonly options$: SDKOptions & { hooks?: SDKHooks };
@@ -38,6 +37,87 @@ export class Variation extends ClientSDK {
         void this.options$;
     }
 
+    /**
+     * Create using Universal Upscaler
+     *
+     * @remarks
+     * This endpoint will create a high resolution image using Universal Upscaler
+     */
+    async createUniversalUpscalerJob(
+        request: operations.CreateUniversalUpscalerJobRequestBody,
+        options?: RequestOptions
+    ): Promise<operations.CreateUniversalUpscalerJobResponse> {
+        const input$ = request;
+
+        const payload$ = schemas$.parse(
+            input$,
+            (value$) =>
+                operations.CreateUniversalUpscalerJobRequestBody$outboundSchema.parse(value$),
+            "Input validation failed"
+        );
+        const body$ = encodeJSON$("body", payload$, { explode: true });
+
+        const path$ = this.templateURLComponent("/variations/universal-upscaler")();
+
+        const query$ = "";
+
+        const headers$ = new Headers({
+            "Content-Type": "application/json",
+            Accept: "application/json",
+        });
+
+        let security$;
+        if (typeof this.options$.bearerAuth === "function") {
+            security$ = { bearerAuth: await this.options$.bearerAuth() };
+        } else if (this.options$.bearerAuth) {
+            security$ = { bearerAuth: this.options$.bearerAuth };
+        } else {
+            security$ = {};
+        }
+        const context = {
+            operationID: "CreateUniversalUpscalerJob",
+            oAuth2Scopes: [],
+            securitySource: this.options$.bearerAuth,
+        };
+        const securitySettings$ = this.resolveGlobalSecurity(security$);
+
+        const request$ = this.createRequest$(
+            context,
+            {
+                security: securitySettings$,
+                method: "POST",
+                path: path$,
+                headers: headers$,
+                query: query$,
+                body: body$,
+                timeoutMs: options?.timeoutMs || this.options$.timeoutMs || -1,
+            },
+            options
+        );
+
+        const response = await this.do$(request$, {
+            context,
+            errorCodes: [],
+            retryConfig: options?.retries || this.options$.retryConfig,
+            retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+        });
+
+        const responseFields$ = {
+            ContentType: response.headers.get("content-type") ?? "application/octet-stream",
+            StatusCode: response.status,
+            RawResponse: response,
+            Headers: {},
+        };
+
+        const [result$] = await this.matcher<operations.CreateUniversalUpscalerJobResponse>()
+            .json(200, operations.CreateUniversalUpscalerJobResponse$inboundSchema, {
+                key: "object",
+            })
+            .match(response, { extraFields: responseFields$ });
+
+        return result$;
+    }
+
     /**
      * Create no background
      *
@@ -45,25 +125,27 @@ export class Variation extends ClientSDK {
      * This endpoint will create a no background variation of the provided image ID
      */
     async createVariationNoBG(
-        input: operations.CreateVariationNoBGRequestBody,
+        request: operations.CreateVariationNoBGRequestBody,
         options?: RequestOptions
     ): Promise<operations.CreateVariationNoBGResponse> {
-        const headers$ = new Headers();
-        headers$.set("user-agent", SDK_METADATA.userAgent);
-        headers$.set("Content-Type", "application/json");
-        headers$.set("Accept", "application/json");
+        const input$ = request;
 
         const payload$ = schemas$.parse(
-            input,
-            (value$) => operations.CreateVariationNoBGRequestBody$.outboundSchema.parse(value$),
+            input$,
+            (value$) => operations.CreateVariationNoBGRequestBody$outboundSchema.parse(value$),
             "Input validation failed"
         );
-        const body$ = enc$.encodeJSON("body", payload$, { explode: true });
+        const body$ = encodeJSON$("body", payload$, { explode: true });
 
         const path$ = this.templateURLComponent("/variations/nobg")();
 
         const query$ = "";
 
+        const headers$ = new Headers({
+            "Content-Type": "application/json",
+            Accept: "application/json",
+        });
+
         let security$;
         if (typeof this.options$.bearerAuth === "function") {
             security$ = { bearerAuth: await this.options$.bearerAuth() };
@@ -79,8 +161,8 @@ export class Variation extends ClientSDK {
         };
         const securitySettings$ = this.resolveGlobalSecurity(security$);
 
-        const doOptions = { context, errorCodes: [] };
-        const request = this.createRequest$(
+        const request$ = this.createRequest$(
+            context,
             {
                 security: securitySettings$,
                 method: "POST",
@@ -88,35 +170,30 @@ export class Variation extends ClientSDK {
                 headers: headers$,
                 query: query$,
                 body: body$,
+                timeoutMs: options?.timeoutMs || this.options$.timeoutMs || -1,
             },
             options
         );
 
-        const response = await this.do$(request, doOptions);
+        const response = await this.do$(request$, {
+            context,
+            errorCodes: [],
+            retryConfig: options?.retries || this.options$.retryConfig,
+            retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+        });
 
         const responseFields$ = {
             ContentType: response.headers.get("content-type") ?? "application/octet-stream",
             StatusCode: response.status,
             RawResponse: response,
+            Headers: {},
         };
 
-        if (this.matchResponse(response, 200, "application/json")) {
-            const responseBody = await response.json();
-            const result = schemas$.parse(
-                responseBody,
-                (val$) => {
-                    return operations.CreateVariationNoBGResponse$.inboundSchema.parse({
-                        ...responseFields$,
-                        object: val$,
-                    });
-                },
-                "Response validation failed"
-            );
-            return result;
-        } else {
-            const responseBody = await response.text();
-            throw new errors.SDKError("Unexpected API response", response, responseBody);
-        }
+        const [result$] = await this.matcher<operations.CreateVariationNoBGResponse>()
+            .json(200, operations.CreateVariationNoBGResponse$inboundSchema, { key: "object" })
+            .match(response, { extraFields: responseFields$ });
+
+        return result$;
     }
 
     /**
@@ -126,29 +203,29 @@ export class Variation extends ClientSDK {
      * This endpoint will create an unzoom variation for the provided image ID
      */
     async createVariationUnzoom(
-        input: operations.CreateVariationUnzoomRequestBody | undefined,
+        request?: operations.CreateVariationUnzoomRequestBody | undefined,
         options?: RequestOptions
     ): Promise<operations.CreateVariationUnzoomResponse> {
-        const headers$ = new Headers();
-        headers$.set("user-agent", SDK_METADATA.userAgent);
-        headers$.set("Content-Type", "application/json");
-        headers$.set("Accept", "application/json");
+        const input$ = request;
 
         const payload$ = schemas$.parse(
-            input,
+            input$,
             (value$) =>
-                operations.CreateVariationUnzoomRequestBody$.outboundSchema
-                    .optional()
-                    .parse(value$),
+                operations.CreateVariationUnzoomRequestBody$outboundSchema.optional().parse(value$),
             "Input validation failed"
         );
         const body$ =
-            payload$ === undefined ? null : enc$.encodeJSON("body", payload$, { explode: true });
+            payload$ === undefined ? null : encodeJSON$("body", payload$, { explode: true });
 
         const path$ = this.templateURLComponent("/variations/unzoom")();
 
         const query$ = "";
 
+        const headers$ = new Headers({
+            "Content-Type": "application/json",
+            Accept: "application/json",
+        });
+
         let security$;
         if (typeof this.options$.bearerAuth === "function") {
             security$ = { bearerAuth: await this.options$.bearerAuth() };
@@ -164,8 +241,8 @@ export class Variation extends ClientSDK {
         };
         const securitySettings$ = this.resolveGlobalSecurity(security$);
 
-        const doOptions = { context, errorCodes: [] };
-        const request = this.createRequest$(
+        const request$ = this.createRequest$(
+            context,
             {
                 security: securitySettings$,
                 method: "POST",
@@ -173,35 +250,30 @@ export class Variation extends ClientSDK {
                 headers: headers$,
                 query: query$,
                 body: body$,
+                timeoutMs: options?.timeoutMs || this.options$.timeoutMs || -1,
             },
             options
         );
 
-        const response = await this.do$(request, doOptions);
+        const response = await this.do$(request$, {
+            context,
+            errorCodes: [],
+            retryConfig: options?.retries || this.options$.retryConfig,
+            retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+        });
 
         const responseFields$ = {
             ContentType: response.headers.get("content-type") ?? "application/octet-stream",
             StatusCode: response.status,
             RawResponse: response,
+            Headers: {},
         };
 
-        if (this.matchResponse(response, 200, "application/json")) {
-            const responseBody = await response.json();
-            const result = schemas$.parse(
-                responseBody,
-                (val$) => {
-                    return operations.CreateVariationUnzoomResponse$.inboundSchema.parse({
-                        ...responseFields$,
-                        object: val$,
-                    });
-                },
-                "Response validation failed"
-            );
-            return result;
-        } else {
-            const responseBody = await response.text();
-            throw new errors.SDKError("Unexpected API response", response, responseBody);
-        }
+        const [result$] = await this.matcher<operations.CreateVariationUnzoomResponse>()
+            .json(200, operations.CreateVariationUnzoomResponse$inboundSchema, { key: "object" })
+            .match(response, { extraFields: responseFields$ });
+
+        return result$;
     }
 
     /**
@@ -211,29 +283,31 @@ export class Variation extends ClientSDK {
      * This endpoint will create an upscale for the provided image ID
      */
     async createVariationUpscale(
-        input: operations.CreateVariationUpscaleRequestBody | undefined,
+        request?: operations.CreateVariationUpscaleRequestBody | undefined,
         options?: RequestOptions
     ): Promise<operations.CreateVariationUpscaleResponse> {
-        const headers$ = new Headers();
-        headers$.set("user-agent", SDK_METADATA.userAgent);
-        headers$.set("Content-Type", "application/json");
-        headers$.set("Accept", "application/json");
+        const input$ = request;
 
         const payload$ = schemas$.parse(
-            input,
+            input$,
             (value$) =>
-                operations.CreateVariationUpscaleRequestBody$.outboundSchema
-                    .optional()
-                    .parse(value$),
+                operations.CreateVariationUpscaleRequestBody$outboundSchema.optional().parse(
+                    value$
+                ),
             "Input validation failed"
         );
         const body$ =
-            payload$ === undefined ? null : enc$.encodeJSON("body", payload$, { explode: true });
+            payload$ === undefined ? null : encodeJSON$("body", payload$, { explode: true });
 
         const path$ = this.templateURLComponent("/variations/upscale")();
 
         const query$ = "";
 
+        const headers$ = new Headers({
+            "Content-Type": "application/json",
+            Accept: "application/json",
+        });
+
         let security$;
         if (typeof this.options$.bearerAuth === "function") {
             security$ = { bearerAuth: await this.options$.bearerAuth() };
@@ -249,8 +323,8 @@ export class Variation extends ClientSDK {
         };
         const securitySettings$ = this.resolveGlobalSecurity(security$);
 
-        const doOptions = { context, errorCodes: [] };
-        const request = this.createRequest$(
+        const request$ = this.createRequest$(
+            context,
             {
                 security: securitySettings$,
                 method: "POST",
@@ -258,35 +332,30 @@ export class Variation extends ClientSDK {
                 headers: headers$,
                 query: query$,
                 body: body$,
+                timeoutMs: options?.timeoutMs || this.options$.timeoutMs || -1,
             },
             options
         );
 
-        const response = await this.do$(request, doOptions);
+        const response = await this.do$(request$, {
+            context,
+            errorCodes: [],
+            retryConfig: options?.retries || this.options$.retryConfig,
+            retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+        });
 
         const responseFields$ = {
             ContentType: response.headers.get("content-type") ?? "application/octet-stream",
             StatusCode: response.status,
             RawResponse: response,
+            Headers: {},
         };
 
-        if (this.matchResponse(response, 200, "application/json")) {
-            const responseBody = await response.json();
-            const result = schemas$.parse(
-                responseBody,
-                (val$) => {
-                    return operations.CreateVariationUpscaleResponse$.inboundSchema.parse({
-                        ...responseFields$,
-                        object: val$,
-                    });
-                },
-                "Response validation failed"
-            );
-            return result;
-        } else {
-            const responseBody = await response.text();
-            throw new errors.SDKError("Unexpected API response", response, responseBody);
-        }
+        const [result$] = await this.matcher<operations.CreateVariationUpscaleResponse>()
+            .json(200, operations.CreateVariationUpscaleResponse$inboundSchema, { key: "object" })
+            .match(response, { extraFields: responseFields$ });
+
+        return result$;
     }
 
     /**
@@ -302,24 +371,25 @@ export class Variation extends ClientSDK {
         const input$: operations.GetVariationByIdRequest = {
             id: id,
         };
-        const headers$ = new Headers();
-        headers$.set("user-agent", SDK_METADATA.userAgent);
-        headers$.set("Accept", "application/json");
 
         const payload$ = schemas$.parse(
             input$,
-            (value$) => operations.GetVariationByIdRequest$.outboundSchema.parse(value$),
+            (value$) => operations.GetVariationByIdRequest$outboundSchema.parse(value$),
             "Input validation failed"
         );
         const body$ = null;
 
         const pathParams$ = {
-            id: enc$.encodeSimple("id", payload$.id, { explode: false, charEncoding: "percent" }),
+            id: encodeSimple$("id", payload$.id, { explode: false, charEncoding: "percent" }),
         };
         const path$ = this.templateURLComponent("/variations/{id}")(pathParams$);
 
         const query$ = "";
 
+        const headers$ = new Headers({
+            Accept: "application/json",
+        });
+
         let security$;
         if (typeof this.options$.bearerAuth === "function") {
             security$ = { bearerAuth: await this.options$.bearerAuth() };
@@ -335,8 +405,8 @@ export class Variation extends ClientSDK {
         };
         const securitySettings$ = this.resolveGlobalSecurity(security$);
 
-        const doOptions = { context, errorCodes: [] };
-        const request = this.createRequest$(
+        const request$ = this.createRequest$(
+            context,
             {
                 security: securitySettings$,
                 method: "GET",
@@ -344,34 +414,29 @@ export class Variation extends ClientSDK {
                 headers: headers$,
                 query: query$,
                 body: body$,
+                timeoutMs: options?.timeoutMs || this.options$.timeoutMs || -1,
             },
             options
         );
 
-        const response = await this.do$(request, doOptions);
+        const response = await this.do$(request$, {
+            context,
+            errorCodes: [],
+            retryConfig: options?.retries || this.options$.retryConfig,
+            retryCodes: options?.retryCodes || ["429", "500", "502", "503", "504"],
+        });
 
         const responseFields$ = {
             ContentType: response.headers.get("content-type") ?? "application/octet-stream",
             StatusCode: response.status,
             RawResponse: response,
+            Headers: {},
         };
 
-        if (this.matchResponse(response, 200, "application/json")) {
-            const responseBody = await response.json();
-            const result = schemas$.parse(
-                responseBody,
-                (val$) => {
-                    return operations.GetVariationByIdResponse$.inboundSchema.parse({
-                        ...responseFields$,
-                        object: val$,
-                    });
-                },
-                "Response validation failed"
-            );
-            return result;
-        } else {
-            const responseBody = await response.text();
-            throw new errors.SDKError("Unexpected API response", response, responseBody);
-        }
+        const [result$] = await this.matcher<operations.GetVariationByIdResponse>()
+            .json(200, operations.GetVariationByIdResponse$inboundSchema, { key: "object" })
+            .match(response, { extraFields: responseFields$ });
+
+        return result$;
     }
 }