SSRF Fix Updated

590fd129 · root · 1f8d08ea · 590fd129
Commit 590fd129 authored Aug 07, 2024 by root
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 2 deletions

backend/apps/images/main.py backend/apps/images/main.py +1 -2

No files found.
--- a/backend/apps/images/main.py
+++ b/backend/apps/images/main.py
@@ -150,8 +150,7 @@ async def update_engine_url(
    else:
        url = form_data.AUTOMATIC1111_BASE_URL.strip("/")
        try:
-            r = requests.head(url)
-	    r.raise_for_status() 
+            r = requests.head(url) 
            app.state.config.AUTOMATIC1111_BASE_URL = url
        except Exception as e:
            raise HTTPException(status_code=400, detail="Invalid URL provided.")