From 54a837784c3df7e73c7c95adfef96e68c842984b Mon Sep 17 00:00:00 2001 From: sigoden Date: Thu, 30 May 2024 17:10:02 +0800 Subject: refactor: rename `SendData` to `CompletionData` (#553) --- src/client/common.rs | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) (limited to 'src/client/common.rs') diff --git a/src/client/common.rs b/src/client/common.rs index 044ed2e..336fa78 100644 --- a/src/client/common.rs +++ b/src/client/common.rs @@ -203,7 +203,7 @@ macro_rules! impl_client_trait { async fn send_message_inner( &self, client: &reqwest::Client, - data: $crate::client::SendData, + data: $crate::client::CompletionData, ) -> anyhow::Result<$crate::client::CompletionOutput> { let builder = self.request_builder(client, data)?; $send_message(builder).await @@ -213,7 +213,7 @@ macro_rules! impl_client_trait { &self, client: &reqwest::Client, handler: &mut $crate::client::SseHandler, - data: $crate::client::SendData, + data: $crate::client::CompletionData, ) -> Result<()> { let builder = self.request_builder(client, data)?; $send_message_streaming(builder, handler).await @@ -289,7 +289,7 @@ pub trait Client: Sync + Send { } let client = self.build_client()?; - let data = input.prepare_send_data(self.model(), false)?; + let data = input.prepare_completion_data(self.model(), false)?; self.send_message_inner(&client, data) .await .with_context(|| "Failed to get answer") @@ -318,7 +318,7 @@ pub trait Client: Sync + Send { return Ok(()); } let client = self.build_client()?; - let data = input.prepare_send_data(self.model(), true)?; + let data = input.prepare_completion_data(self.model(), true)?; self.send_message_streaming_inner(&client, handler, data).await } => { handler.done()?; @@ -343,14 +343,14 @@ pub trait Client: Sync + Send { async fn send_message_inner( &self, client: &ReqwestClient, - data: SendData, + data: CompletionData, ) -> Result; async fn send_message_streaming_inner( &self, client: &ReqwestClient, handler: &mut SseHandler, - data: SendData, + data: CompletionData, ) -> Result<()>; } @@ -391,7 +391,7 @@ pub fn select_model_patch<'a>( } #[derive(Debug)] -pub struct SendData { +pub struct CompletionData { pub messages: Vec, pub temperature: Option, pub top_p: Option, -- cgit v1.2.3