klinsmaya/SQLBot
1
0
Fork 0
mirror of https://gh-proxy.org/https://github.com/klinsmaya/SQLBot.git synced 2026-11-10 02:07:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

perf: Optimize embedded compatibility for multi-domain verification

Browse Source
This commit is contained in:
fit2cloud-chenyw
2025-11-26 13:29:26 +08:00
committed by fit2cloud-chenyw
parent e881b3ef47
commit 0e418d2ee8
7 changed files with 29 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
backend/apps/system/crud/assistant.py
+1 -1
View File
@@ -149,7 +149,7 @@ class AssistantOutDs:
if not domain_text:
return None
if ',' in domain_text:
return self.request_origin.strip('/') if self.request_origin else domain_text.split(',')[0].strip('/') + endpoint
return (self.request_origin.strip('/') if self.request_origin else domain_text.split(',')[0].strip('/')) + endpoint
else:
return f"{domain_text}{endpoint}"
backend/apps/system/middleware/auth.py
+1 -1
View File
@@ -40,7 +40,7 @@ class TokenMiddleware(BaseHTTPMiddleware):
if validator[0]:
request.state.current_user = validator[1]
request.state.assistant = validator[2]
origin = request.headers.get("origin") or get_origin_from_referer(request)
origin = request.headers.get("X-SQLBOT-HOST-ORIGIN") or get_origin_from_referer(request)
if origin and validator[2]:
request.state.assistant.request_origin = origin
return await call_next(request)