feat: Add use_profile_image and detailed token usage fields to generation models.

This commit is contained in:
xds
2026-02-05 22:19:57 +03:00
parent 002c949f08
commit c0debab0cb
8 changed files with 21 additions and 3 deletions

View File

@@ -131,9 +131,17 @@ class GoogleAdapter:
else:
logger.warning("No images text generated from parts")
input_tokens = 0
output_tokens = 0
if response.usage_metadata:
input_tokens = response.usage_metadata.prompt_token_count
output_tokens = response.usage_metadata.candidates_token_count
metrics = {
"api_execution_time_seconds": api_duration,
"token_usage": token_usage
"token_usage": token_usage,
"input_token_usage": input_tokens,
"output_token_usage": output_tokens
}
return generated_images, metrics