updated click prompt to remove default number

This commit is contained in:
Rip&Tear
2024-10-12 13:05:55 +08:00
parent 6c29ebafea
commit 1ea8115d56

View File

@@ -194,7 +194,7 @@ def create_crew(name, parent_folder=None):
while True:
try:
selected_index = click.prompt(
"Enter the number of your choice", type=int, default=1
"Enter the number of your choice", type=int
) - 1
if 0 <= selected_index < len(PROVIDERS) + 1:
selected_provider = (PROVIDERS + ['other'])[selected_index]
@@ -259,7 +259,7 @@ def create_crew(name, parent_folder=None):
while True:
try:
selected_model_index = click.prompt(
"Enter the number of your choice", type=int, default=1
"Enter the number of your choice", type=int
) - 1
if 0 <= selected_model_index < len(available_models):
model = available_models[selected_model_index]
@@ -322,4 +322,4 @@ def create_crew(name, parent_folder=None):
dst_file = src_folder / file_name
copy_template(src_file, dst_file, name, class_name, folder_name)
click.secho(f"Crew {name} created successfully!", fg="green", bold=True)
click.secho(f"Crew {name} created successfully!", fg="green", bold=True)