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

Import multiprocessing and fix run_langflow function

Browse source
This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-01-14 15:21:30 -03:00
commit 88c75e29b2
1 changed files with 21 additions and 11 deletions
Download patch file Download diff file Expand all files Collapse all files

32
src/backend/langflow/__main__.py
View file

@ -1,3 +1,4 @@
import multiprocessing
import platform import platform
import socket import socket
import sys import sys
@ -9,18 +10,16 @@ from typing import Optional
import httpx import httpx
import typer import typer
from dotenv import load_dotenv from dotenv import load_dotenv
from multiprocess import Process, cpu_count # type: ignore from multiprocess import cpu_count # type: ignore
from rich import box from rich import box
from rich import print as rprint from rich import print as rprint
from rich.console import Console from rich.console import Console
from rich.panel import Panel from rich.panel import Panel
from rich.table import Table from rich.table import Table
from sqlmodel import select
from langflow.main import setup_app from langflow.main import setup_app
from langflow.services.database.utils import session_getter from langflow.services.deps import get_settings_service
from langflow.services.deps import get_db_service, get_settings_service from langflow.services.utils import initialize_settings_service
from langflow.services.utils import initialize_services, initialize_settings_service
from langflow.utils.logger import configure, logger from langflow.utils.logger import configure, logger
console = Console() console = Console()
@ -216,7 +215,8 @@ def run(
def run_on_mac_or_linux(host, port, log_level, options, app, open_browser=True): def run_on_mac_or_linux(host, port, log_level, options, app, open_browser=True):
webapp_process = Process(target=run_langflow, args=(host, port, log_level, options, app)) ctx = multiprocessing.get_context("spawn")
webapp_process = ctx.Process(target=run_langflow, args=(host, port, log_level, options))
webapp_process.start() webapp_process.start()
status_code = 0 status_code = 0
while status_code != 200: while status_code != 200:
@ -229,6 +229,7 @@ def run_on_mac_or_linux(host, port, log_level, options, app, open_browser=True):
print_banner(host, port) print_banner(host, port)
if open_browser: if open_browser:
webbrowser.open(f"http://{host}:{port}") webbrowser.open(f"http://{host}:{port}")
webapp_process.join()
def run_on_windows(host, port, log_level, options, app): def run_on_windows(host, port, log_level, options, app):
@ -298,24 +299,33 @@ def print_banner(host, port):
rprint(panel) rprint(panel)
def run_langflow(host, port, log_level, options, app): def run_langflow(host, port, log_level, options):
""" """
Run Langflow server on localhost Run Langflow server on localhost
""" """
try: try:
if platform.system() in ["Windows"]: if platform.system() in ["Windows", "Darwin"]:
# Run using uvicorn on MacOS and Windows # Run using uvicorn on MacOS and Windows
# Windows doesn't support gunicorn # Windows doesn't support gunicorn
# MacOS requires an env variable to be set to use gunicorn # MacOS requires an env variable to be set to use gunicorn
import uvicorn import uvicorn
uvicorn.run(app, host=host, port=port, log_level=log_level) uvicorn.run(
"langflow.main:create_app",
factory=True,
host=host,
port=port,
log_level=log_level,
workers=options["workers"],
)
else: else:
from langflow.server import LangflowApplication from langflow.server import LangflowApplication
LangflowApplication(app, options).run() LangflowApplication(options).run()
except KeyboardInterrupt: except KeyboardInterrupt:
pass logger.info("Shutting down server")
sys.exit(0)
except Exception as e: except Exception as e:
logger.exception(e) logger.exception(e)
sys.exit(1) sys.exit(1)