Fix BaseOpenAIChatCompletionClient token usage (#4770)

* Fix openai client token usage (#4769) * Include actual_usage in add_usage function * Make sure result is assigned * remove method --------- Co-authored-by: Jack Gerrits <[email protected]>
microsoft · Dec 30, 2024 · 0569689 · 0569689
1 parent cb1633b
commit 0569689
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/python/packages/autogen-ext/src/autogen_ext/models/openai/_openai_client.py b/python/packages/autogen-ext/src/autogen_ext/models/openai/_openai_client.py
@@ -558,8 +558,8 @@ async def create(
             logprobs=logprobs,
         )
 
-        _add_usage(self._actual_usage, usage)
-        _add_usage(self._total_usage, usage)
+        self._total_usage = _add_usage(self._total_usage, usage)
+        self._actual_usage = _add_usage(self._actual_usage, usage)
 
         # TODO - why is this cast needed?
         return response
@@ -775,8 +775,8 @@ async def create_stream(
             logprobs=logprobs,
         )
 
-        _add_usage(self._actual_usage, usage)
-        _add_usage(self._total_usage, usage)
+        self._total_usage = _add_usage(self._total_usage, usage)
+        self._actual_usage = _add_usage(self._actual_usage, usage)
 
         yield result