NovelAI: Adjust Kayra max length on subscription tier

2b904bb1707141e3484484f2d6dacf81cf471a46

Cohee <18619528+Cohee1207@users.noreply.github.com>

public/scripts/nai-settings.js+16 -1 Viewed

     return null;
+}
+export function getKayraMaxResponseTokens() {
+    switch (novel_data?.tier) {
+        case 1:
+            return 100;
+        case 2:
+            return 100;
+        case 3:
+            return 150;
+    }
+    return maximum_output_length;
+}
 export function getNovelTier() {
     return nai_tiers[novel_data?.tier] ?? 'no_connection';
+}

         console.log(finalPrompt);
+    }
+    const adjustedMaxLength = nai_settings.model_novel.includes('kayra') ? getKayraMaxResponseTokens() : maximum_output_length;
     return {
         'input': finalPrompt,
         'model': nai_settings.model_novel,
         'use_string': true,
         'temperature': Number(nai_settings.temperature),
-        'max_length': maxLength < maximum_output_length ? maxLength : maximum_output_length,
+        'max_length': maxLength < adjustedMaxLength ? maxLength : adjustedMaxLength,
         'min_length': Number(nai_settings.min_length),
         'tail_free_sampling': Number(nai_settings.tail_free_sampling),
         'repetition_penalty': Number(nai_settings.repetition_penalty),