@@ -209,18 +209,12 @@ pub fn process_rerank_response(
209
209
} )
210
210
. collect ( ) ;
211
211
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
+ } ) ;
224
218
225
219
Ok ( GolemRerankResponse {
226
220
results,
@@ -357,14 +351,16 @@ mod tests {
357
351
assert_eq ! (
358
352
embedding_response. embeddings[ 0 ] . vector,
359
353
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
361
356
]
362
357
) ;
363
358
assert_eq ! ( embedding_response. embeddings[ 1 ] . index, 0 ) ;
364
359
assert_eq ! (
365
360
embedding_response. embeddings[ 1 ] . vector,
366
361
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
368
364
]
369
365
) ;
370
366
assert_eq ! (
@@ -458,7 +454,7 @@ mod tests {
458
454
Some ( Usage {
459
455
input_tokens: Some ( 11 ) ,
460
456
total_tokens: Some ( 111 ) ,
461
- } )
457
+ } )
462
458
) ;
463
459
}
464
460
}
0 commit comments