25 lines
452 B
Python
25 lines
452 B
Python
from fastapi import APIRouter
|
|
from fastapi.responses import StreamingResponse
|
|
from src.generators import conversation_streamer
|
|
|
|
|
|
router = APIRouter()
|
|
|
|
|
|
@router.get("")
|
|
async def hello_world():
|
|
return "Hello World!"
|
|
|
|
|
|
@router.get("/talk")
|
|
async def talk():
|
|
return StreamingResponse(
|
|
content=conversation_streamer(), media_type="text/event-stream"
|
|
)
|
|
|
|
|
|
@router.get("/json")
|
|
async def json_streamer():
|
|
# Your solution here
|
|
pass
|