Skip to content

Commit f4c4245

Browse files
fix: append image mime_type in v2 random api
1 parent e4b098a commit f4c4245

File tree

1 file changed

+9
-1
lines changed

1 file changed

+9
-1
lines changed

src/awsl_blob_v2.py

Lines changed: 9 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,13 @@
1111
_logger = logging.getLogger(__name__)
1212

1313

14+
def append_image_mime_type(url: str) -> str:
15+
if "mime_type=" in url:
16+
return url
17+
separator = "&" if "?" in url else "?"
18+
return f"{url}{separator}mime_type=image/jpeg"
19+
20+
1421
@router.get("/v2/list", response_model=List[BlobItem], responses={404: {"model": Message}}, tags=["AwslV2"])
1522
def awsl_v2_list(uid: Optional[str] = "", limit: Optional[int] = 10, offset: Optional[int] = 0):
1623
if limit > 1000:
@@ -28,7 +35,8 @@ def awsl_v2_list_count(uid: Optional[str] = "") -> int:
2835

2936
@router.get("/v2/random", response_model=str, tags=["AwslV2"])
3037
def awsl_v2_random(uid: Optional[str] = "") -> str:
31-
return DBClientBase.get_client().awsl_v2_random(uid)
38+
url = DBClientBase.get_client().awsl_v2_random(uid)
39+
return append_image_mime_type(url)
3240

3341

3442
@router.get("/v2/random_json", response_model=BlobItem, tags=["AwslV2"])

0 commit comments

Comments
 (0)