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 '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] |