fix ner

app/dialogflow/api/v1/dialogflow.py

@@ -816,12 +816,12 @@ async def check_exist_user_info(request: Request) -> Response:
     session_info = body.get("sessionInfo", {})
     parameters = session_info.get("parameters")
     
-    is_exist_user_info = dialog_service.check_exist_user_info()
+    is_exist_user_info = await dialog_service.check_exist_user_info()
     
     user_info = {}
     
     if is_exist_user_info:
-        user_info = dialog_service.get_user_info()
+        user_info = await dialog_service.get_user_info()
 
     user_name = user_info.get("name")
     phone_number = user_info.get("phone_number")
@@ -837,7 +837,7 @@ async def check_exist_user_info(request: Request) -> Response:
     return DialogFlowResponseAPI(parameters=parameters)
 
 @router.post('/trip/extract-user-name')
-async def extract_user_name(request: Request) -> Response:
+async def get_user_name(request: Request) -> Response:
     body = await request.json()
     session_info = body.get("sessionInfo", {})
     parameters = session_info.get("parameters")
@@ -846,7 +846,7 @@ async def extract_user_name(request: Request) -> Response:
     
     ner: NER = request.app.state.ner
     
-    user_name = dialog_service.extract_user_name(text=raw_text_user_name, ner=ner)
+    user_name = await dialog_service.extract_user_name(text=raw_text_user_name, ner=ner)
     
     parameters = {
         "user_name": user_name

app/dialogflow/services/dialog_service.py

@@ -396,12 +396,12 @@ class DialogService:
         except Exception as e:
             logger.error(f"Error fetching user info: {e}")
             return None
-    
-    def extract_user_name(text: str, ner: NER):
+    @staticmethod
+    async def extract_user_name(text: str, ner: NER):
         if text is None:
             return None
         
-        user_name_pred = ner.predict(text=text, entity_tag="PERSON")
+        user_name_pred = await ner.predict(text=text, entity_tag="PERSON")
         
         if user_name_pred:
             user_name = user_name_pred[0]

app/ner/services/ner.py

@@ -19,10 +19,10 @@ class NER:
                                  tokenizer=self.tokenizer,
                                  device=settings.DEVICE)
         
-    def predict(self, text: str, entity_tag: str = None):
+    async def predict(self, text: str, entity_tag: str = None):
         if self.pipeline is None:
             raise ValueError("Model not loaded. Please call load_model() first.")
-        pred = self.pipeline(text)
+        pred = await self.pipeline(text)
         if entity_tag:
             return self.extract_entities(pred, entity_tag)
         return pred