Fix error handling (#624)

fix error message
refuel-ai · Nov 8, 2023 · 7ac1132 · 7ac1132
1 parent 262d5d9
commit 7ac1132
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/autolabel/tasks/attribute_extraction.py b/src/autolabel/tasks/attribute_extraction.py
@@ -145,7 +145,9 @@ def parse_llm_response(
         except Exception as e:
             logger.error(f"Error parsing LLM response: {response.text}, Error: {e}")
             llm_label = self.NULL_LABEL
-            error = LabelingError(error_type=ErrorType.PARSING_ERROR, error_msg=str(e))
+            error = LabelingError(
+                error_type=ErrorType.PARSING_ERROR, error_message=str(e)
+            )
 
         # TODO(rajas): Handle output guidelines not followed error (for options case)