Yuchan5386 commited on
Commit
6a8da97
·
verified ·
1 Parent(s): 0a400c9

Update api.py

Browse files
Files changed (1) hide show
  1. api.py +8 -1
api.py CHANGED
@@ -294,4 +294,11 @@ async def async_generator_wrapper(prompt: str):
294
  @app.get("/generate")
295
  async def generate(request: Request):
296
  prompt = request.query_params.get("prompt", "안녕하세요")
297
- return StreamingResponse(async_generator_wrapper(prompt), media_type="text/plain")
 
 
 
 
 
 
 
 
294
  @app.get("/generate")
295
  async def generate(request: Request):
296
  prompt = request.query_params.get("prompt", "안녕하세요")
297
+ response_text = respond(prompt)
298
+
299
+ async def stream_response():
300
+ for chunk in split_response(response_text):
301
+ yield chunk
302
+ await asyncio.sleep(0.1)
303
+
304
+ return StreamingResponse(stream_response(), media_type="text/plain")