summaryrefslogtreecommitdiffstats
path: root/src/client/vertexai.rs
diff options
context:
space:
mode:
Diffstat (limited to 'src/client/vertexai.rs')
-rw-r--r--src/client/vertexai.rs44
1 files changed, 22 insertions, 22 deletions
diff --git a/src/client/vertexai.rs b/src/client/vertexai.rs
index 0c0608e..566067f 100644
--- a/src/client/vertexai.rs
+++ b/src/client/vertexai.rs
@@ -26,28 +26,6 @@ pub struct VertexAIConfig {
pub extra: Option<ExtraConfig>,
}
-#[async_trait]
-impl Client for VertexAIClient {
- client_common_fns!();
-
- async fn send_message_inner(&self, client: &ReqwestClient, data: SendData) -> Result<String> {
- self.prepare_access_token().await?;
- let builder = self.request_builder(client, data)?;
- gemini_send_message(builder).await
- }
-
- async fn send_message_streaming_inner(
- &self,
- client: &ReqwestClient,
- handler: &mut ReplyHandler,
- data: SendData,
- ) -> Result<()> {
- self.prepare_access_token().await?;
- let builder = self.request_builder(client, data)?;
- gemini_send_message_streaming(builder, handler).await
- }
-}
-
impl VertexAIClient {
list_models_fn!(
VertexAIConfig,
@@ -100,6 +78,28 @@ impl VertexAIClient {
}
}
+#[async_trait]
+impl Client for VertexAIClient {
+ client_common_fns!();
+
+ async fn send_message_inner(&self, client: &ReqwestClient, data: SendData) -> Result<String> {
+ self.prepare_access_token().await?;
+ let builder = self.request_builder(client, data)?;
+ gemini_send_message(builder).await
+ }
+
+ async fn send_message_streaming_inner(
+ &self,
+ client: &ReqwestClient,
+ handler: &mut ReplyHandler,
+ data: SendData,
+ ) -> Result<()> {
+ self.prepare_access_token().await?;
+ let builder = self.request_builder(client, data)?;
+ gemini_send_message_streaming(builder, handler).await
+ }
+}
+
pub async fn gemini_send_message(builder: RequestBuilder) -> Result<String> {
let res = builder.send().await?;
let status = res.status();