diff options
| author | sigoden <sigoden@gmail.com> | 2024-09-06 07:09:14 +0800 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2024-09-06 07:09:14 +0800 |
| commit | 791b6150afd626baabc2d71acc1dc3dcb70ad4d0 (patch) | |
| tree | 08eb8fe26088f0b0cdb2bf6a021307f3168fc293 /src/serve.rs | |
| parent | b16913fec38f604ab045d6f16ddba62481b09a6c (diff) | |
| download | aichat-791b6150afd626baabc2d71acc1dc3dcb70ad4d0.tar.gz | |
feat: add config `serve_addr` & env $SERVE_ADDR for specifying serve addr (#839)
Diffstat (limited to 'src/serve.rs')
| -rw-r--r-- | src/serve.rs | 3 |
1 files changed, 1 insertions, 2 deletions
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?; |