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

- Renamed `model_identity` to `dify_model_identity` for clearer specificity - Added `proxy` parameter to `httpx.HTTPTransport` for SSRF protection adjustments
langgenius · Oct 1, 2024 · 22f7744 · 22f7744
1 parent 1e62ad2
commit 22f7744
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/api/core/file/models.py b/api/core/file/models.py
@@ -34,7 +34,7 @@ class FileExtraConfig(BaseModel):
 
 
 class File(BaseModel):
- model_identity: str = FILE_MODEL_IDENTITY
+ dify_model_identity: str = FILE_MODEL_IDENTITY
 
  id: Optional[str] = None # message file id
  tenant_id: str

diff --git a/api/core/helper/ssrf_proxy.py b/api/core/helper/ssrf_proxy.py
@@ -15,8 +15,8 @@
 
 proxy_mounts = (
  {
- "http://": httpx.HTTPTransport(SSRF_PROXY_HTTP_URL),
- "https://": httpx.HTTPTransport(SSRF_PROXY_HTTPS_URL),
+ "http://": httpx.HTTPTransport(proxy=SSRF_PROXY_HTTP_URL),
+ "https://": httpx.HTTPTransport(proxy=SSRF_PROXY_HTTPS_URL),
  }
  if SSRF_PROXY_HTTP_URL and SSRF_PROXY_HTTPS_URL
  else None