From 05748bdf104c9f70b87ac14fd2d681493fd0ffa7 Mon Sep 17 00:00:00 2001
From: leehuwuj <leehuwuj@gmail.com>
Date: Mon, 27 May 2024 14:53:01 +0700
Subject: [PATCH] refactor code
---
templates/types/streaming/fastapi/main.py | 15 +++++++--------
1 file changed, 7 insertions(+), 8 deletions(-)
diff --git a/templates/types/streaming/fastapi/main.py b/templates/types/streaming/fastapi/main.py
index b095b616..9ed2037d 100644
--- a/templates/types/streaming/fastapi/main.py
+++ b/templates/types/streaming/fastapi/main.py
@@ -38,16 +38,15 @@ if environment == "dev":
return RedirectResponse(url="/docs")
+def mount_static_files(directory, path):
+ if os.path.exists(directory):
+ app.mount(path, StaticFiles(directory=directory), name=f"{directory}-static")
+
+
# Mount the data files to serve the file viewer
-if os.path.exists("data"):
- app.mount("/api/files/data", StaticFiles(directory="data"), name="data-static")
+mount_static_files("data", "/api/files/data")
# Mount the output files from tools
-if os.path.exists("tool-output"):
- app.mount(
- "/api/files/tool-output",
- StaticFiles(directory="tool-output"),
- name="tool-output-static",
- )
+mount_static_files("tool-output", "/api/files/tool-output")
app.include_router(chat_router, prefix="/api/chat")
--
GitLab