added logs for regex (#555)

Co-authored-by: ankur.jain <[email protected]>
hellohaptik · Sep 2, 2024 · d49fba4 · d49fba4
1 parent dff55ee
commit d49fba4
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/ner_v1/api.py b/ner_v1/api.py
@@ -341,6 +341,7 @@ def regex(request):
     """
     try:
         parameters_dict = parse_parameters_from_request(request)
+        ner_logger.info(f"[Regex Detection] Detection started for {parameters_dict[PARAMETER_ENTITY_NAME]}")
         entity_output = get_regex(parameters_dict[PARAMETER_MESSAGE],
                                   parameters_dict[PARAMETER_ENTITY_NAME],
                                   parameters_dict[PARAMETER_STRUCTURED_VALUE],
@@ -351,6 +352,7 @@ def regex(request):
                                   parameters_dict[PARAMETER_SOURCE_LANGUAGE]
                                   )
         ner_logger.debug('Finished %s : %s ' % (parameters_dict[PARAMETER_ENTITY_NAME], entity_output))
+        ner_logger.info(f"[Regex Detection] Detection done for {parameters_dict[PARAMETER_ENTITY_NAME]}")
     except TypeError as e:
         ner_logger.exception('Exception for regex: %s ' % e)
         return HttpResponse(status=500)

diff --git a/ner_v1/chatbot/entity_detection.py b/ner_v1/chatbot/entity_detection.py
@@ -663,14 +663,17 @@ def get_regex(message, entity_name, structured_value, fallback_value, bot_messag
 
     """
     regex_detector = RegexDetector(entity_name=entity_name, pattern=pattern, asr_enabled=is_asr, language=language)
+    ner_logger.info(f"[Regex Detection] Step 1 {entity_name}")
     if structured_value:
         entity_list, original_text_list = regex_detector.detect_entity(text=structured_value)
+        ner_logger.info(f"[Regex Detection] Step 2 structured_value {entity_name}")
         if entity_list:
             return output_entity_dict_list(entity_list, original_text_list, FROM_STRUCTURE_VALUE_VERIFIED)
         else:
             return output_entity_dict_list([structured_value], [structured_value], FROM_STRUCTURE_VALUE_NOT_VERIFIED)
     else:
         entity_list, original_text_list = regex_detector.detect_entity(text=message)
+        ner_logger.info(f"[Regex Detection] Step 2 {entity_name}")
         if entity_list:
             return output_entity_dict_list(entity_list, original_text_list, FROM_MESSAGE)
         elif fallback_value: