Procházet zdrojové kódy

fix: provider model not delete when reset key pair (#946)

takatost před 1 rokem
rodič
revize
a3aba7a9aa
1 změnil soubory, kde provedl 2 přidání a 1 odebrání
  1. 2 1
      api/commands.py

+ 2 - 1
api/commands.py

@@ -20,7 +20,7 @@ from models.model import Account
 import secrets
 import base64
 
-from models.provider import Provider, ProviderType, ProviderQuotaType
+from models.provider import Provider, ProviderType, ProviderQuotaType, ProviderModel
 
 
 @click.command('reset-password', help='Reset the account password.')
@@ -102,6 +102,7 @@ def reset_encrypt_key_pair():
     tenant.encrypt_public_key = generate_key_pair(tenant.id)
 
     db.session.query(Provider).filter(Provider.provider_type == 'custom').delete()
+    db.session.query(ProviderModel).delete()
     db.session.commit()
 
     click.echo(click.style('Congratulations! '