Skip to content

Commit 687ef1e

Browse files
committed
fix(oci): always populate billed units from OCI usage
1 parent 9f1e924 commit 687ef1e

1 file changed

Lines changed: 4 additions & 6 deletions

File tree

src/cohere/oci_client.py

Lines changed: 4 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -337,18 +337,16 @@ def _usage_from_oci(usage_data: typing.Optional[typing.Dict[str, typing.Any]]) -
337337
input_tokens = usage_data.get("inputTokens", 0)
338338
output_tokens = usage_data.get("completionTokens", usage_data.get("outputTokens", 0))
339339

340-
usage: typing.Dict[str, typing.Any] = {
340+
return {
341341
"tokens": {
342342
"input_tokens": input_tokens,
343343
"output_tokens": output_tokens,
344-
}
345-
}
346-
if input_tokens or output_tokens:
347-
usage["billed_units"] = {
344+
},
345+
"billed_units": {
348346
"input_tokens": input_tokens,
349347
"output_tokens": output_tokens,
350348
}
351-
return usage
349+
}
352350

353351

354352
def get_event_hooks(

0 commit comments

Comments
 (0)