From 79642cfca1e712ed0a8c678609ee8f2dbf196078 Mon Sep 17 00:00:00 2001 From: Leon Luithlen Date: Mon, 4 Nov 2024 15:53:40 +0100 Subject: [PATCH] Replace update_forward_refs with model_rebuild calls --- cognee/shared/GithubTopology.py | 2 +- cognee/tasks/infer_data_ontology/models/models.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/cognee/shared/GithubTopology.py b/cognee/shared/GithubTopology.py index a84aa5983..3315c72e4 100644 --- a/cognee/shared/GithubTopology.py +++ b/cognee/shared/GithubTopology.py @@ -18,7 +18,7 @@ class Directory(BaseModel): directories: List['Directory'] = [] # Allows recursive Directory Model -Directory.update_forward_refs() +Directory.model_rebuild() class RepositoryProperties(BaseModel): custom_properties: Optional[Dict[str, Any]] = None diff --git a/cognee/tasks/infer_data_ontology/models/models.py b/cognee/tasks/infer_data_ontology/models/models.py index b62bf3ac0..9c086b5c7 100644 --- a/cognee/tasks/infer_data_ontology/models/models.py +++ b/cognee/tasks/infer_data_ontology/models/models.py @@ -12,7 +12,7 @@ class NodeModel(BaseModel): default_relationship: Optional[RelationshipModel] = None children: List[Union[Dict[str, Any], "NodeModel"]] = Field(default_factory=list) -NodeModel.update_forward_refs() +NodeModel.model_rebuild() class OntologyNode(BaseModel):