Merge commit from fork

SSRF Fix
This commit is contained in:
Timothy Jaeryang Baek 2024-08-08 11:47:33 +02:00 committed by GitHub
commit 8d257ed596
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -150,10 +150,10 @@ async def update_engine_url(
else:
url = form_data.AUTOMATIC1111_BASE_URL.strip("/")
try:
r = requests.head(url)
r = requests.head(url)
app.state.config.AUTOMATIC1111_BASE_URL = url
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:
app.state.config.COMFYUI_BASE_URL = COMFYUI_BASE_URL