From 3b5843fe2e6b4e3725bbc1b269c23115c953b2b0 Mon Sep 17 00:00:00 2001 From: sigoden Date: Thu, 11 Apr 2024 05:03:58 +0800 Subject: refactor: all clients use openai token counter (#402) --- src/client/gemini.rs | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) (limited to 'src/client/gemini.rs') 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, @@ -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() } -- cgit v1.2.3