diff options
| author | sigoden <sigoden@gmail.com> | 2024-05-30 17:10:02 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-05-30 17:10:02 +0800 |
| commit | 54a837784c3df7e73c7c95adfef96e68c842984b (patch) | |
| tree | e090da40436bd3c1220d5ab88071ac33397ce57c /src/client/openai.rs | |
| parent | fa4bf14e02b1e6361b38014fe6a986738cf06837 (diff) | |
| download | aichat-54a837784c3df7e73c7c95adfef96e68c842984b.tar.gz | |
refactor: rename `SendData` to `CompletionData` (#553)
Diffstat (limited to 'src/client/openai.rs')
| -rw-r--r-- | src/client/openai.rs | 16 |
1 files changed, 10 insertions, 6 deletions
diff --git a/src/client/openai.rs b/src/client/openai.rs index 881365f..bf719ae 100644 --- a/src/client/openai.rs +++ b/src/client/openai.rs @@ -1,7 +1,7 @@ use super::{ - catch_error, message::*, sse_stream, Client, CompletionOutput, ExtraConfig, Model, ModelData, - ModelPatches, OpenAIClient, PromptAction, PromptKind, SendData, SseHandler, SseMmessage, - ToolCall, + catch_error, message::*, sse_stream, Client, CompletionData, CompletionOutput, ExtraConfig, + Model, ModelData, ModelPatches, OpenAIClient, PromptAction, PromptKind, SseHandler, + SseMmessage, ToolCall, }; use anyhow::{bail, Result}; @@ -30,7 +30,11 @@ impl OpenAIClient { pub const PROMPTS: [PromptAction<'static>; 1] = [("api_key", "API Key:", true, PromptKind::String)]; - fn request_builder(&self, client: &ReqwestClient, data: SendData) -> Result<RequestBuilder> { + fn request_builder( + &self, + client: &ReqwestClient, + data: CompletionData, + ) -> Result<RequestBuilder> { let api_key = self.get_api_key()?; let api_base = self.get_api_base().unwrap_or_else(|_| API_BASE.to_string()); @@ -121,8 +125,8 @@ pub async fn openai_send_message_streaming( sse_stream(builder, handle).await } -pub fn openai_build_body(data: SendData, model: &Model) -> Value { - let SendData { +pub fn openai_build_body(data: CompletionData, model: &Model) -> Value { + let CompletionData { messages, temperature, top_p, |
