From a194448369d4e48ec41df415112b5ceb1de8af90 Mon Sep 17 00:00:00 2001 From: Jason Date: Tue, 25 Jul 2023 23:53:13 +0800 Subject: [PATCH] add done --- examples/citation_with_extraction/main.py | 1 + 1 file changed, 1 insertion(+) diff --git a/examples/citation_with_extraction/main.py b/examples/citation_with_extraction/main.py index 7d01588..1c8b6d1 100644 --- a/examples/citation_with_extraction/main.py +++ b/examples/citation_with_extraction/main.py @@ -124,5 +124,6 @@ async def extract(question: Question, openai_key=Depends(get_api_key)): } resp_json = json.dumps(resp) yield f"data: {resp_json}" + yield "data: [DONE]" return StreamingResponse(generate(), media_type="text/event-stream")