Unverified Commit a3aba7a9 authored by takatost's avatar takatost Committed by GitHub

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

parent 866ee5da
...@@ -20,7 +20,7 @@ from models.model import Account ...@@ -20,7 +20,7 @@ from models.model import Account
import secrets import secrets
import base64 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.') @click.command('reset-password', help='Reset the account password.')
...@@ -102,6 +102,7 @@ def reset_encrypt_key_pair(): ...@@ -102,6 +102,7 @@ def reset_encrypt_key_pair():
tenant.encrypt_public_key = generate_key_pair(tenant.id) tenant.encrypt_public_key = generate_key_pair(tenant.id)
db.session.query(Provider).filter(Provider.provider_type == 'custom').delete() db.session.query(Provider).filter(Provider.provider_type == 'custom').delete()
db.session.query(ProviderModel).delete()
db.session.commit() db.session.commit()
click.echo(click.style('Congratulations! ' click.echo(click.style('Congratulations! '
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment