scriptoriumadmin/litellm
1
0
Fork 0
mirror of https://github.com/BerriAI/litellm.git synced 2026-09-12 23:01:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 88

fix: use bearer auth for rust image edit

Browse source
This commit is contained in:
Ishaan Jaff 2026-06-25 13:02:31 -07:00
parent a8b44db0b7
commit f3629b82c4
No known key found for this signature in database
1 changed files with 7 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

9
litellm-rust/crates/providers/src/image_edit.rs
View file

@ -140,7 +140,7 @@ pub async fn image_edit(request: ImageEditRequest<'_>) -> CoreResult<Value> {
let mut request_builder = http_client().post(&url);
if let Some(api_key) = api_key {
if !has_header(&headers, "x-api-key") && !has_header(&headers, "authorization") {
request_builder = request_builder.header("x-api-key", api_key);
request_builder = request_builder.bearer_auth(api_key);
}
}
for (key, value) in headers {
@ -284,7 +284,12 @@ mod tests {
let request = server.await.expect("server task completes");
assert!(request.starts_with("POST /v1/images/edits "), "{request}");
assert!(request.contains("x-api-key: sk-test"), "{request}");
assert!(
request
.to_ascii_lowercase()
.contains("authorization: bearer sk-test"),
"{request}"
);
assert!(request.contains("name=\"model\""), "{request}");
assert!(request.contains("qwen-image-edit"), "{request}");
assert!(request.contains("name=\"prompt\""), "{request}");