only website address to be refered entity url

congbo · Aug 8, 2024 · c265587 · c265587
1 parent 9d49115
commit c265587
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/webserver/configs/settings.py b/webserver/configs/settings.py
@@ -10,8 +10,8 @@
 
 
 class Settings(BaseSettings):
-    server_host: str = "http://localhost"
     server_port: int = 20213
+    website_address: str = f"http://127.0.0.1:{server_port}"
     data: str = (
         "./output"
     )

diff --git a/webserver/main.py b/webserver/main.py
@@ -332,4 +332,4 @@ def get_latest_model(model: str):
 if __name__ == "__main__":
     import uvicorn
 
-    uvicorn.run(app, host="localhost", port=settings.server_port)
+    uvicorn.run(app, port=settings.server_port)
diff --git a/webserver/utils/refer.py b/webserver/utils/refer.py
@@ -20,7 +20,7 @@ def get_reference(text: str) -> dict:
 
 
 def generate_ref_links(data: Dict[str, Set[int]], index_id: str) -> str:
-    base_url = f"{settings.server_host}:{settings.server_port}/v1/references"
+    base_url = f"{settings.website_address}/v1/references"
     lines = []
     for key, values in data.items():
         for value in values: