Skip to content

Commit ee8d312

Browse files
authored
Add back usage.tokens
1 parent 87b7eac commit ee8d312

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

src/cohere/types/usage.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -5,9 +5,13 @@
55
import pydantic
66
from ..core.pydantic_utilities import IS_PYDANTIC_V2
77
from ..core.unchecked_base_model import UncheckedBaseModel
8+
from .usage_billed_units import UsageBilledUnits
9+
from .usage_tokens import UsageTokens
810

911

1012
class Usage(UncheckedBaseModel):
13+
billed_units: typing.Optional[UsageBilledUnits] = None
14+
tokens: typing.Optional[UsageTokens] = None
1115
cached_tokens: typing.Optional[float] = pydantic.Field(default=None)
1216
"""
1317
The number of prompt tokens that hit the inference cache.

0 commit comments

Comments
 (0)