fix: coderabbit issues

cognee/modules/pipelines/exceptions.py

@@ -1,8 +1,18 @@
 class WrongTaskOrderException(Exception):
     message: str
 
+    def __init__(self, message: str):
+        self.message = message
+        super().__init__(message)
+
 
 class TaskExecutionException(Exception):
     type: str
     message: str
     traceback: str
+
+    def __init__(self, type: str, message: str, traceback: str):
+        self.message = message
+        self.type = type
+        self.traceback = traceback
+        super().__init__(message)

cognee/modules/pipelines/operations/run_tasks.py

@@ -107,7 +107,7 @@ async def run_tasks_with_telemetry(
 
 async def run_tasks(
     tasks: list[Task],
-    dataset_id: UUID,
+    dataset_id: UUID = None,
     data: Any = None,
     pipeline_name: str = "unknown_pipeline",
     context: dict = None,

cognee/modules/pipelines/tasks/Task.py

@@ -124,11 +124,11 @@ async def run(self, *args, **kwargs):
         except Exception as error:
             import traceback
 
-            error_details = {
-                "type": type(error).__name__,
-                "message": str(error),
-                "traceback": traceback.format_exc(),
-            }
+            error_details = TaskExecutionException(
+                type=type(error).__name__,
+                message=str(error),
+                traceback=traceback.format_exc(),
+            )
 
             yield TaskExecutionErrored(
                 task=self.executable,