This commit is contained in:
Timothy Jaeryang Baek 2025-09-07 02:02:21 +04:00
parent 3d6d050ad8
commit c9282135c4

View file

@ -411,6 +411,7 @@ async def get_file_process_status(
MAX_FILE_PROCESSING_DURATION = 3600 * 2
async def event_stream(file_item):
if file_item:
for _ in range(MAX_FILE_PROCESSING_DURATION):
file_item = Files.get_file_by_id(file_item.id)
if file_item:
@ -430,6 +431,8 @@ async def get_file_process_status(
break
await asyncio.sleep(0.5)
else:
yield f"data: {json.dumps({'status': 'not_found'})}\n\n"
return StreamingResponse(
event_stream(file),