From a732291f3321f2feb31fb7b1105e223e2fa33976 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Wed, 5 Jun 2024 09:10:12 +0800
Subject: refactor: rename `pass_max_tokens` to `require_max_tokens` (#562)

---
 src/client/model.rs | 8 ++++----
 src/serve.rs        | 4 ++--
 2 files changed, 6 insertions(+), 6 deletions(-)

(limited to 'src')

diff --git a/src/client/model.rs b/src/client/model.rs
index e16cb4e..22ad4d9 100644
--- a/src/client/model.rs
+++ b/src/client/model.rs
@@ -153,7 +153,7 @@ impl Model {
     }
 
     pub fn max_tokens_param(&self) -> Option<isize> {
-        if self.data.pass_max_tokens {
+        if self.data.require_max_tokens {
             self.data.max_output_tokens
         } else {
             None
@@ -163,13 +163,13 @@ impl Model {
     pub fn set_max_tokens(
         &mut self,
         max_output_tokens: Option<isize>,
-        pass_max_tokens: bool,
+        require_max_tokens: bool,
     ) -> &mut Self {
         match max_output_tokens {
             None | Some(0) => self.data.max_output_tokens = None,
             _ => self.data.max_output_tokens = max_output_tokens,
         }
-        self.data.pass_max_tokens = pass_max_tokens;
+        self.data.require_max_tokens = require_max_tokens;
         self
     }
 
@@ -227,7 +227,7 @@ pub struct ModelData {
     // chat-only properties
     pub max_output_tokens: Option<isize>,
     #[serde(default)]
-    pub pass_max_tokens: bool,
+    pub require_max_tokens: bool,
     #[serde(default)]
     pub supports_vision: bool,
     #[serde(default)]
diff --git a/src/serve.rs b/src/serve.rs
index 116942a..3f33fc2 100644
--- a/src/serve.rs
+++ b/src/serve.rs
@@ -85,7 +85,7 @@ impl Server {
                 let ModelData {
                     max_input_tokens,
                     max_output_tokens,
-                    pass_max_tokens,
+                    require_max_tokens,
                     input_price,
                     output_price,
                     supports_vision,
@@ -96,7 +96,7 @@ impl Server {
                     "id": id,
                     "max_input_tokens": max_input_tokens,
                     "max_output_tokens": max_output_tokens,
-                    "pass_max_tokens": pass_max_tokens,
+                    "require_max_tokens": require_max_tokens,
                     "input_price": input_price,
                     "output_price": output_price,
                     "supports_vision": supports_vision,
-- 
cgit v1.2.3