1 files changed, 5 insertions, 5 deletions
diff --git a/assets/playground.html b/assets/playground.html
index 485d7b6..0036fb4 100644
--- a/assets/playground.html
+++ b/assets/playground.html
@@ -939,8 +939,8 @@
               body[body_key || setting_key] = this.settings[setting_key];
             }
           });
-          const { max_output_token, need_max_tokens } = this.currentModel;
-          if (!body["max_tokens"] && need_max_tokens) {
+          const { max_output_token, pass_max_tokens } = this.currentModel;
+          if (!body["max_tokens"] && pass_max_tokens) {
             body["max_tokens"] = max_output_token;
           };
           return body;
@@ -1013,14 +1013,14 @@
     function retrieveModel(models, id) {
       const model = models.find(model => model.id === id);
       if (!model) return {};
-      const max_output_token = model.max_output_tokens || model["max_output_tokens?"] || null;
-      const need_max_tokens = !!model.max_output_tokens;
+      const max_output_token = model.max_output_tokens;
       const supports_vision = !!model.supports_vision;
+      const pass_max_tokens = !!model.pass_max_tokens;
       return {
         id,
         max_output_token,
-        need_max_tokens,
         supports_vision,
+        pass_max_tokens,
       }
     }