jyapayne/langflow
1
0
Fork 0
Code Issues Pull requests Projects Releases 1 Packages Wiki Activity Actions

Add custom Uvicorn worker for Langflow application

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-04-01 16:07:50 -03:00
commit 23e08f1f5c
1 changed files with 6 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

7
src/backend/base/langflow/server.py
View file

@ -1,11 +1,16 @@
from gunicorn.app.base import BaseApplication # type: ignore from gunicorn.app.base import BaseApplication # type: ignore
from uvicorn.workers import UvicornWorker
class LangflowUvicornWorker(UvicornWorker):
CONFIG_KWARGS = {"loop": "asyncio"}
class LangflowApplication(BaseApplication): class LangflowApplication(BaseApplication):
def __init__(self, app, options=None): def __init__(self, app, options=None):
self.options = options or {} self.options = options or {}
self.options["worker_class"] = "uvicorn.workers.UvicornWorker" self.options["worker_class"] = "langflow.server.LangflowUvicornWorker"
self.application = app self.application = app
super().__init__() super().__init__()