nova-api/api/provider_auth.py

"""This module contains functions for authenticating with providers."""

import asyncio

async def invalidate_key(provider_and_key: str) -> none:
    """Invalidates a key stored in the secret/ folder by storing it in the associated .invalid.txt file.
The schmea in which <provider_and_key> should be passed is:
<provider_name><key>, e.g.
closed4>sk-...
"""

    if not provider_and_key:
        return

    provider = provider_and_key.split('>')[0]
    provider_file = f'secret/{provider}.txt'
    key = provider_and_key.split('>')[1]

    with open(provider_file, encoding='utf8') as f_in:
        text = f_in.read()

    with open(provider_file, 'w', encoding='utf8') as f_out:
        f_out.write(text.replace(key, ''))

    with open(f'secret/{provider}.invalid.txt', 'a', encoding='utf8') as f:
        f.write(key + '\n')

if __name__ == '__main__':
    asyncio.run(invalidate_key('closed>sk-...'))