This commit is contained in:
Jason
2023-07-24 15:24:25 +08:00
parent 135a174997
commit a56bdb077a
+3 -1
View File
@@ -1,3 +1,4 @@
import json
from typing import Iterable, List
from fastapi import FastAPI, Request, HTTPException
from fastapi.params import Depends
@@ -121,6 +122,7 @@ async def extract(question: Question, openai_key=Depends(get_api_key)):
"spans": spans,
"citation": [question.context[a:b] for (a, b) in spans],
}
yield f"data: {resp}"
resp_json = json.dumps(resp)
yield f"data: {resp_json}"
return StreamingResponse(generate(), media_type="text/event-stream")