feat: add summarization to pipeline (summarize + summary_sentences + summary_language)

This commit is contained in:
hyungi
2025-08-13 08:50:06 +09:00
parent 6346635ac1
commit 8d87b1f46b
3 changed files with 70 additions and 7 deletions

View File

@@ -63,6 +63,9 @@ class PipelineIngestRequest(BaseModel):
generate_html: bool = True
translate: bool = True
target_language: str = "ko"
summarize: bool = False
summary_sentences: int = 5
summary_language: str | None = None
@app.get("/health")
@@ -171,6 +174,9 @@ def pipeline_ingest(req: PipelineIngestRequest, _: None = Depends(require_api_ke
generate_html=req.generate_html,
translate=req.translate,
target_language=req.target_language,
summarize=req.summarize,
summary_sentences=req.summary_sentences,
summary_language=req.summary_language,
)
return {"status": "ok", "doc_id": result.doc_id, "added": result.added_chunks, "chunks": result.chunks, "html_path": result.html_path}