diff --git a/cognee/tasks/summarization/models.py b/cognee/tasks/summarization/models.py
index 8cee2ade3..8420cfaa5 100644
--- a/cognee/tasks/summarization/models.py
+++ b/cognee/tasks/summarization/models.py
@@ -1,7 +1,5 @@
 
-from pydantic import BaseModel, Field
 from typing import Union
-from datetime import datetime, timezone  
 from cognee.infrastructure.engine import DataPoint
 from cognee.modules.chunking.models import DocumentChunk
 from cognee.shared.CodeGraphEntities import CodeFile, CodePart