summaryrefslogtreecommitdiffstats
path: root/src/client/gemini.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/client/gemini.rs')
-rw-r--r--src/client/gemini.rs8
1 files changed, 4 insertions, 4 deletions
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<String> {
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;