@hautechai/sdk 2.8.0 → 2.9.1

package/dist/index.js

@@ -10467,6 +10467,7 @@ __export(index_exports, {
   GPTV3WebSearchToolDtoType: () => GPTV3WebSearchToolDtoType,
   GetImageRepresentationParamsDtoType: () => GetImageRepresentationParamsDtoType,
   GiseleVtonV1InputCategory: () => GiseleVtonV1InputCategory,
+  GoogleNanoBananaV1InputOutputFormat: () => GoogleNanoBananaV1InputOutputFormat,
   GoogleNanoBananaV1ResponseKind: () => GoogleNanoBananaV1ResponseKind,
   GoogleNanoBananaV1ResponseStatus: () => GoogleNanoBananaV1ResponseStatus,
   GptV1ResponseKind: () => GptV1ResponseKind,
@@ -10569,6 +10570,8 @@ __export(index_exports, {
   SelfAccountDtoType: () => SelfAccountDtoType,
   StackEntityKind: () => StackEntityKind,
   StacksControllerListStacksV1OrderBy: () => StacksControllerListStacksV1OrderBy,
+  StartNaomiTrainingParamsDtoLrScheduler: () => StartNaomiTrainingParamsDtoLrScheduler,
+  StartNaomiTrainingParamsDtoMixedPrecision: () => StartNaomiTrainingParamsDtoMixedPrecision,
   StorageEntityKind: () => StorageEntityKind,
   StringsTemplateV1ResponseKind: () => StringsTemplateV1ResponseKind,
   StringsTemplateV1ResponseStatus: () => StringsTemplateV1ResponseStatus,
@@ -13206,6 +13209,12 @@ var GiseleVtonV1InputCategory = {
   upper_body: "upper_body"
 };
 
+// src/autogenerated/schemas/googleNanoBananaV1InputOutputFormat.ts
+var GoogleNanoBananaV1InputOutputFormat = {
+  jpg: "jpg",
+  png: "png"
+};
+
 // src/autogenerated/schemas/googleNanoBananaV1ResponseKind.ts
 var GoogleNanoBananaV1ResponseKind = {
   operation: "operation"
@@ -13944,6 +13953,20 @@ var StacksControllerListStacksV1OrderBy = {
   updatedAt_DESC: "updatedAt_DESC"
 };
 
+// src/autogenerated/schemas/startNaomiTrainingParamsDtoLrScheduler.ts
+var StartNaomiTrainingParamsDtoLrScheduler = {
+  linear: "linear",
+  cosine: "cosine",
+  polynomial: "polynomial"
+};
+
+// src/autogenerated/schemas/startNaomiTrainingParamsDtoMixedPrecision.ts
+var StartNaomiTrainingParamsDtoMixedPrecision = {
+  no: "no",
+  fp16: "fp16",
+  bf16: "bf16"
+};
+
 // src/autogenerated/schemas/storageEntityKind.ts
 var StorageEntityKind = {
   storage: "storage"
@@ -15316,13 +15339,35 @@ var usePipelinesApi = () => {
 
 // src/autogenerated/lora/lora.ts
 var getLora = () => {
-  const loraControllerStartV1 = (startLoraTrainingParamsDto, options) => {
+  const loraControllerStartFluxV1 = (startFluxTrainingParamsDto, options) => {
+    return axiosMutator(
+      {
+        url: `/v1/lora/train/flux`,
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        data: startFluxTrainingParamsDto
+      },
+      options
+    );
+  };
+  const loraControllerStartNaomiV1 = (startNaomiTrainingParamsDto, options) => {
+    return axiosMutator(
+      {
+        url: `/v1/lora/train/naomi`,
+        method: "POST",
+        headers: { "Content-Type": "application/json" },
+        data: startNaomiTrainingParamsDto
+      },
+      options
+    );
+  };
+  const loraControllerStartGeneralV1 = (startTrainingParamsDto, options) => {
     return axiosMutator(
       {
         url: `/v1/lora/train`,
         method: "POST",
         headers: { "Content-Type": "application/json" },
-        data: startLoraTrainingParamsDto
+        data: startTrainingParamsDto
       },
       options
     );
@@ -15357,7 +15402,7 @@ var getLora = () => {
       options
     );
   };
-  return { loraControllerStartV1, loraControllerListV1, loraControllerGetV1, loraControllerTrainingsWebhookV1 };
+  return { loraControllerStartFluxV1, loraControllerStartNaomiV1, loraControllerStartGeneralV1, loraControllerListV1, loraControllerGetV1, loraControllerTrainingsWebhookV1 };
 };
 
 // src/sdk/api-definitions/lora.ts
@@ -15380,7 +15425,9 @@ var waitLoraStatusChange = wrapCustomMethod(async function(lora, timeoutMs = 12e
 var useLoraApi = () => {
   const hautechApi = getLora();
   return useApi({
-    start: hautechApi.loraControllerStartV1,
+    start: hautechApi.loraControllerStartGeneralV1,
+    startFlux: hautechApi.loraControllerStartFluxV1,
+    startNaomi: hautechApi.loraControllerStartNaomiV1,
     list: hautechApi.loraControllerListV1,
     get: hautechApi.loraControllerGetV1,
     wait: waitLoraStatusChange
@@ -15828,6 +15875,7 @@ var createTokenSigner = (options) => {
   GPTV3WebSearchToolDtoType,
   GetImageRepresentationParamsDtoType,
   GiseleVtonV1InputCategory,
+  GoogleNanoBananaV1InputOutputFormat,
   GoogleNanoBananaV1ResponseKind,
   GoogleNanoBananaV1ResponseStatus,
   GptV1ResponseKind,
@@ -15930,6 +15978,8 @@ var createTokenSigner = (options) => {
   SelfAccountDtoType,
   StackEntityKind,
   StacksControllerListStacksV1OrderBy,
+  StartNaomiTrainingParamsDtoLrScheduler,
+  StartNaomiTrainingParamsDtoMixedPrecision,
   StorageEntityKind,
   StringsTemplateV1ResponseKind,
   StringsTemplateV1ResponseStatus,