From 081080233a68b8b3dd7d77d795c084a23229d809 Mon Sep 17 00:00:00 2001
From: WayChan <waychan23@hotmail.com>
Date: Wed, 3 Dec 2025 15:32:32 +0000
Subject: [PATCH] fix: saving document from browser extension fails due to
 missing and mismatch fields of backend data models

---
 surfsense_backend/app/routes/documents_routes.py           | 6 +++++-
 surfsense_backend/app/tasks/celery_tasks/document_tasks.py | 6 +++++-
 2 files changed, 10 insertions(+), 2 deletions(-)

diff --git a/surfsense_backend/app/routes/documents_routes.py b/surfsense_backend/app/routes/documents_routes.py
index 67015243f..1d2a3ca78 100644
--- a/surfsense_backend/app/routes/documents_routes.py
+++ b/surfsense_backend/app/routes/documents_routes.py
@@ -71,8 +71,12 @@ async def create_documents(
                     "metadata": {
                         "VisitedWebPageTitle": individual_document.metadata.VisitedWebPageTitle,
                         "VisitedWebPageURL": individual_document.metadata.VisitedWebPageURL,
+                        "BrowsingSessionId": individual_document.metadata.BrowsingSessionId,
+                        "VisitedWebPageDateWithTimeInISOString": individual_document.metadata.VisitedWebPageDateWithTimeInISOString,
+                        "VisitedWebPageVisitDurationInMilliseconds": individual_document.metadata.VisitedWebPageVisitDurationInMilliseconds,
+                        "VisitedWebPageReffererURL": individual_document.metadata.VisitedWebPageReffererURL,
                     },
-                    "content": individual_document.content,
+                    "pageContent": individual_document.content,
                 }
                 process_extension_document_task.delay(
                     document_dict, request.search_space_id, str(user.id)
diff --git a/surfsense_backend/app/tasks/celery_tasks/document_tasks.py b/surfsense_backend/app/tasks/celery_tasks/document_tasks.py
index 5cf5a662a..a7b750673 100644
--- a/surfsense_backend/app/tasks/celery_tasks/document_tasks.py
+++ b/surfsense_backend/app/tasks/celery_tasks/document_tasks.py
@@ -69,10 +69,14 @@ async def _process_extension_document(
     class DocumentMetadata(BaseModel):
         VisitedWebPageTitle: str
         VisitedWebPageURL: str
+        BrowsingSessionId: str
+        VisitedWebPageDateWithTimeInISOString: str
+        VisitedWebPageReffererURL: str
+        VisitedWebPageVisitDurationInMilliseconds: str
 
     class IndividualDocument(BaseModel):
         metadata: DocumentMetadata
-        content: str
+        pageContent: str
 
     individual_document = IndividualDocument(**individual_document_dict)