From a193710a7f88de12ef11c5a5c66a5591150fe247 Mon Sep 17 00:00:00 2001 From: sigoden Date: Thu, 25 Apr 2024 14:03:16 +0800 Subject: refactor: extract common catch_error (#437) --- src/client/gemini.rs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) (limited to 'src/client/gemini.rs') diff --git a/src/client/gemini.rs b/src/client/gemini.rs index b930d79..05e19a3 100644 --- a/src/client/gemini.rs +++ b/src/client/gemini.rs @@ -1,4 +1,4 @@ -use super::vertexai::{build_body, send_message, send_message_streaming}; +use super::vertexai::{gemini_build_body, gemini_send_message, gemini_send_message_streaming}; use super::{ Client, ExtraConfig, GeminiClient, Model, ModelConfig, PromptType, ReplyHandler, SendData, }; @@ -28,7 +28,7 @@ impl Client for GeminiClient { async fn send_message_inner(&self, client: &ReqwestClient, data: SendData) -> Result { let builder = self.request_builder(client, data)?; - send_message(builder).await + gemini_send_message(builder).await } async fn send_message_streaming_inner( @@ -38,7 +38,7 @@ impl Client for GeminiClient { data: SendData, ) -> Result<()> { let builder = self.request_builder(client, data)?; - send_message_streaming(builder, handler).await + gemini_send_message_streaming(builder, handler).await } } @@ -67,7 +67,7 @@ impl GeminiClient { let block_threshold = self.config.block_threshold.clone(); - let body = build_body(data, &self.model, block_threshold)?; + let body = gemini_build_body(data, &self.model, block_threshold)?; let model = &self.model.name; -- cgit v1.2.3