From 056f50ece5453a439d37d9c9ea860f3247e96122 Mon Sep 17 00:00:00 2001
From: Adil Hafeez <adil.hafeez@gmail.com>
Date: Mon, 23 Jun 2025 14:12:28 -0700
Subject: [PATCH] make max_tokens i32

---
 crates/hermesllm/src/providers/openai/builder.rs | 4 ++--
 crates/hermesllm/src/providers/openai/types.rs   | 2 +-
 2 files changed, 3 insertions(+), 3 deletions(-)

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<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
     }
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<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>,