feat: add summarization to pipeline (summarize + summary_sentences + summary_language)
This commit is contained in:
@@ -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}
|
||||
|
||||
|
||||
Reference in New Issue
Block a user