fix: Resolve issue with Data object serialization (#1407)
<!-- .github/pull_request_template.md --> ## Description Enable Data object serialization when returning PipelineRunInfo type objects in backend ## Type of Change <!-- Please check the relevant option --> - [ ] Bug fix (non-breaking change that fixes an issue) - [ ] New feature (non-breaking change that adds functionality) - [ ] Breaking change (fix or feature that would cause existing functionality to change) - [ ] Documentation update - [ ] Code refactoring - [ ] Performance improvement - [ ] Other (please specify): ## Changes Made <!-- List the specific changes made in this PR --> - - - ## Testing <!-- Describe how you tested your changes --> ## Screenshots/Videos (if applicable) <!-- Add screenshots or videos to help explain your changes --> ## Pre-submission Checklist <!-- Please check all boxes that apply before submitting your PR --> - [ ] **I have tested my changes thoroughly before submitting this PR** - [ ] **This PR contains minimal changes necessary to address the issue/feature** - [ ] My code follows the project's coding standards and style guidelines - [ ] I have added tests that prove my fix is effective or that my feature works - [ ] I have added necessary documentation (if applicable) - [ ] All new and existing tests pass - [ ] I have searched existing PRs to ensure this change hasn't been submitted already - [ ] I have linked any relevant issues in the description - [ ] My commits have clear and descriptive messages ## Related Issues <!-- Link any related issues using "Fixes #issue_number" or "Relates to #issue_number" --> ## Additional Notes <!-- Add any additional notes, concerns, or context for reviewers --> ## DCO Affirmation I affirm that all code in every commit of this pull request conforms to the terms of the Topoteretes Developer Certificate of Origin.
This commit is contained in:
parent
53467da224
commit
f93b7686c3
1 changed files with 7 additions and 2 deletions
|
|
@ -1,6 +1,7 @@
|
|||
from typing import Any, Optional
|
||||
from typing import Any, Optional, List, Union
|
||||
from uuid import UUID
|
||||
from pydantic import BaseModel
|
||||
from cognee.modules.data.models.Data import Data
|
||||
|
||||
|
||||
class PipelineRunInfo(BaseModel):
|
||||
|
|
@ -8,11 +9,15 @@ class PipelineRunInfo(BaseModel):
|
|||
pipeline_run_id: UUID
|
||||
dataset_id: UUID
|
||||
dataset_name: str
|
||||
payload: Optional[Any] = None
|
||||
# Data must be mentioned in typing to allow custom encoders for Data to be activated
|
||||
payload: Optional[Union[Any, List[Data]]] = None
|
||||
data_ingestion_info: Optional[list] = None
|
||||
|
||||
model_config = {
|
||||
"arbitrary_types_allowed": True,
|
||||
"from_attributes": True,
|
||||
# Add custom encoding handler for Data ORM model
|
||||
"json_encoders": {Data: lambda d: d.to_json()},
|
||||
}
|
||||
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue