File tree Expand file tree Collapse file tree 2 files changed +8
-4
lines changed
tests/test_litellm/llms/oci/chat Expand file tree Collapse file tree 2 files changed +8
-4
lines changed Original file line number Diff line number Diff line change @@ -121,20 +121,20 @@ class OCICompletionTokenDetails(BaseModel):
121
121
reasoningTokens : int
122
122
123
123
124
- class OCIPropmtTokensDetails (BaseModel ):
124
+ class OCIPromptTokensDetails (BaseModel ):
125
125
"""Prompt token details in the OCI response."""
126
126
127
127
cachedTokens : int
128
128
129
129
130
130
class OCIResponseUsage (BaseModel ):
131
131
"""Token usage in the OCI response."""
132
-
132
+
133
133
promptTokens : int
134
134
completionTokens : int
135
135
totalTokens : int
136
- completionTokensDetails : OCICompletionTokenDetails
137
- promptTokensDetails : OCIPropmtTokensDetails
136
+ completionTokensDetails : Optional [ OCICompletionTokenDetails ] = None
137
+ promptTokensDetails : Optional [ OCIPromptTokensDetails ] = None
138
138
139
139
140
140
class OCIResponseChoice (BaseModel ):
Original file line number Diff line number Diff line change @@ -213,6 +213,10 @@ def test_transform_response_simple_text(self):
213
213
assert result .usage .prompt_tokens == 10 # type: ignore
214
214
assert result .usage .completion_tokens == 20 # type: ignore
215
215
assert result .usage .total_tokens == 30 # type: ignore
216
+ # These are not handled in the transformer, TBH no idea why they are here
217
+ # but, for now, they seem to be always None
218
+ assert result .usage .completion_tokens_details is None
219
+ assert result .usage .prompt_tokens_details is None
216
220
217
221
def test_transform_response_with_tool_calls (self ):
218
222
"""
You can’t perform that action at this time.
0 commit comments