diff options
Diffstat (limited to 'src/client/openai.rs')
| -rw-r--r-- | src/client/openai.rs | 9 |
1 files changed, 6 insertions, 3 deletions
diff --git a/src/client/openai.rs b/src/client/openai.rs index f2fd222..c8275cc 100644 --- a/src/client/openai.rs +++ b/src/client/openai.rs @@ -1,6 +1,7 @@ use super::{ - catch_error, message::*, sse_stream, CompletionOutput, ExtraConfig, Model, ModelData, - OpenAIClient, PromptAction, PromptKind, SendData, SsMmessage, SseHandler, ToolCall, + catch_error, message::*, sse_stream, Client, CompletionOutput, ExtraConfig, Model, ModelData, + ModelPatches, OpenAIClient, PromptAction, PromptKind, SendData, SsMmessage, SseHandler, + ToolCall, }; use anyhow::{bail, Result}; @@ -18,6 +19,7 @@ pub struct OpenAIConfig { pub organization_id: Option<String>, #[serde(default)] pub models: Vec<ModelData>, + pub patches: Option<ModelPatches>, pub extra: Option<ExtraConfig>, } @@ -32,7 +34,8 @@ impl OpenAIClient { let api_key = self.get_api_key()?; let api_base = self.get_api_base().unwrap_or_else(|_| API_BASE.to_string()); - let body = openai_build_body(data, &self.model); + let mut body = openai_build_body(data, &self.model); + self.patch_request_body(&mut body); let url = format!("{api_base}/chat/completions"); |
