From 4db9b309803796bc5f996d0b3713344eb44207ec Mon Sep 17 00:00:00 2001 From: sigoden Date: Thu, 25 Apr 2024 10:15:54 +0800 Subject: refactor: rewrite list models of all clients (#436) --- src/client/gemini.rs | 17 +++++++++-------- 1 file changed, 9 insertions(+), 8 deletions(-) (limited to 'src/client/gemini.rs') diff --git a/src/client/gemini.rs b/src/client/gemini.rs index 22f32c5..b930d79 100644 --- a/src/client/gemini.rs +++ b/src/client/gemini.rs @@ -12,13 +12,6 @@ use serde::Deserialize; const API_BASE: &str = "https://generativelanguage.googleapis.com/v1beta/models/"; -const MODELS: [(&str, usize, &str); 3] = [ - // https://ai.google.dev/models/gemini - ("gemini-1.0-pro-latest", 30720, "text"), - ("gemini-1.0-pro-vision-latest", 12288, "text,vision"), - ("gemini-1.5-pro-latest", 1048576, "text,vision"), -]; - #[derive(Debug, Clone, Deserialize, Default)] pub struct GeminiConfig { pub name: Option, @@ -50,7 +43,15 @@ impl Client for GeminiClient { } impl GeminiClient { - list_models_fn!(GeminiConfig, &MODELS); + list_models_fn!( + GeminiConfig, + [ + // https://ai.google.dev/models/gemini + ("gemini-1.0-pro-latest", "text", 30720), + ("gemini-1.0-pro-vision-latest", "text,vision", 12288), + ("gemini-1.5-pro-latest", "text,vision", 1048576), + ] + ); config_get_fn!(api_key, get_api_key); pub const PROMPTS: [PromptType<'static>; 1] = -- cgit v1.2.3