From d1aafa11153ab689c21c2c57c47da52337d8e8d1 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Tue, 23 Apr 2024 16:46:48 +0800
Subject: feat: customize model's max_output_tokens (#428)

---
 src/client/mistral.rs | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

(limited to 'src/client/mistral.rs')

diff --git a/src/client/mistral.rs b/src/client/mistral.rs
index 611c25c..4ff4787 100644
--- a/src/client/mistral.rs
+++ b/src/client/mistral.rs
@@ -51,7 +51,7 @@ impl MistralClient {
     fn request_builder(&self, client: &ReqwestClient, data: SendData) -> Result<RequestBuilder> {
         let api_key = self.get_api_key().ok();
 
-        let body = openai_build_body(data, self.model.name.clone());
+        let body = openai_build_body(data, &self.model);
 
         let url = API_URL;
 
-- 
cgit v1.2.3