Merge pull request aden-hive#179 from PatrickChen928/main

feat: add nullable_output_keys, fix: aden-hive#178

Author: PatrickChen928

core/framework/graph/executor.py

@@ -317,6 +317,7 @@ async def execute(
                             output=result.output,
                             expected_keys=node_spec.output_keys,
                             check_hallucination=True,
+                            nullable_keys=node_spec.nullable_output_keys,
                         )
                         if not validation.success:
                             self.logger.error(f"   ✗ Output validation failed: {validation.error}")

core/framework/graph/node.py

@@ -163,6 +163,10 @@ class NodeSpec(BaseModel):
     output_keys: list[str] = Field(
         default_factory=list, description="Keys this node writes to shared memory or output"
     )
+    nullable_output_keys: list[str] = Field(
+        default_factory=list,
+        description="Output keys that can be None without triggering validation errors"
+    )
 
     # Optional schemas for validation and cleansing
     input_schema: dict[str, dict] = Field(

core/framework/graph/validator.py

@@ -102,6 +102,7 @@ def validate_output_keys(
         output: dict[str, Any],
         expected_keys: list[str],
         allow_empty: bool = False,
+        nullable_keys: list[str] | None = None,
     ) -> ValidationResult:
         """
         Validate that all expected keys are present and non-empty.
@@ -110,11 +111,13 @@ def validate_output_keys(
             output: The output dict to validate
             expected_keys: Keys that must be present
             allow_empty: If True, allow empty string values
+            nullable_keys: Keys that are allowed to be None
 
         Returns:
             ValidationResult with success status and any errors
         """
         errors = []
+        nullable_keys = nullable_keys or []
 
         if not isinstance(output, dict):
             return ValidationResult(
@@ -127,7 +130,8 @@ def validate_output_keys(
             elif not allow_empty:
                 value = output[key]
                 if value is None:
-                    errors.append(f"Output key '{key}' is None")
+                    if key not in nullable_keys:
+                        errors.append(f"Output key '{key}' is None")
                 elif isinstance(value, str) and len(value.strip()) == 0:
                     errors.append(f"Output key '{key}' is empty string")
 
@@ -273,6 +277,7 @@ def validate_all(
         expected_keys: list[str] | None = None,
         schema: dict[str, Any] | None = None,
         check_hallucination: bool = True,
+        nullable_keys: list[str] | None = None,
     ) -> ValidationResult:
         """
         Run all applicable validations on output.
@@ -282,6 +287,7 @@ def validate_all(
             expected_keys: Optional list of required keys
             schema: Optional JSON schema
             check_hallucination: Whether to check for hallucination patterns
+            nullable_keys: Keys that are allowed to be None
 
         Returns:
             Combined ValidationResult
@@ -290,7 +296,7 @@ def validate_all(
 
         # Validate keys if provided
         if expected_keys:
-            result = self.validate_output_keys(output, expected_keys)
+            result = self.validate_output_keys(output, expected_keys, nullable_keys=nullable_keys)
             all_errors.extend(result.errors)
 
         # Validate schema if provided