adds exeptions to rasa adapter

Author: Johannes Hötter

kern/adapter/rasa.py

@@ -3,7 +3,7 @@
 from typing import Any, List, Optional
 import pandas as pd
 import yaml
-from kern import Client
+from kern import Client, exceptions
 from collections import OrderedDict
 
 # https://stackoverflow.com/questions/8640959/how-can-i-control-what-scalar-form-pyyaml-uses-for-my-data
@@ -119,11 +119,18 @@ def build_intent_yaml(
         file_name (str, optional): name of the file you want to store the data to. Defaults to "nlu.yml".
         constant_outside (str, optional): constant to be used for outside labels in token-level tasks. Defaults to CONSTANT_OUTSIDE.
         version (str, optional): Rasa version. Defaults to "3.1".
+
+    Raises:
+        exceptions.UnknownItemError: if the item you are looking for is not found.
     """
     msg.info("Building training data for Rasa")
     msg.warn("If you haven't done so yet, please install rasa and run `rasa init`")
     df = client.get_record_export(tokenize=(tokenized_label_task is not None))
 
+    for attribute in [text_name, intent_label_task, metadata_label_task, tokenized_label_task]:
+        if attribute is not None and attribute not in df.columns:
+            raise exceptions.UnknownItemError(f"Can't find argument '{attribute}' in the existing export schema: {df.columns.tolist()}")
+
     if tokenized_label_task is not None:
         text_name_injected = f"{text_name}__injected"
         df[text_name_injected] = df.apply(

kern/exceptions.py

@@ -1,30 +1,36 @@
 # -*- coding: utf-8 -*-
 from typing import Optional
 
+class LocalError(Exception):
+    pass
+
+class UnknownItemError(LocalError):
+    pass
+
 # https://developer.mozilla.org/en-US/docs/Web/HTTP/Status#client_error_responses
-class SDKError(Exception):
+class APIError(Exception):
     def __init__(self, message: Optional[str] = None):
         if message is None:
             message = "Please check the SDK documentation at https://github.com/code-kern-ai/kern-python"
         super().__init__(message)
 
 
 # 401 Unauthorized
-class UnauthorizedError(SDKError):
+class UnauthorizedError(APIError):
     pass
 
 
 # 404 Not Found
-class NotFoundError(SDKError):
+class NotFoundError(APIError):
     pass
 
-class UnknownProjectError(SDKError):
+class UnknownProjectError(APIError):
     def __init__(self, project_id: str):
         super().__init__(message=f"Could not find project '{project_id}'. Please check your input.")
 
 
 # 500 Server Error
-class InternalServerError(SDKError):
+class InternalServerError(APIError):
     pass
 
 
@@ -48,8 +54,8 @@ def get_api_exception_class(
     status_code: int,
     error_code: Optional[str] = None,
     error_message: Optional[str] = None,
-) -> SDKError:
-    exception_or_dict = RESPONSE_CODES_API_EXCEPTION_MAP.get(status_code, SDKError)
+) -> APIError:
+    exception_or_dict = RESPONSE_CODES_API_EXCEPTION_MAP.get(status_code, APIError)
     if isinstance(exception_or_dict, dict):
         exception_class = exception_or_dict.get(error_code, exception_or_dict["*"])
     else: