Merge pull request #521 from waychan23/main

fix: saving document from browser extension fails
This commit is contained in:
Rohan Verma 2025-12-03 14:08:17 -08:00 committed by GitHub
commit de810ecb87
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 10 additions and 2 deletions

View file

@ -71,8 +71,12 @@ async def create_documents(
"metadata": { "metadata": {
"VisitedWebPageTitle": individual_document.metadata.VisitedWebPageTitle, "VisitedWebPageTitle": individual_document.metadata.VisitedWebPageTitle,
"VisitedWebPageURL": individual_document.metadata.VisitedWebPageURL, "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( process_extension_document_task.delay(
document_dict, request.search_space_id, str(user.id) document_dict, request.search_space_id, str(user.id)

View file

@ -69,10 +69,14 @@ async def _process_extension_document(
class DocumentMetadata(BaseModel): class DocumentMetadata(BaseModel):
VisitedWebPageTitle: str VisitedWebPageTitle: str
VisitedWebPageURL: str VisitedWebPageURL: str
BrowsingSessionId: str
VisitedWebPageDateWithTimeInISOString: str
VisitedWebPageReffererURL: str
VisitedWebPageVisitDurationInMilliseconds: str
class IndividualDocument(BaseModel): class IndividualDocument(BaseModel):
metadata: DocumentMetadata metadata: DocumentMetadata
content: str pageContent: str
individual_document = IndividualDocument(**individual_document_dict) individual_document = IndividualDocument(**individual_document_dict)