diff --git a/src/neo4j_graphrag/experimental/pipeline/types/orchestration.py b/src/neo4j_graphrag/experimental/pipeline/types/orchestration.py index 1e1497ae3..ba6f99636 100644 --- a/src/neo4j_graphrag/experimental/pipeline/types/orchestration.py +++ b/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) )