yym68686 commited on
Commit
2a42096
·
1 Parent(s): e6c24d4
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -74,7 +74,7 @@ async def process_request(request: RequestModel, provider: Dict):
74
  if request.stream:
75
  model = provider['model'][request.model]
76
  generator = fetch_response_stream(app.state.client, url, headers, payload, engine, model)
77
- wrapped_generator = error_handling_wrapper(generator, status_code=500)
78
  return StreamingResponse(wrapped_generator, media_type="text/event-stream")
79
  else:
80
  return await fetch_response(app.state.client, url, headers, payload)
 
74
  if request.stream:
75
  model = provider['model'][request.model]
76
  generator = fetch_response_stream(app.state.client, url, headers, payload, engine, model)
77
+ wrapped_generator = await error_handling_wrapper(generator, status_code=500)
78
  return StreamingResponse(wrapped_generator, media_type="text/event-stream")
79
  else:
80
  return await fetch_response(app.state.client, url, headers, payload)