diff --git a/src/khoj/database/models/__init__.py b/src/khoj/database/models/__init__.py
index 2c5f0ada..a854dc91 100644
--- a/src/khoj/database/models/__init__.py
+++ b/src/khoj/database/models/__init__.py
@@ -23,7 +23,7 @@ logger = logging.getLogger(__name__)
 class Context(PydanticBaseModel):
     compiled: str
     file: str
-    query: str
+    query: Optional[str] = None
 
 
 class CodeContextFile(PydanticBaseModel):