diff options
Diffstat (limited to 'src/client/vertexai.rs')
| -rw-r--r-- | src/client/vertexai.rs | 44 |
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(); |
