Update api/utils.py
Browse files- api/utils.py +2 -4
api/utils.py
CHANGED
@@ -203,6 +203,7 @@ async def process_non_streaming_response(request: ChatRequest):
|
|
203 |
response.raise_for_status()
|
204 |
async for chunk in response.aiter_text():
|
205 |
full_response += chunk
|
|
|
206 |
except httpx.HTTPStatusError as e:
|
207 |
logger.error(f"HTTP error occurred for Request ID {request_id}: {e}")
|
208 |
raise HTTPException(status_code=e.response.status_code, detail=str(e))
|
@@ -215,12 +216,10 @@ async def process_non_streaming_response(request: ChatRequest):
|
|
215 |
|
216 |
if BLOCKED_MESSAGE in full_response:
|
217 |
full_response = full_response.replace(BLOCKED_MESSAGE, '').strip()
|
218 |
-
if not full_response:
|
219 |
-
raise HTTPException(status_code=500, detail="Blocked message detected in response.")
|
220 |
|
221 |
cleaned_full_response = strip_model_prefix(full_response, model_prefix)
|
222 |
|
223 |
-
# Append advertisement
|
224 |
if ADVERTISEMENT_TEXT:
|
225 |
cleaned_full_response += "\n\n" + ADVERTISEMENT_TEXT
|
226 |
|
@@ -238,4 +237,3 @@ async def process_non_streaming_response(request: ChatRequest):
|
|
238 |
],
|
239 |
"usage": None,
|
240 |
}
|
241 |
-
|
|
|
203 |
response.raise_for_status()
|
204 |
async for chunk in response.aiter_text():
|
205 |
full_response += chunk
|
206 |
+
|
207 |
except httpx.HTTPStatusError as e:
|
208 |
logger.error(f"HTTP error occurred for Request ID {request_id}: {e}")
|
209 |
raise HTTPException(status_code=e.response.status_code, detail=str(e))
|
|
|
216 |
|
217 |
if BLOCKED_MESSAGE in full_response:
|
218 |
full_response = full_response.replace(BLOCKED_MESSAGE, '').strip()
|
|
|
|
|
219 |
|
220 |
cleaned_full_response = strip_model_prefix(full_response, model_prefix)
|
221 |
|
222 |
+
# Append advertisement once
|
223 |
if ADVERTISEMENT_TEXT:
|
224 |
cleaned_full_response += "\n\n" + ADVERTISEMENT_TEXT
|
225 |
|
|
|
237 |
],
|
238 |
"usage": None,
|
239 |
}
|
|