andre/crewAI
1
0
Fork 0
mirror of https://github.com/crewAIInc/crewAI.git synced 2026-05-02 15:52:34 +00:00
Code Issues Actions 6 Packages Projects Releases Wiki Activity

refactered select_choice function for early return

Browse Source
This commit is contained in:
Rip&Tear
2024-10-13 18:09:33 +08:00
parent 1f9baf9b2c
commit 5d18c6312d
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
src/crewai/cli/create_crew.py
View File

@@ -69,12 +69,13 @@ def select_choice(prompt_message, choices):
except click.exceptions.Abort: except click.exceptions.Abort:
click.secho("Operation aborted by the user.", fg="red") click.secho("Operation aborted by the user.", fg="red")
return None return None
if 0 <= selected_index < len(choices):
return choices[selected_index] if not (0 <= selected_index < len(choices)):
else:
click.secho("Invalid selection.", fg="red") click.secho("Invalid selection.", fg="red")
return None return None
return choices[selected_index]
def select_provider(provider, all_providers, PROVIDERS): def select_provider(provider, all_providers, PROVIDERS):
if provider and provider.lower() not in all_providers and provider.lower() != 'other': if provider and provider.lower() not in all_providers and provider.lower() != 'other':
click.secho(f"Invalid provider: {provider}", fg="red") click.secho(f"Invalid provider: {provider}", fg="red")