added exception to track complete error (#1662)

digiteinfotech · Dec 18, 2024 · e31e904 · e31e904
1 parent be8816b
commit e31e904
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/kairon/events/definitions/model_testing.py b/kairon/events/definitions/model_testing.py
@@ -58,6 +58,6 @@ def execute(self, **kwargs):
                                                      nlu_result=nlu_results,
                                                      event_status=EVENT_STATUS.COMPLETED.value)
         except Exception as e:
-            logger.error(str(e))
+            logger.exception(str(e))
             ModelTestingLogProcessor.log_test_result(self.bot, self.user, exception=str(e),
                                                      event_status=EVENT_STATUS.FAIL.value)
diff --git a/kairon/nlu/classifiers/llm.py b/kairon/nlu/classifiers/llm.py
@@ -31,7 +31,11 @@
     pass
 
 @DefaultV1Recipe.register(
-    DefaultV1Recipe.ComponentType.INTENT_CLASSIFIER, is_trainable=True
+    [
+        DefaultV1Recipe.ComponentType.INTENT_CLASSIFIER,
+        DefaultV1Recipe.ComponentType.ENTITY_EXTRACTOR,
+     ],
+    is_trainable=True
 )
 class LLMClassifier(IntentClassifier, GraphComponent, EntityExtractorMixin, ABC):
     """Intent and Entity classifiers using the OpenAI Completion framework"""