diff --git a/crates/hermesllm/src/providers/openai/builder.rs b/crates/hermesllm/src/providers/openai/builder.rs index 43c4176f..98b632f6 100644 --- a/crates/hermesllm/src/providers/openai/builder.rs +++ b/crates/hermesllm/src/providers/openai/builder.rs @@ -9,7 +9,7 @@ pub struct OpenAIRequestBuilder { temperature: Option, top_p: Option, n: Option, - max_tokens: Option, + max_tokens: Option, stream: Option, stop: Option>, presence_penalty: Option, @@ -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 } diff --git a/crates/hermesllm/src/providers/openai/types.rs b/crates/hermesllm/src/providers/openai/types.rs index 6f4e38d7..c622c700 100644 --- a/crates/hermesllm/src/providers/openai/types.rs +++ b/crates/hermesllm/src/providers/openai/types.rs @@ -102,7 +102,7 @@ pub struct ChatCompletionsRequest { pub temperature: Option, pub top_p: Option, pub n: Option, - pub max_tokens: Option, + pub max_tokens: Option, pub stream: Option, pub stop: Option>, pub presence_penalty: Option,