Browse Source

Merge commit from fork

SSRF Fix
Timothy Jaeryang Baek 9 months ago
parent
commit
8d257ed596
1 changed files with 2 additions and 2 deletions
  1. 2 2
      backend/apps/images/main.py

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

@@ -150,10 +150,10 @@ async def update_engine_url(
     else:
     else:
         url = form_data.AUTOMATIC1111_BASE_URL.strip("/")
         url = form_data.AUTOMATIC1111_BASE_URL.strip("/")
         try:
         try:
-            r = requests.head(url)
+            r = requests.head(url) 
             app.state.config.AUTOMATIC1111_BASE_URL = url
             app.state.config.AUTOMATIC1111_BASE_URL = url
         except Exception as e:
         except Exception as e:
-            raise HTTPException(status_code=400, detail=ERROR_MESSAGES.DEFAULT(e))
+            raise HTTPException(status_code=400, detail="Invalid URL provided.")
 
 
     if form_data.COMFYUI_BASE_URL == None:
     if form_data.COMFYUI_BASE_URL == None:
         app.state.config.COMFYUI_BASE_URL = COMFYUI_BASE_URL
         app.state.config.COMFYUI_BASE_URL = COMFYUI_BASE_URL