From 6b5d51016ad4e99f962510e34ba1a495ee079661 Mon Sep 17 00:00:00 2001 From: Gabriel Luiz Freitas Almeida Date: Mon, 15 Jan 2024 23:29:57 -0300 Subject: [PATCH] Fix LangflowApplication initialization --- src/backend/langflow/__main__.py | 2 +- src/backend/langflow/server.py | 5 ++--- 2 files changed, 3 insertions(+), 4 deletions(-) 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):