Skip to content

Commit 664f4c7

Browse files
committed
clippy
1 parent d8aafd8 commit 664f4c7

File tree

1 file changed

+11
-15
lines changed

1 file changed

+11
-15
lines changed

golem-embed/embed-cohere/src/conversions.rs

Lines changed: 11 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -209,18 +209,12 @@ pub fn process_rerank_response(
209209
})
210210
.collect();
211211

212-
let usage = if let Some(meta) = response.clone().meta {
213-
if let Some(billed_units) = meta.billed_units {
214-
Some(Usage {
215-
input_tokens: billed_units.input_tokens,
216-
total_tokens: billed_units.output_tokens,
217-
})
218-
} else {
219-
None
220-
}
221-
} else {
222-
None
223-
};
212+
let usage = response.clone().meta.and_then(|meta| {
213+
meta.billed_units.map(|billed_units| Usage {
214+
input_tokens: billed_units.input_tokens,
215+
total_tokens: billed_units.output_tokens,
216+
})
217+
});
224218

225219
Ok(GolemRerankResponse {
226220
results,
@@ -357,14 +351,16 @@ mod tests {
357351
assert_eq!(
358352
embedding_response.embeddings[0].vector,
359353
vec![
360-
-15.0, -65.0, 0.0, -31.0, -43.0, -14.0, -48.0, 59.0, -34.0, 15.0, 36.0, 49.0, -5.0, 3.0, -49.0, -34.0, -74.0, 21.0
354+
-15.0, -65.0, 0.0, -31.0, -43.0, -14.0, -48.0, 59.0, -34.0, 15.0, 36.0, 49.0, -5.0,
355+
3.0, -49.0, -34.0, -74.0, 21.0
361356
]
362357
);
363358
assert_eq!(embedding_response.embeddings[1].index, 0);
364359
assert_eq!(
365360
embedding_response.embeddings[1].vector,
366361
vec![
367-
14.0, 38.0, -30.0, -13.0, -49.0, 4.0, -33.0, -49.0, 48.0, 9.0, -84.0, 8.0, 0.0, -84.0, -46.0, -20.0, 24.0, -26.0, -98.0, 28.0
362+
14.0, 38.0, -30.0, -13.0, -49.0, 4.0, -33.0, -49.0, 48.0, 9.0, -84.0, 8.0, 0.0,
363+
-84.0, -46.0, -20.0, 24.0, -26.0, -98.0, 28.0
368364
]
369365
);
370366
assert_eq!(
@@ -458,7 +454,7 @@ mod tests {
458454
Some(Usage {
459455
input_tokens: Some(11),
460456
total_tokens: Some(111),
461-
})
457+
})
462458
);
463459
}
464460
}

0 commit comments

Comments
 (0)