diff options
Diffstat (limited to 'bot/utils/commands.py')
| -rw-r--r-- | bot/utils/commands.py | 11 | 
1 files changed, 11 insertions, 0 deletions
| diff --git a/bot/utils/commands.py b/bot/utils/commands.py new file mode 100644 index 00000000..7c04a25a --- /dev/null +++ b/bot/utils/commands.py @@ -0,0 +1,11 @@ +from typing import Optional + +from rapidfuzz import process + + +def get_command_suggestions( +        all_commands: list[str], query: str, *, cutoff: int = 60, limit: int = 3 +) -> Optional[list]: +    """Get similar command names.""" +    results = process.extract(query, all_commands, score_cutoff=cutoff, limit=limit) +    return [result[0] for result in results] | 
