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/arena.html | 10 +++++-----
 1 file changed, 5 insertions(+), 5 deletions(-)

(limited to 'assets/arena.html')

diff --git a/assets/arena.html b/assets/arena.html
index 6424542..b20553d 100644
--- a/assets/arena.html
+++ b/assets/arena.html
@@ -746,8 +746,8 @@
             messages: messages,
             stream: true,
           };
-          const { max_output_token, need_max_tokens } = retrieveModel(this.models, chat.model_id);
-          if (!body["max_tokens"] && need_max_tokens) {
+          const { max_output_token, pass_max_tokens } = retrieveModel(this.models, chat.model_id);
+          if (!body["max_tokens"] && pass_max_tokens) {
             body["max_tokens"] = max_output_token;
           };
           return body;
@@ -819,14 +819,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