diff options
Diffstat (limited to 'src/config')
| -rw-r--r-- | src/config/input.rs | 10 |
1 files changed, 7 insertions, 3 deletions
diff --git a/src/config/input.rs b/src/config/input.rs index 44bbb4b..ae94799 100644 --- a/src/config/input.rs +++ b/src/config/input.rs @@ -1,7 +1,7 @@ use super::{role::Role, session::Session, GlobalConfig}; use crate::client::{ - init_client, list_models, Client, CompletionData, ImageUrl, Message, MessageContent, + init_client, list_models, ChatCompletionsData, Client, ImageUrl, Message, MessageContent, MessageContentPart, MessageRole, Model, }; use crate::function::{ToolCallResult, ToolResults}; @@ -149,7 +149,11 @@ impl Input { init_client(&self.config, Some(self.model())) } - pub fn prepare_completion_data(&self, model: &Model, stream: bool) -> Result<CompletionData> { + pub fn prepare_completion_data( + &self, + model: &Model, + stream: bool, + ) -> Result<ChatCompletionsData> { if !self.medias.is_empty() && !model.supports_vision() { bail!("The current model does not support vision."); } @@ -176,7 +180,7 @@ impl Input { }; functions = config.function.select(function_matcher); }; - Ok(CompletionData { + Ok(ChatCompletionsData { messages, temperature, top_p, |