feat(docs): add versioned document update and versions APIs
This commit is contained in:
@@ -6,6 +6,8 @@ Endpoints:
|
||||
- POST /api/doc/parse - Parse a document
|
||||
- POST /api/doc/ingest - Ingest document to RAG
|
||||
- POST /api/doc/ask - Ask question about document
|
||||
- POST /api/doc/update - Update existing document text (versioned)
|
||||
- GET /api/doc/versions/{doc_id} - List document versions
|
||||
"""
|
||||
import logging
|
||||
from typing import Optional, Dict, Any
|
||||
@@ -17,9 +19,12 @@ from services.doc_service import (
|
||||
parse_document,
|
||||
ingest_document,
|
||||
ask_about_document,
|
||||
update_document,
|
||||
list_document_versions,
|
||||
get_doc_context,
|
||||
ParsedResult,
|
||||
IngestResult,
|
||||
UpdateResult,
|
||||
QAResult,
|
||||
DocContext
|
||||
)
|
||||
@@ -52,6 +57,7 @@ class IngestDocumentRequest(BaseModel):
|
||||
file_name: Optional[str] = None
|
||||
dao_id: Optional[str] = None
|
||||
user_id: Optional[str] = None
|
||||
agent_id: str = "daarwizz"
|
||||
|
||||
|
||||
class AskDocumentRequest(BaseModel):
|
||||
@@ -61,6 +67,21 @@ class AskDocumentRequest(BaseModel):
|
||||
doc_id: Optional[str] = None
|
||||
dao_id: Optional[str] = None
|
||||
user_id: Optional[str] = None
|
||||
agent_id: str = "daarwizz"
|
||||
|
||||
|
||||
class UpdateDocumentRequest(BaseModel):
|
||||
"""Request to update existing document content."""
|
||||
session_id: str
|
||||
doc_id: Optional[str] = None
|
||||
doc_url: Optional[str] = None
|
||||
file_name: Optional[str] = None
|
||||
text: Optional[str] = None
|
||||
dao_id: Optional[str] = None
|
||||
user_id: Optional[str] = None
|
||||
agent_id: str = "daarwizz"
|
||||
storage_ref: Optional[str] = None
|
||||
metadata: Optional[Dict[str, Any]] = None
|
||||
|
||||
|
||||
# ========================================
|
||||
@@ -167,7 +188,8 @@ async def ingest_document_endpoint(request: IngestDocumentRequest):
|
||||
doc_url=request.doc_url,
|
||||
file_name=request.file_name,
|
||||
dao_id=request.dao_id,
|
||||
user_id=request.user_id
|
||||
user_id=request.user_id,
|
||||
agent_id=request.agent_id,
|
||||
)
|
||||
|
||||
if not result.success:
|
||||
@@ -209,7 +231,8 @@ async def ask_about_document_endpoint(request: AskDocumentRequest):
|
||||
question=request.question,
|
||||
doc_id=doc_id,
|
||||
dao_id=request.dao_id,
|
||||
user_id=request.user_id
|
||||
user_id=request.user_id,
|
||||
agent_id=request.agent_id,
|
||||
)
|
||||
|
||||
if not result.success:
|
||||
@@ -227,6 +250,59 @@ async def ask_about_document_endpoint(request: AskDocumentRequest):
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.post("/api/doc/update")
|
||||
async def update_document_endpoint(request: UpdateDocumentRequest):
|
||||
"""
|
||||
Update a document and bump its version.
|
||||
If text is omitted and doc_url exists, text is re-parsed from the source document.
|
||||
"""
|
||||
try:
|
||||
result = await update_document(
|
||||
session_id=request.session_id,
|
||||
doc_id=request.doc_id,
|
||||
doc_url=request.doc_url,
|
||||
file_name=request.file_name,
|
||||
text=request.text,
|
||||
dao_id=request.dao_id,
|
||||
user_id=request.user_id,
|
||||
agent_id=request.agent_id,
|
||||
storage_ref=request.storage_ref,
|
||||
metadata=request.metadata,
|
||||
)
|
||||
if not result.success:
|
||||
raise HTTPException(status_code=400, detail=result.error)
|
||||
return {
|
||||
"ok": True,
|
||||
"doc_id": result.doc_id,
|
||||
"version_no": result.version_no,
|
||||
"version_id": result.version_id,
|
||||
"updated_chunks": result.updated_chunks,
|
||||
"status": result.status,
|
||||
}
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"Update document error: {e}", exc_info=True)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.get("/api/doc/versions/{doc_id}")
|
||||
async def list_document_versions_endpoint(doc_id: str, agent_id: str = "daarwizz", limit: int = 20):
|
||||
"""
|
||||
List document versions for agent/doc pair.
|
||||
"""
|
||||
try:
|
||||
data = await list_document_versions(agent_id=agent_id, doc_id=doc_id, limit=limit)
|
||||
if not data.get("ok"):
|
||||
raise HTTPException(status_code=400, detail=data.get("error", "Failed to load versions"))
|
||||
return data
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"List document versions error: {e}", exc_info=True)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.get("/api/doc/context/{session_id}")
|
||||
async def get_document_context(session_id: str):
|
||||
"""
|
||||
@@ -257,4 +333,3 @@ async def get_document_context(session_id: str):
|
||||
except Exception as e:
|
||||
logger.error(f"Get document context error: {e}", exc_info=True)
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user