diff options
| author | 2022-08-18 17:52:36 +0100 | |
|---|---|---|
| committer | 2022-08-18 17:52:36 +0100 | |
| commit | 96f84a147c5e862d6153f48b2de83613aa3bd654 (patch) | |
| tree | 6962c76ff7b9a9b8b732e5663b807e4a3d70f2de /bot/utils/commands.py | |
| parent | add more packages to .latex (diff) | |
| parent | Help command fix, normalize suggestions for unknown commands (#1064) (diff) | |
Merge branch 'main' into feat/latex-enhancement
Diffstat (limited to '')
| -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] |