fix: typing issues in plugins

1 files changed, 5 insertions, 3 deletions

diff --git a/plugins/b64.py b/plugins/b64.py
index d56ac5b3..1fec6123 100644
--- a/plugins/b64.py
+++ b/plugins/b64.py
@@ -24,7 +24,9 @@ This plugin also respects security guidelines listed in XEP-0419.
 
 from base64 import b64decode, b64encode
 from poezio.plugin_e2ee import E2EEPlugin
-from slixmpp import Message
+from poezio.tabs import ChatTab
+from slixmpp import Message, JID
+from typing import Optional
 
 
 class Plugin(E2EEPlugin):
@@ -37,14 +39,14 @@ class Plugin(E2EEPlugin):
     # This encryption mechanism is using <body/> as a container
     replace_body_with_eme = False
 
-    def decrypt(self, message: Message, _tab) -> None:
+    async def decrypt(self, message: Message, jid: Optional[JID], _tab: ChatTab) -> None:
         """
             Decrypt base64
         """
         body = message['body']
         message['body'] = b64decode(body.encode()).decode()
 
-    def encrypt(self, message: Message, _tab) -> None:
+    async def encrypt(self, message: Message, jid: Optional[JID], _tab: ChatTab) -> None:
         """
             Encrypt to base64
         """