fourstring
diff --git a/‎pyrogram/client/ext/base_client.py‎
Lines changed: 2 additions & 2 deletions b/‎pyrogram/client/ext/base_client.py‎
Lines changed: 2 additions & 2 deletions
diff --git a/‎pyrogram/client/style/html.py‎
Lines changed: 9 additions & 3 deletions b/‎pyrogram/client/style/html.py‎
Lines changed: 9 additions & 3 deletions
diff --git a/‎pyrogram/client/style/markdown.py‎
Lines changed: 10 additions & 5 deletions b/‎pyrogram/client/style/markdown.py‎
Lines changed: 10 additions & 5 deletions
@@ -81,8 +81,8 @@ def __init__(self):
         self.peers_by_username = {}
         self.peers_by_phone = {}
 
-        self.markdown = Markdown(self.peers_by_id)
-        self.html = HTML(self.peers_by_id)
+        self.markdown = Markdown(self)
+        self.html = HTML(self)
 
         self.session = None
         self.media_sessions = {}
 
@@ -19,6 +19,7 @@
 import re
 from collections import OrderedDict
 
+import pyrogram
 from pyrogram.api.types import (
     MessageEntityBold as Bold,
     MessageEntityItalic as Italic,
@@ -28,15 +29,16 @@
     MessageEntityMentionName as MentionInvalid,
     InputMessageEntityMentionName as Mention,
 )
+from pyrogram.errors import PeerIdInvalid
 from . import utils
 
 
 class HTML:
     HTML_RE = re.compile(r"<(\w+)(?: href=([\"'])([^<]+)\2)?>([^>]+)</\1>")
     MENTION_RE = re.compile(r"tg://user\?id=(\d+)")
 
-    def __init__(self, peers_by_id: dict = None):
-        self.peers_by_id = peers_by_id or {}
+    def __init__(self, client: "pyrogram.BaseClient" = None):
+        self.client = client
 
     def parse(self, message: str):
         entities = []
@@ -52,7 +54,11 @@ def parse(self, message: str):
 
                 if mention:
                     user_id = int(mention.group(1))
-                    input_user = self.peers_by_id.get(user_id, None)
+
+                    try:
+                        input_user = self.client.resolve_peer(user_id)
+                    except PeerIdInvalid:
+                        input_user = None
 
                     entity = (
                         Mention(offset=start, length=len(body), user_id=input_user)
 
@@ -19,6 +19,7 @@
 import re
 from collections import OrderedDict
 
+import pyrogram
 from pyrogram.api.types import (
     MessageEntityBold as Bold,
     MessageEntityItalic as Italic,
@@ -28,6 +29,7 @@
     MessageEntityMentionName as MentionInvalid,
     InputMessageEntityMentionName as Mention
 )
+from pyrogram.errors import PeerIdInvalid
 from . import utils
 
 
@@ -52,8 +54,8 @@ class Markdown:
     ))
     MENTION_RE = re.compile(r"tg://user\?id=(\d+)")
 
-    def __init__(self, peers_by_id: dict = None):
-        self.peers_by_id = peers_by_id or {}
+    def __init__(self, client: "pyrogram.BaseClient" = None):
+        self.client = client
 
     def parse(self, message: str):
         message = utils.add_surrogates(str(message or "")).strip()
@@ -69,12 +71,15 @@ def parse(self, message: str):
 
                 if mention:
                     user_id = int(mention.group(1))
-                    input_user = self.peers_by_id.get(user_id, None)
+
+                    try:
+                        input_user = self.client.resolve_peer(user_id)
+                    except PeerIdInvalid:
+                        input_user = None
 
                     entity = (
                         Mention(offset=start, length=len(text), user_id=input_user)
-                        if input_user
-                        else MentionInvalid(offset=start, length=len(text), user_id=user_id)
+                        if input_user else MentionInvalid(offset=start, length=len(text), user_id=user_id)
                     )
                 else:
                     entity = Url(offset=start, length=len(text), url=url)