feat(docs): add document write-back publish pipeline
This commit is contained in:
@@ -7,6 +7,7 @@ Endpoints:
|
||||
- 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)
|
||||
- POST /api/doc/publish - Publish physical file version via artifact registry
|
||||
- GET /api/doc/versions/{doc_id} - List document versions
|
||||
"""
|
||||
import logging
|
||||
@@ -21,6 +22,7 @@ from services.doc_service import (
|
||||
ask_about_document,
|
||||
update_document,
|
||||
list_document_versions,
|
||||
publish_document_artifact,
|
||||
get_doc_context,
|
||||
ParsedResult,
|
||||
IngestResult,
|
||||
@@ -81,6 +83,25 @@ class UpdateDocumentRequest(BaseModel):
|
||||
user_id: Optional[str] = None
|
||||
agent_id: str = "daarwizz"
|
||||
storage_ref: Optional[str] = None
|
||||
publish_artifact: bool = False
|
||||
artifact_id: Optional[str] = None
|
||||
target_format: Optional[str] = None
|
||||
artifact_label: Optional[str] = None
|
||||
metadata: Optional[Dict[str, Any]] = None
|
||||
|
||||
|
||||
class PublishDocumentRequest(BaseModel):
|
||||
"""Request to publish document as physical artifact version."""
|
||||
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
|
||||
artifact_id: Optional[str] = None
|
||||
target_format: Optional[str] = None
|
||||
artifact_label: Optional[str] = None
|
||||
metadata: Optional[Dict[str, Any]] = None
|
||||
|
||||
|
||||
@@ -267,18 +288,29 @@ async def update_document_endpoint(request: UpdateDocumentRequest):
|
||||
user_id=request.user_id,
|
||||
agent_id=request.agent_id,
|
||||
storage_ref=request.storage_ref,
|
||||
publish_artifact=request.publish_artifact,
|
||||
artifact_id=request.artifact_id,
|
||||
target_format=request.target_format,
|
||||
artifact_label=request.artifact_label,
|
||||
metadata=request.metadata,
|
||||
)
|
||||
if not result.success:
|
||||
raise HTTPException(status_code=400, detail=result.error)
|
||||
return {
|
||||
response = {
|
||||
"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,
|
||||
"publish_error": result.publish_error,
|
||||
"artifact_id": result.artifact_id,
|
||||
"artifact_version_id": result.artifact_version_id,
|
||||
"artifact_storage_key": result.artifact_storage_key,
|
||||
"artifact_mime": result.artifact_mime,
|
||||
"artifact_download_url": result.artifact_download_url,
|
||||
}
|
||||
return response
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
@@ -286,6 +318,43 @@ async def update_document_endpoint(request: UpdateDocumentRequest):
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@router.post("/api/doc/publish")
|
||||
async def publish_document_endpoint(request: PublishDocumentRequest):
|
||||
"""
|
||||
Publish current document text as physical file artifact version.
|
||||
"""
|
||||
try:
|
||||
result = await publish_document_artifact(
|
||||
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,
|
||||
artifact_id=request.artifact_id,
|
||||
target_format=request.target_format,
|
||||
artifact_label=request.artifact_label,
|
||||
metadata=request.metadata,
|
||||
)
|
||||
if not result.success:
|
||||
raise HTTPException(status_code=400, detail=result.error)
|
||||
return {
|
||||
"ok": True,
|
||||
"artifact_id": result.artifact_id,
|
||||
"version_id": result.version_id,
|
||||
"storage_key": result.storage_key,
|
||||
"mime": result.mime,
|
||||
"file_name": result.file_name,
|
||||
"download_url": result.download_url,
|
||||
}
|
||||
except HTTPException:
|
||||
raise
|
||||
except Exception as e:
|
||||
logger.error(f"Publish 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):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user