Merge pull request #214 from ollama-webui/server-error-handling

feat: better error handling for ollama reverse proxy

backend/apps/ollama/main.py

@@ -59,6 +59,7 @@ def proxy(path):
     else:
         pass
 
+    try:
         # Make a request to the target server
         target_response = requests.request(
             method=request.method,
@@ -68,6 +69,8 @@ def proxy(path):
             stream=True,  # Enable streaming for server-sent events
         )
 
+        target_response.raise_for_status()
+
         # Proxy the target server's response to the client
         def generate():
             for chunk in target_response.iter_content(chunk_size=8192):
@@ -80,6 +83,8 @@ def proxy(path):
             response.headers[key] = value
 
         return response
+    except Exception as e:
+        return jsonify({"detail": "Server Connection Error", "message": str(e)}), 400
 
 
 if __name__ == "__main__":