From fa09e75a0979cabd3318d82cc1012b03ea585d24 Mon Sep 17 00:00:00 2001
From: LucasX Ubuntu <lucas.deletang.legravier@gmail.com>
Date: Thu, 9 Apr 2026 21:17:29 +0200
Subject: [PATCH] feat: use env var for ollama url

---
 src/main.rs | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/src/main.rs b/src/main.rs
index 0b7e930..11e70de 100644
--- a/src/main.rs
+++ b/src/main.rs
@@ -7,9 +7,13 @@ use crate::providers::ollama::OllamaProvider;
 use crate::state::app_state::AppState;
 
 use axum::Router;
+use once_cell::sync::Lazy;
+use std::env;
 use std::net::SocketAddr;
 use std::sync::Arc;
 
+static OLLAMA_URL: Lazy<String> = Lazy::new(|| env::var("OLLAMA_URL").expect("OLLAMA_URL not set"));
+
 #[tokio::main]
 async fn main() {
     #[cfg(debug_assertions)]
@@ -18,7 +22,7 @@ async fn main() {
     }
 
     let state = AppState {
-        ollama: Arc::new(OllamaProvider::new("http://localhost:11434")),
+        ollama: Arc::new(OllamaProvider::new(OLLAMA_URL.as_str())),
     };
 
     let app = Router::new()