update

Author: qbc2016

src/agentscope/models/gemini_model.py

@@ -245,7 +245,8 @@ def _save_model_invocation_and_update_monitor(
                 **kwargs,
             ).total_tokens
             completion_tokens = self.client.models.count_tokens(
-                **kwargs,
+                model=self.model_name,
+                contents=response.candidates[0].content.parts,
             ).total_tokens
 
         formatted_usage = ChatUsage(