AI/dify
1
0
Fork 0
mirror of https://git.mirrors.martin98.com/https://github.com/langgenius/dify.git synced 2025-08-19 16:19:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor(models, helper): update identifiers and proxy transport code

Browse Source 
- Renamed `model_identity` to `dify_model_identity` for clearer specificity
- Added `proxy` parameter to `httpx.HTTPTransport` for SSRF protection adjustments
This commit is contained in:
-LAN- 2024-10-01 13:06:30 +08:00
parent afb3e317d3
commit 5c64792793
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
api/core/file/models.py
View File

@ -34,7 +34,7 @@ class FileExtraConfig(BaseModel):
class File(BaseModel): class File(BaseModel):
model_identity: str = FILE_MODEL_IDENTITY dify_model_identity: str = FILE_MODEL_IDENTITY
id: Optional[str] = None # message file id id: Optional[str] = None # message file id
tenant_id: str tenant_id: str

4
api/core/helper/ssrf_proxy.py
View File

@ -15,8 +15,8 @@ SSRF_DEFAULT_MAX_RETRIES = int(os.getenv("SSRF_DEFAULT_MAX_RETRIES", "3"))
proxy_mounts = ( proxy_mounts = (
{ {
"http://": httpx.HTTPTransport(SSRF_PROXY_HTTP_URL), "http://": httpx.HTTPTransport(proxy=SSRF_PROXY_HTTP_URL),
"https://": httpx.HTTPTransport(SSRF_PROXY_HTTPS_URL), "https://": httpx.HTTPTransport(proxy=SSRF_PROXY_HTTPS_URL),
} }
if SSRF_PROXY_HTTP_URL and SSRF_PROXY_HTTPS_URL if SSRF_PROXY_HTTP_URL and SSRF_PROXY_HTTPS_URL
else None else None