Spaces:
Runtime error
Runtime error
File size: 2,129 Bytes
099df87 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 |
from fastapi import APIRouter
from fastapi.responses import StreamingResponse
from .models import ResearchRequest, SearchRequest # relative import
from . import db, crew # relative import
router = APIRouter()
@router.post("/start-job-stream")
async def start_research_job_stream(request: ResearchRequest):
async def event_generator():
# Researcher
yield f"data: {crew.stream_event('stage', 'researcher:start')}\n\n"
researcher_notes = await crew.run_researcher_async(request.topic)
yield f"data: {crew.stream_event('stage', 'researcher:done')}\n\n"
# Analyst
yield f"data: {crew.stream_event('stage', 'analyst:start')}\n\n"
analyst_notes = await crew.run_analyst_async(researcher_notes)
yield f"data: {crew.stream_event('stage', 'analyst:done')}\n\n"
# Writer: token stream
yield f"data: {crew.stream_event('stage', 'writer:start')}\n\n"
final_accum = []
async for token in crew.run_writer_token_stream(
topic=request.topic,
researcher_notes=researcher_notes,
analyst_notes=analyst_notes,
):
final_accum.append(token)
yield f"data: {crew.stream_event('token', token)}\n\n"
full_text = "".join(final_accum)
yield f"data: {crew.stream_event('stage', 'writer:done')}\n\n"
# Save exactly what was streamed
report_id = crew.generate_report_id()
db.save_report(report_id, full_text, title=request.topic)
# Final event (build dict first to avoid f-string brace issues)
meta = {"report_id": report_id, "title": request.topic}
yield f"data: {crew.stream_event('final', meta)}\n\n"
yield "event: close\ndata: done\n\n"
# Important for SSE: ensure streaming MIME and no buffering on your proxy
return StreamingResponse(event_generator(), media_type="text/event-stream")
@router.get("/list-reports")
async def list_reports():
return db.list_reports()
@router.post("/search-reports")
async def search_reports(request: SearchRequest):
return db.search_reports(request.query)
|