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 |
}
|
|
|