Add `usageMetadata` to `GenerateContentResponse`

google-gemini / generative-ai-swift

The official Swift library for the Google Gemini API

Apache License 2.0

864 stars 131 forks source link

Closed andrewheard closed 2 months ago

andrewheard commented 2 months ago

Added usageMetadata to GenerateContentResponse. This is a port of https://github.com/firebase/firebase-ios-sdk/pull/12777.

Notes:

Removed promptFeedback checks from testGenerateContent_success_basicReplyShort since it is no longer populated by the backend in this scenario.
Will re-run the prompts when promptTokenCount and totalTokenCount are added to the responses.
- They are intentionally parsed as 0 when omitted.