From 5e4210980d6ed92c3850042c7b57ca7eef028be0 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Thu, 15 Feb 2024 08:10:30 +0800
Subject: feat: support vertexai (#308)

---
 src/client/gemini.rs | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

(limited to 'src/client/gemini.rs')

diff --git a/src/client/gemini.rs b/src/client/gemini.rs
index 6f3141c..e846ab1 100644
--- a/src/client/gemini.rs
+++ b/src/client/gemini.rs
@@ -89,7 +89,7 @@ impl GeminiClient {
     }
 }
 
-async fn send_message(builder: RequestBuilder) -> Result<String> {
+pub(crate) async fn send_message(builder: RequestBuilder) -> Result<String> {
     let res = builder.send().await?;
     let status = res.status();
     let data: Value = res.json().await?;
@@ -102,7 +102,7 @@ async fn send_message(builder: RequestBuilder) -> Result<String> {
     Ok(output.to_string())
 }
 
-async fn send_message_streaming(builder: RequestBuilder, handler: &mut ReplyHandler) -> Result<()> {
+pub(crate) async fn send_message_streaming(builder: RequestBuilder, handler: &mut ReplyHandler) -> Result<()> {
     let res = builder.send().await?;
     if res.status() != 200 {
         let data: Value = res.json().await?;
@@ -178,7 +178,7 @@ fn check_error(data: &Value) -> Result<()> {
     }
 }
 
-fn build_body(data: SendData, _model: String) -> Result<Value> {
+pub(crate) fn build_body(data: SendData, _model: String) -> Result<Value> {
     let SendData {
         mut messages,
         temperature,
-- 
cgit v1.2.3