From 76362da42bd395d39828506ab7300c94f3b3f605 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 7 Aug 2023 09:44:45 -0300 Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A7=20chore(main.py):=20refactor=20set?= =?UTF-8?q?up=5Fapp=20function=20to=20add=20support=20for=20backend=5Fonly?= =?UTF-8?q?=20flag=20=E2=9C=A8=20feat(main.py):=20add=20backend=5Fonly=20f?= =?UTF-8?q?lag=20to=20setup=5Fapp=20function=20to=20allow=20running=20the?= =?UTF-8?q?=20app=20without=20serving=20static=20files?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/backend/langflow/main.py | 9 ++++++--- 1 file changed, 6 insertions(+), 3 deletions(-) diff --git a/src/backend/langflow/main.py b/src/backend/langflow/main.py index a2da92c93..222873275 100644 --- a/src/backend/langflow/main.py +++ b/src/backend/langflow/main.py @@ -70,16 +70,19 @@ def get_static_files_dir(): return frontend_path / "frontend" -def setup_app(static_files_dir: Optional[Path] = None) -> FastAPI: +def setup_app( + static_files_dir: Optional[Path] = None, backend_only: bool = False +) -> FastAPI: """Setup the FastAPI app.""" # get the directory of the current file if not static_files_dir: static_files_dir = get_static_files_dir() - if not static_files_dir or not static_files_dir.exists(): + if not backend_only and (not static_files_dir or not static_files_dir.exists()): raise RuntimeError(f"Static files directory {static_files_dir} does not exist.") app = create_app() - setup_static_files(app, static_files_dir) + if not backend_only: + setup_static_files(app, static_files_dir) return app