#1424:gunicorn can't run in windows

This commit is contained in:
XD3000/高瑞雲 2024-01-12 14:30:23 +08:00
parent cbdb8d68a3
commit 574d042655

View file

@ -343,6 +343,9 @@ def run_server(
) )
try: try:
import uvicorn import uvicorn
if os.name == "nt":
pass
else:
import gunicorn.app.base import gunicorn.app.base
except: except:
raise ImportError( raise ImportError(
@ -427,6 +430,11 @@ def run_server(
f"\033[1;34mSee all Router/Swagger docs on http://0.0.0.0:{port} \033[0m\n" f"\033[1;34mSee all Router/Swagger docs on http://0.0.0.0:{port} \033[0m\n"
) # noqa ) # noqa
from litellm.proxy.proxy_server import app
if os.name == "nt":
uvicorn.run(app, host=host, port=port) # run uvicorn
else:
import gunicorn.app.base
# Gunicorn Application Class # Gunicorn Application Class
class StandaloneApplication(gunicorn.app.base.BaseApplication): class StandaloneApplication(gunicorn.app.base.BaseApplication):
def __init__(self, app, options=None): def __init__(self, app, options=None):
@ -454,8 +462,6 @@ def run_server(
"worker_class": "uvicorn.workers.UvicornWorker", "worker_class": "uvicorn.workers.UvicornWorker",
"preload": True, # Add the preload flag "preload": True, # Add the preload flag
} }
from litellm.proxy.proxy_server import app
StandaloneApplication(app=app, options=gunicorn_options).run() # Run gunicorn StandaloneApplication(app=app, options=gunicorn_options).run() # Run gunicorn