refactor: all clients use openai token counter (#402)

1 files changed, 1 insertions, 4 deletions

diff --git a/src/client/gemini.rs b/src/client/gemini.rs
index 19e656a..bcd10e3 100644
--- a/src/client/gemini.rs
+++ b/src/client/gemini.rs
@@ -1,5 +1,5 @@
 use super::vertexai::{build_body, send_message, send_message_streaming};
-use super::{Client, ExtraConfig, GeminiClient, Model, PromptType, SendData, TokensCountFactors};
+use super::{Client, ExtraConfig, GeminiClient, Model, PromptType, SendData};
 
 use crate::{render::ReplyHandler, utils::PromptKind};
 
@@ -17,8 +17,6 @@ const MODELS: [(&str, usize, &str); 3] = [
     ("gemini-1.5-pro-latest", 1048576, "text,vision"),
 ];
 
-const TOKENS_COUNT_FACTORS: TokensCountFactors = (5, 2);
-
 #[derive(Debug, Clone, Deserialize, Default)]
 pub struct GeminiConfig {
     pub name: Option<String>,
@@ -61,7 +59,6 @@ impl GeminiClient {
                 Model::new(client_name, name)
                     .set_capabilities(capabilities.into())
                     .set_max_input_tokens(Some(max_input_tokens))
-                    .set_tokens_count_factors(TOKENS_COUNT_FACTORS)
             })
             .collect()
     }