@hautechai/sdk 2.8.0 → 2.9.1

package/dist/index.mjs

@@ -12901,6 +12901,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"
@@ -13639,6 +13645,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"
@@ -15011,13 +15031,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
     );
@@ -15052,7 +15094,7 @@ var getLora = () => {
       options
     );
   };
-  return { loraControllerStartV1, loraControllerListV1, loraControllerGetV1, loraControllerTrainingsWebhookV1 };
+  return { loraControllerStartFluxV1, loraControllerStartNaomiV1, loraControllerStartGeneralV1, loraControllerListV1, loraControllerGetV1, loraControllerTrainingsWebhookV1 };
 };
 
 // src/sdk/api-definitions/lora.ts
@@ -15075,7 +15117,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
@@ -15522,6 +15566,7 @@ export {
   GPTV3WebSearchToolDtoType,
   GetImageRepresentationParamsDtoType,
   GiseleVtonV1InputCategory,
+  GoogleNanoBananaV1InputOutputFormat,
   GoogleNanoBananaV1ResponseKind,
   GoogleNanoBananaV1ResponseStatus,
   GptV1ResponseKind,
@@ -15624,6 +15669,8 @@ export {
   SelfAccountDtoType,
   StackEntityKind,
   StacksControllerListStacksV1OrderBy,
+  StartNaomiTrainingParamsDtoLrScheduler,
+  StartNaomiTrainingParamsDtoMixedPrecision,
   StorageEntityKind,
   StringsTemplateV1ResponseKind,
   StringsTemplateV1ResponseStatus,