def get_usage_tokens()

in shared/util.py [0:0]

• 10 lines of code
• 10 McCabe index (conditional complexity)

def get_usage_tokens(function_result, token_type='total'):
    metadata = function_result.metadata['metadata']
    usage_tokens = 0
    if token_type == 'completion':
        usage_tokens = sum(item['usage'].completion_tokens for item in metadata if 'usage' in item)
    elif token_type == 'prompt':
        usage_tokens = sum(item['usage'].prompt_tokens for item in metadata if 'usage' in item)
    elif token_type == 'total':
        usage_tokens = sum(item['usage'].total_tokens for item in metadata if 'usage' in item)        
    return usage_tokens