Fix type so that subclasses are dumped with all their fields (was causing issues in logging with prettify function that calls model_dump where all fields were removed because not in DataModel) (neo4j#346)

Author: stellasia

src/neo4j_graphrag/experimental/pipeline/types/orchestration.py

@@ -18,7 +18,7 @@
 import enum
 from typing import Optional
 
-from pydantic import BaseModel, Field
+from pydantic import BaseModel, Field, SerializeAsAny
 
 from neo4j_graphrag.experimental.pipeline.component import DataModel
 
@@ -40,7 +40,7 @@ def possible_next_status(self) -> list[RunStatus]:
 
 class RunResult(BaseModel):
     status: RunStatus = RunStatus.DONE
-    result: Optional[DataModel] = None
+    result: Optional[SerializeAsAny[DataModel]] = None
     timestamp: datetime.datetime = Field(
         default_factory=lambda: datetime.datetime.now(datetime.timezone.utc)
     )