diff options
Diffstat (limited to 'src/client/ernie.rs')
| -rw-r--r-- | src/client/ernie.rs | 44 |
1 files changed, 22 insertions, 22 deletions
diff --git a/src/client/ernie.rs b/src/client/ernie.rs index 22f9362..5cc546c 100644 --- a/src/client/ernie.rs +++ b/src/client/ernie.rs @@ -30,28 +30,6 @@ pub struct ErnieConfig { pub extra: Option<ExtraConfig>, } -#[async_trait] -impl Client for ErnieClient { - 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)?; - 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)?; - send_message_streaming(builder, handler).await - } -} - impl ErnieClient { list_models_fn!( ErnieConfig, @@ -118,6 +96,28 @@ impl ErnieClient { } } +#[async_trait] +impl Client for ErnieClient { + 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)?; + 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)?; + send_message_streaming(builder, handler).await + } +} + async fn send_message(builder: RequestBuilder) -> Result<String> { let data: Value = builder.send().await?.json().await?; maybe_catch_error(&data)?; |
