From 6ecbc8a5d88536a718de6370e1171d6f900dd39a Mon Sep 17 00:00:00 2001
From: Michelle Tilley <michelle@michelletilley.net>
Date: Tue, 19 May 2026 10:32:46 -0700
Subject: Allow overriding model with env var (#3493)

This PR allows overriding the Atuin AI model for feature-flagged users
with the `ATUIN_AI__MODEL` environment variable. Any OpenRouter model is
accepted; it's recommended to include an explicit `openrouter:` provider
prefix, so models with colons in their names are interpreted correctly,
e.g. `ATUIN_AI__MODEL=openrouter:deepseek/deepseek-v4-flash:free`.
---
 crates/atuin-ai/src/stream.rs | 7 +++++++
 1 file changed, 7 insertions(+)

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,
-- 
cgit v1.3.1