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 = 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()