diff options
| author | sigoden <sigoden@gmail.com> | 2024-04-23 16:46:48 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-04-23 16:46:48 +0800 |
| commit | d1aafa11153ab689c21c2c57c47da52337d8e8d1 (patch) | |
| tree | dc20dc033e9d376aab09941835a842b22fe32c02 /src/client/gemini.rs | |
| parent | 1cc89eff514669d6459f62cc4eed8e0b34d8ef0c (diff) | |
| download | aichat-d1aafa11153ab689c21c2c57c47da52337d8e8d1.tar.gz | |
feat: customize model's max_output_tokens (#428)
Diffstat (limited to 'src/client/gemini.rs')
| -rw-r--r-- | src/client/gemini.rs | 4 |
1 files changed, 2 insertions, 2 deletions
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); |