feat: Add use_profile_image and detailed token usage fields to generation models.
This commit is contained in:
@@ -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
|
||||
|
||||
|
||||
Reference in New Issue
Block a user