Return autocomplete function after wrapping it

Voxel-Fox-Ltd · Sep 18, 2023 · 3abdc55 · 3abdc55
1 parent 966feaa
commit 3abdc55
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/novus/ext/client/command.py b/novus/ext/client/command.py
@@ -279,7 +279,7 @@ async def run(
     async def __call__(self, *args, **kwargs) -> Any:
         return await self.callback(self.owner, *args, **kwargs)
 
-    def autocomplete(self, func: AutocompleteCallback) -> None:
+    def autocomplete(self, func: AutocompleteCallback) -> AutocompleteCallback:
         """
         Add an autocomplete to this command.
 
@@ -302,6 +302,7 @@ async def echo_autocomplete(self, ctx: novus.Interaction):
         """
 
         self._autocomplete = func
+        return func
 
     async def run_autocomplete(
             self,