From 7762cd6bed7f95fa855144e01aa94e932869fe12 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 8 May 2024 13:46:26 +0800
Subject: refactor: model pass_max_tokens (#493)

---
 assets/playground.html | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'assets/playground.html')

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,
       }
     }
 
-- 
cgit v1.2.3