fix(awel): Fix awel check empty data bug (#1311)

dbgpt/core/awel/task/base.py

@@ -33,6 +33,19 @@ def __bool__(self):
     def __str__(self):
         return f"EmptyData({self.name})"
 
+    def is_same(self, obj: Any) -> bool:
+        """Check if the object is the same as the current object.
+
+        Args:
+            obj (Any): The object to compare with.
+
+        Returns:
+            bool: True if the object is the same as the current object, False otherwise.
+        """
+        if not isinstance(obj, _EMPTY_DATA_TYPE):
+            return False
+        return self == obj
+
 
 EMPTY_DATA = _EMPTY_DATA_TYPE("EMPTY_DATA")
 SKIP_DATA = _EMPTY_DATA_TYPE("SKIP_DATA")

dbgpt/core/awel/task/task_impl.py

@@ -76,7 +76,7 @@ def __init__(self, data: Union[T, _EMPTY_DATA_TYPE] = EMPTY_DATA) -> None:
     @property
     def output(self) -> T:
         """Return the output data."""
-        if self._data == EMPTY_DATA:
+        if EMPTY_DATA.is_same(self._data):
             raise ValueError("No output data for current task output")
         return cast(T, self._data)
 
@@ -188,7 +188,7 @@ def output_stream(self) -> AsyncIterator[T]:
         Raises:
             ValueError: If the output data is empty.
         """
-        if self._data == EMPTY_DATA:
+        if EMPTY_DATA.is_same(self._data):
             raise ValueError("No output data for current task output")
         return cast(AsyncIterator[T], self._data)