diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/client/model.rs | 8 | ||||
| -rw-r--r-- | src/serve.rs | 4 |
2 files changed, 6 insertions, 6 deletions
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, |