diff --git a/main.py b/main.py index 6d167d8..6c031e8 100644 --- a/main.py +++ b/main.py @@ -111,6 +111,10 @@ async def convert_stream(stream: Stream[ChatCompletionChunk]) -> AsyncIterable[s if __name__ == "__main__": import uvicorn + import asyncio - uvicorn.run("main:app", host="127.0.0.1", port=int(os.environ.get("PORT", "8000")), + try: + uvicorn.run("main:app", host="127.0.0.1", port=int(os.environ.get("PORT", "8000")), log_level="debug" if debug else "critical", reload=debug, access_log=debug) + except (KeyboardInterrupt, asyncio.CancelledError): + pass