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

Add model_rebuild

Browse source
This commit is contained in:
Leon Luithlen 2024-11-15 17:57:03 +01:00
parent a3342918d9
commit 8a2cf2075a
1 changed files with 3 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
cognee/modules/storage/utils/__init__.py
View file

@ -29,7 +29,9 @@ def copy_model(model: DataPoint, include_fields: dict = {}, exclude_fields: list
**include_fields
}
return create_model(model.__name__, **final_fields)
model = create_model(model.__name__, **final_fields)
model.model_rebuild()
return model
def get_own_properties(data_point: DataPoint):
properties = {}