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/claude.rs | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) (limited to 'src/client/claude.rs') diff --git a/src/client/claude.rs b/src/client/claude.rs index ccedb51..3194fe2 100644 --- a/src/client/claude.rs +++ b/src/client/claude.rs @@ -1,7 +1,7 @@ use super::{ catch_error, extract_system_message, message::*, sse_stream, ClaudeClient, Client, - CompletionOutput, ExtraConfig, ImageUrl, MessageContent, MessageContentPart, Model, ModelData, - ModelPatches, PromptAction, PromptKind, SendData, SseHandler, SseMmessage, ToolCall, + CompletionData, CompletionOutput, ExtraConfig, ImageUrl, MessageContent, MessageContentPart, + Model, ModelData, ModelPatches, PromptAction, PromptKind, SseHandler, SseMmessage, ToolCall, }; use anyhow::{bail, Context, Result}; @@ -27,7 +27,11 @@ impl ClaudeClient { pub const PROMPTS: [PromptAction<'static>; 1] = [("api_key", "API Key:", true, PromptKind::String)]; - fn request_builder(&self, client: &ReqwestClient, data: SendData) -> Result { + fn request_builder( + &self, + client: &ReqwestClient, + data: CompletionData, + ) -> Result { let api_key = self.get_api_key().ok(); let mut body = claude_build_body(data, &self.model)?; @@ -131,8 +135,8 @@ pub async fn claude_send_message_streaming( sse_stream(builder, handle).await } -pub fn claude_build_body(data: SendData, model: &Model) -> Result { - let SendData { +pub fn claude_build_body(data: CompletionData, model: &Model) -> Result { + let CompletionData { mut messages, temperature, top_p, -- cgit v1.2.3