diff options
Diffstat (limited to 'crates/atuin-ai/src')
| -rw-r--r-- | crates/atuin-ai/src/stream.rs | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/crates/atuin-ai/src/stream.rs b/crates/atuin-ai/src/stream.rs index d921b39c..084e8238 100644 --- a/crates/atuin-ai/src/stream.rs +++ b/crates/atuin-ai/src/stream.rs @@ -137,6 +137,13 @@ pub(crate) fn create_chat_stream( } } + if let Ok(model) = std::env::var("ATUIN_AI__MODEL") + && !model.trim().is_empty() { + config["model"] = serde_json::json!(model.trim()); + + } + + let mut request_body = serde_json::json!({ "messages": request.messages, "context": context, |