diff options
| author | sigoden <sigoden@gmail.com> | 2024-05-08 13:46:26 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-08 13:46:26 +0800 |
| commit | 7762cd6bed7f95fa855144e01aa94e932869fe12 (patch) | |
| tree | 66e3d9371f4db081a2ce87e763c5c717c355adfc /assets | |
| parent | 1c6c740381d0da4bcb70c6531bf93b406b94d9a6 (diff) | |
| download | aichat-7762cd6bed7f95fa855144e01aa94e932869fe12.tar.gz | |
refactor: model pass_max_tokens (#493)
Diffstat (limited to 'assets')
| -rw-r--r-- | assets/arena.html | 10 | ||||
| -rw-r--r-- | assets/playground.html | 10 |
2 files changed, 10 insertions, 10 deletions
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, } } 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, } } |