gmakstutis/cognee
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Replace update_forward_refs with model_rebuild calls

Browse source
This commit is contained in:
Leon Luithlen 2024-11-04 15:53:40 +01:00
parent 287427a240
commit 79642cfca1
2 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
cognee/shared/GithubTopology.py
View file

@ -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

2
cognee/tasks/infer_data_ontology/models/models.py
View file

@ -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):