Update main.py

main.py

@@ -88,10 +88,10 @@ async def chat(request: ChatCompletionRequest):
             yield f"data: {json.dumps(response)}\n\n"
         yield "event: done\ndata: {}\n\n"
 
-    return StreamingResponse(format_response(chat_chunks), media_type="text/event-stream")
+    return EventSourceResponse(format_response(chat_chunks), media_type="text/event-stream")
 
 @app.post("/v0/chat/completions")
-async def chat(request: ChatCompletionRequest, response_mode=None):
+async def chatV0(request: ChatCompletionRequest, response_mode=None):
     kwargs = request.dict()
     dialogue_template = DialogueTemplate(
         system=system_message, messages=kwargs['messages']