fix bug
Browse files
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)
|