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/gemini.rs | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) (limited to 'src/client/gemini.rs') diff --git a/src/client/gemini.rs b/src/client/gemini.rs index 0d82a48..05614ce 100644 --- a/src/client/gemini.rs +++ b/src/client/gemini.rs @@ -1,6 +1,6 @@ use super::{ - vertexai::*, Client, ExtraConfig, GeminiClient, Model, ModelData, ModelPatches, PromptAction, - PromptKind, SendData, + vertexai::*, Client, CompletionData, ExtraConfig, GeminiClient, Model, ModelData, ModelPatches, + PromptAction, PromptKind, }; use anyhow::Result; @@ -27,7 +27,11 @@ impl GeminiClient { 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()?; let func = match data.stream { -- cgit v1.2.3