A potential fix for messy error boxes by RuntimeError

Signed-off-by: dafnapension <dafnashein@yahoo.com>

Author: dafnapension

src/unitxt/error_utils.py

@@ -298,5 +298,7 @@ def error_context(context_object: Any = None, **context):
     try:
         yield
     except Exception as e:
+        if e.__class__.__name__ == "KeyError":
+            e = RuntimeError(e.__class__.__name__ + ": '" + e.args[0] + "'")
         _add_context_to_exception(e, context_object, **context)
-        raise
+        raise e

tests/library/test_error_utils.py

@@ -47,7 +47,7 @@ class TestProcessor:
 
         processor = TestProcessor()
 
-        with self.assertRaises(KeyError) as cm:
+        with self.assertRaises(RuntimeError) as cm:
             with error_context(processor):
                 raise KeyError("Missing key")
 
@@ -186,12 +186,12 @@ class TestProcessor:
 
     def test_error_context_without_object(self):
         """Test error_context without a context object."""
-        with self.assertRaises(KeyError) as cm:
+        with self.assertRaises(RuntimeError) as cm:
             with error_context(input_file="data.json", line_number=156):
                 raise KeyError("Missing field")
 
         error = cm.exception
-        self.assertIsInstance(error, KeyError)
+        self.assertIsInstance(error, RuntimeError)
         self.assertIsNone(error.context_object)
         # Context now includes version info plus the specified context
         self.assertIn("Unitxt", error.context)