From 8a2cf2075a148a2d9f8d8c9420187952b41897cb Mon Sep 17 00:00:00 2001
From: Leon Luithlen <leon@topoteretes.com>
Date: Fri, 15 Nov 2024 17:57:03 +0100
Subject: [PATCH] Add model_rebuild

---
 cognee/modules/storage/utils/__init__.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/cognee/modules/storage/utils/__init__.py b/cognee/modules/storage/utils/__init__.py
index 7073e6470..a399e8a82 100644
--- a/cognee/modules/storage/utils/__init__.py
+++ b/cognee/modules/storage/utils/__init__.py
@@ -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 = {}