Add nest_asyncio to fix event loop issue

This commit is contained in:
Gabriel Luiz Freitas Almeida 2024-04-01 00:13:38 -03:00
commit be9a8c20fc

View file

@ -3,6 +3,7 @@ from pathlib import Path
from typing import Optional from typing import Optional
from urllib.parse import urlencode from urllib.parse import urlencode
import nest_asyncio
import socketio # type: ignore import socketio # type: ignore
from fastapi import FastAPI, Request from fastapi import FastAPI, Request
from fastapi.middleware.cors import CORSMiddleware from fastapi.middleware.cors import CORSMiddleware
@ -20,6 +21,7 @@ from langflow.utils.logger import configure
def get_lifespan(fix_migration=False, socketio_server=None): def get_lifespan(fix_migration=False, socketio_server=None):
@asynccontextmanager @asynccontextmanager
async def lifespan(app: FastAPI): async def lifespan(app: FastAPI):
nest_asyncio.apply()
initialize_services(fix_migration=fix_migration, socketio_server=socketio_server) initialize_services(fix_migration=fix_migration, socketio_server=socketio_server)
setup_llm_caching() setup_llm_caching()
LangfuseInstance.update() LangfuseInstance.update()