Update api/utils.py

api/utils.py

@@ -85,7 +85,7 @@ async def process_streaming_response(request: ChatRequest):
         "clickedForceWebSearch": False,
         "visitFromDelta": False,
         "mobileClient": False,
-        "userSelectedModel": MODEL_MAPPING.get(request.model),
+        "userSelectedModel": MODEL_MAPPING.get(request.model, request.model),  # Added default
         "validated": validate.getHid()
     }
 
@@ -151,17 +151,25 @@ async def process_non_streaming_response(request: ChatRequest):
         "clickedForceWebSearch": False,
         "visitFromDelta": False,
         "mobileClient": False,
-        "userSelectedModel": MODEL_MAPPING.get(request.model),
+        "userSelectedModel": MODEL_MAPPING.get(request.model, request.model),  # Added default
         "validated": validate.getHid()
     }
 
     full_response = ""
     async with httpx.AsyncClient() as client:
-        async with client.stream(
-            method="POST", url=f"{BASE_URL}/api/chat", headers=headers, json=json_data
-        ) as response:
-            async for chunk in response.aiter_text():
-                full_response += chunk
+        try:
+            async with client.stream(
+                method="POST", url=f"{BASE_URL}/api/chat", headers=headers, json=json_data
+            ) as response:
+                response.raise_for_status()
+                async for chunk in response.aiter_text():
+                    full_response += chunk
+        except httpx.HTTPStatusError as e:
+            logger.error(f"HTTP error occurred: {e}")
+            raise HTTPException(status_code=e.response.status_code, detail=str(e))
+        except httpx.RequestError as e:
+            logger.error(f"Error occurred during request: {e}")
+            raise HTTPException(status_code=500, detail=str(e))
     if "https://www.blackbox.ai" in full_response:
         validate.getHid(True)
         full_response = "hid已刷新，重新对话即可"