diff --git a/MakerSuite.py b/MakerSuite.py index a418f8e..9d5e588 100644 --- a/MakerSuite.py +++ b/MakerSuite.py @@ -98,7 +98,8 @@ def pretty_print_makersuite_keys(keys): billing_count += 1 if "(" in key.tier: unknown_keys.add(key) - keys_by_tier.setdefault(key.tier, []).append(key) + else: + keys_by_tier.setdefault(key.tier, []).append(key) for tier in output_order: if tier in keys_by_tier: diff --git a/OpenAI.py b/OpenAI.py index 9ecf27c..af2c4c2 100644 --- a/OpenAI.py +++ b/OpenAI.py @@ -342,7 +342,7 @@ def pretty_print_oai_keys(keys, cloned_keys): print(format_key_details(key)) if unknown in quota_keys_by_tier: - print(f"\n{unknown}:") + print(f"\nFound {len(quota_keys_by_tier[unknown])} keys of {unknown}:") for key in quota_keys_by_tier[unknown]: print(format_key_details(key))