apunkt/dograh
1
0
Fork 0
mirror of https://github.com/dograh-hq/dograh.git synced 2026-06-22 08:38:13 +02:00
Code Issues Projects Releases Packages Wiki Activity

feat: add full document mode in knowledge base

Browse source
This commit is contained in:
Abhishek Kumar 2026-04-09 13:49:20 +05:30
parent c085398933
commit 87c8c5e2c8
26 changed files with 1144 additions and 351 deletions
Download patch file Download diff file Expand all files Collapse all files

5
api/schemas/knowledge_base.py
View file

@ -29,6 +29,10 @@ class ProcessDocumentRequestSchema(BaseModel):
document_uuid: str = Field(..., description="Document UUID to process")
s3_key: str = Field(..., description="S3 key of the uploaded file")
retrieval_mode: str = Field(
default="chunked",
description="Retrieval mode: 'chunked' for vector search or 'full_document' for full text retrieval",
)
class DocumentResponseSchema(BaseModel):
@ -43,6 +47,7 @@ class DocumentResponseSchema(BaseModel):
processing_status: str # pending, processing, completed, failed
processing_error: Optional[str] = None
total_chunks: int
retrieval_mode: str = "chunked"
custom_metadata: Dict[str, Any]
docling_metadata: Dict[str, Any]
source_url: Optional[str] = None