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

fix: if on macOS use uvicorn

Browse source
This commit is contained in:
Gabriel Almeida 2023-03-14 10:48:18 -03:00
commit f9a049a84d
1 changed files with 13 additions and 9 deletions
Download patch file Download diff file Expand all files Collapse all files

22
langflow/backend/langflow_backend/__main__.py
View file

@ -8,16 +8,10 @@ import typer
from fastapi.staticfiles import StaticFiles from fastapi.staticfiles import StaticFiles
from pathlib import Path from pathlib import Path
from langflow_backend.server import LangflowApplication
def get_number_of_workers(workers=None): def get_number_of_workers(workers=None):
if workers is None: if workers is None:
workers = ( workers = 1
int(os.environ.get("OBJC_DISABLE_INITIALIZE_FORK_SAFETY", "0") == "YES")
if platform.system() == "Darwin"
else 1
)
elif workers == -1: elif workers == -1:
workers = (multiprocessing.cpu_count() * 2) + 1 workers = (multiprocessing.cpu_count() * 2) + 1
return workers return workers
@ -40,14 +34,24 @@ def serve(
if not timeout: if not timeout:
timeout = 60 timeout = 60
host = "127.0.0.1"
port = 5003
options = { options = {
"bind": "127.0.0.1:5003", "bind": f"{host}:{port}",
"workers": get_number_of_workers(workers), "workers": get_number_of_workers(workers),
"worker_class": "uvicorn.workers.UvicornWorker", "worker_class": "uvicorn.workers.UvicornWorker",
"timeout": timeout, "timeout": timeout,
} }
LangflowApplication(app, options).run() if platform.system() == "Darwin":
# Run using uvicorn on MacOS
import uvicorn
uvicorn.run(app, host=host, port=port, log_level="info")
else:
from langflow_backend.server import LangflowApplication
LangflowApplication(app, options).run()
def main(): def main():