mirror of
https://github.com/katanemo/plano.git
synced 2026-06-17 15:25:17 +02:00
make max_tokens i32
This commit is contained in:
parent
7baec20772
commit
056f50ece5
2 changed files with 3 additions and 3 deletions
|
|
@ -9,7 +9,7 @@ pub struct OpenAIRequestBuilder {
|
|||
temperature: Option<f32>,
|
||||
top_p: Option<f32>,
|
||||
n: Option<u32>,
|
||||
max_tokens: Option<u32>,
|
||||
max_tokens: Option<i32>,
|
||||
stream: Option<bool>,
|
||||
stop: Option<Vec<String>>,
|
||||
presence_penalty: Option<f32>,
|
||||
|
|
@ -51,7 +51,7 @@ impl OpenAIRequestBuilder {
|
|||
self
|
||||
}
|
||||
|
||||
pub fn max_tokens(mut self, max_tokens: u32) -> Self {
|
||||
pub fn max_tokens(mut self, max_tokens: i32) -> Self {
|
||||
self.max_tokens = Some(max_tokens);
|
||||
self
|
||||
}
|
||||
|
|
|
|||
|
|
@ -102,7 +102,7 @@ pub struct ChatCompletionsRequest {
|
|||
pub temperature: Option<f32>,
|
||||
pub top_p: Option<f32>,
|
||||
pub n: Option<u32>,
|
||||
pub max_tokens: Option<u32>,
|
||||
pub max_tokens: Option<i32>,
|
||||
pub stream: Option<bool>,
|
||||
pub stop: Option<Vec<String>>,
|
||||
pub presence_penalty: Option<f32>,
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue