diff --git a/src/backend/langflow/__main__.py b/src/backend/langflow/__main__.py index c1644c088..8e12a3cf9 100644 --- a/src/backend/langflow/__main__.py +++ b/src/backend/langflow/__main__.py @@ -305,7 +305,7 @@ def run_langflow(host, port, log_level, options, app): else: from langflow.server import LangflowApplication - LangflowApplication(options).run() + LangflowApplication(app, options).run() except KeyboardInterrupt: logger.info("Shutting down server") sys.exit(0) diff --git a/src/backend/langflow/server.py b/src/backend/langflow/server.py index 4f566af3e..9fe432744 100644 --- a/src/backend/langflow/server.py +++ b/src/backend/langflow/server.py @@ -2,12 +2,11 @@ from gunicorn.app.base import BaseApplication # type: ignore class LangflowApplication(BaseApplication): - def __init__(self, options=None): + def __init__(self, app, options=None): self.options = options or {} - from langflow.main import create_app self.options["worker_class"] = "uvicorn.workers.UvicornWorker" - self.application = create_app() + self.application = app super().__init__() def load_config(self):