From 791b6150afd626baabc2d71acc1dc3dcb70ad4d0 Mon Sep 17 00:00:00 2001
From: sigoden <sigoden@gmail.com>
Date: Fri, 6 Sep 2024 07:09:14 +0800
Subject: feat: add config `serve_addr` & env $SERVE_ADDR for specifying serve
 addr (#839)

---
 src/serve.rs | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

(limited to 'src/serve.rs')

diff --git a/src/serve.rs b/src/serve.rs
index dc16d84..59507a4 100644
--- a/src/serve.rs
+++ b/src/serve.rs
@@ -32,7 +32,6 @@ use tokio::{
 use tokio_graceful::Shutdown;
 use tokio_stream::wrappers::UnboundedReceiverStream;
 
-const DEFAULT_ADDRESS: &str = "127.0.0.1:8000";
 const DEFAULT_MODEL_NAME: &str = "default";
 const PLAYGROUND_HTML: &[u8] = include_bytes!("../assets/playground.html");
 const ARENA_HTML: &[u8] = include_bytes!("../assets/arena.html");
@@ -50,7 +49,7 @@ pub async fn run(config: GlobalConfig, addr: Option<String>) -> Result<()> {
                 addr
             }
         }
-        None => DEFAULT_ADDRESS.to_string(),
+        None => config.read().serve_addr(),
     };
     let server = Arc::new(Server::new(&config));
     let listener = TcpListener::bind(&addr).await?;
-- 
cgit v1.2.3