fix

app/dialogflow/api/router.py

@@ -5,6 +5,8 @@ from fastapi import APIRouter
 from app.dialogflow.api.v1.dialogflow import router as dialogflow_router
 from core.conf import settings
 
-v1 = APIRouter(prefix=settings.FASTAPI_API_V1_PATH)
+from utils.life_span import lifespan
+
+v1 = APIRouter(prefix=settings.FASTAPI_API_V1_PATH, lifespan=lifespan)
 
 v1.include_router(dialogflow_router)

core/conf.py

@@ -30,6 +30,10 @@ class Settings(BaseSettings):
     # DATABASE_PORT: int
     # DATABASE_USER: str 
     # DATABASE_PASSWORD: str
+    
+    # MODEl NER
+    NER_MODEL_NAME: str = 'ner'
+    TASK_NAME: str = 'ner'
 
     # FastAPI
     FASTAPI_API_V1_PATH: str = '/api/v1'

core/registrar.py

@@ -19,6 +19,7 @@ from core.path_conf import STATIC_DIR, UPLOAD_DIR
 from utils.health_check import ensure_unique_route_names, http_limit_callback
 from utils.openapi import simplify_operation_ids
 from utils.serializers import MsgSpecJSONResponse
+from utils.life_span import lifespan
 
 
 # @asynccontextmanager
@@ -53,6 +54,7 @@ def register_app() -> FastAPI:
         redoc_url=settings.FASTAPI_REDOC_URL,
         openapi_url=settings.FASTAPI_OPENAPI_URL,
         default_response_class=MsgSpecJSONResponse,
+        lifespan=lifespan
     )
 
     register_logger()

utils/life_span.py

@@ -2,7 +2,6 @@ from contextlib import asynccontextmanager
 
 from fastapi import FastAPI
 
-
 @asynccontextmanager
 async def lifespan(app: FastAPI):
     # Load the ML model