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/gemini.rs | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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

diff --git a/src/client/gemini.rs b/src/client/gemini.rs
index 0c60bee..5e42cd5 100644
--- a/src/client/gemini.rs
+++ b/src/client/gemini.rs
@@ -73,9 +73,9 @@ impl GeminiClient {
 
         let block_threshold = self.config.block_threshold.clone();
 
-        let body = build_body(data, self.model.name.clone(), block_threshold)?;
+        let body = build_body(data, &self.model, block_threshold)?;
 
-        let model = self.model.name.clone();
+        let model = &self.model.name;
 
         let url = format!("{API_BASE}{}:{}?key={}", model, func, api_key);
 
-- 
cgit v1.2.3