feat(db): add user bio and phone_number fields with context integration

- Add bio and phone_number to users schema and db migrations
- Extract bio and phone_number using Pyrogram and store in DB
- Inject 'You bio' and 'Them bio' into the LLM chat history context
- Exclude phone_number from LLM context to preserve privacy
- Auto-sync owner full profile (first_name, username, bio) on all bot command interactions
- Skip fetching bio for group/channel chats to prevent leaking group descriptions

Author: oponfil

clients/pyrogram_client.py

@@ -254,6 +254,7 @@ async def read_chat_history(user_id: int, chat_id: int, limit: int = MAX_CONTEXT
                 "name": sender.first_name if sender else None,
                 "last_name": sender.last_name if sender else None,
                 "username": sender.username if sender else None,
+                "phone_number": sender.phone_number if sender else None,
             })
 
             if msg.voice and text is None:
@@ -360,6 +361,31 @@ async def get_dialog_info(user_id: int, limit: int) -> list[dict]:
     return dialogs
 
 
+async def get_chat_bio(user_id: int, chat_id: int) -> str | None:
+    """Возвращает bio (описание профиля) собеседника через get_chat().
+    Для групп и каналов (chat_id < 0) всегда возвращает None.
+
+    Args:
+        user_id: Telegram user ID (владелец сессии)
+        chat_id: ID чата (собеседника)
+
+    Returns:
+        Строка bio или None если недоступно / ошибка.
+    """
+    if chat_id < 0:
+        return None
+
+    client = _active_clients.get(user_id)
+    if not client:
+        return None
+
+    try:
+        chat = await client.get_chat(chat_id)
+        return getattr(chat, "bio", None) or None
+    except Exception:
+        return None
+
+
 def get_active_user_ids() -> list[int]:
     """Возвращает ID пользователей с активными Pyrogram-клиентами."""
     return list(_active_clients.keys())

database/users.py

@@ -30,6 +30,8 @@ async def upsert_user(
     is_bot: bool = False,
     is_premium: bool = False,
     language_code: Optional[str] = None,
+    phone_number: Optional[str] = None,
+    bio: Optional[str] = None,
 ) -> bool:
     """Создаёт или обновляет пользователя в БД.
 
@@ -45,6 +47,10 @@ async def upsert_user(
         data["last_name"] = last_name
     if language_code is not None:
         data["language_code"] = language_code
+    if phone_number is not None:
+        data["phone_number"] = phone_number
+    if bio is not None:
+        data["bio"] = bio
 
     try:
         await run_supabase(
@@ -220,6 +226,8 @@ async def ensure_user_exists(
     is_bot: bool = False,
     is_premium: bool = False,
     language_code: Optional[str] = None,
+    phone_number: Optional[str] = None,
+    bio: Optional[str] = None,
 ) -> dict:
     """Возвращает пользователя, создавая запись при отсутствии.
 
@@ -238,6 +246,8 @@ async def ensure_user_exists(
         is_bot=is_bot,
         is_premium=is_premium,
         language_code=language_code,
+        phone_number=phone_number,
+        bio=bio,
     )
     if not created:
         raise UserStorageError(f"Failed to create user {user_id}")
@@ -250,6 +260,8 @@ async def ensure_user_exists(
         "is_bot": is_bot,
         "is_premium": is_premium,
         "language_code": language_code,
+        "phone_number": phone_number,
+        "bio": bio,
         "settings": {},
     }
 

handlers/pyrogram_handlers.py

@@ -347,6 +347,8 @@ async def on_pyrogram_message(user_id: int, pyrogram_client_instance, message) -
             "username": opponent.username,
             "language_code": opponent.language_code,
             "is_premium": opponent.is_premium,
+            "bio": await pyrogram_client.get_chat_bio(user_id, chat_id),
+            "phone_number": opponent.phone_number,
         } if opponent else None
 
         # Генерируем ответ
@@ -420,6 +422,8 @@ async def _clear_probe_draft() -> None:
                     "first_name": msg.get("name"),
                     "last_name": msg.get("last_name"),
                     "username": msg.get("username"),
+                    "bio": await pyrogram_client.get_chat_bio(user_id, chat_id),
+                    "phone_number": msg.get("phone_number"),
                 }
                 break
 
@@ -509,6 +513,8 @@ async def _regenerate_reply(user_id: int, chat_id: int) -> None:
                     "first_name": msg.get("name"),
                     "last_name": msg.get("last_name"),
                     "username": msg.get("username"),
+                    "bio": await pyrogram_client.get_chat_bio(user_id, chat_id),
+                    "phone_number": msg.get("phone_number"),
                 }
                 break
 
@@ -778,6 +784,8 @@ async def on_pyrogram_draft(user_id: int, chat_id: int, draft_text: str) -> None
                     "first_name": msg.get("name"),
                     "last_name": msg.get("last_name"),
                     "username": msg.get("username"),
+                    "bio": await pyrogram_client.get_chat_bio(user_id, chat_id),
+                    "phone_number": msg.get("phone_number"),
                 }
                 break
 

schema.sql

@@ -12,6 +12,8 @@ create table if not exists public.users (
   first_seen    timestamptz not null default now(), -- Время первого контакта
   last_msg_at   timestamptz,                       -- Время последнего сообщения
   language_code text default 'en',                 -- Язык пользователя (ISO 639-1)
+  phone_number  text,                              -- Номер телефона пользователя
+  bio           text,                              -- Биография пользователя (из getChat)
   tg_rating     integer,                           -- Рейтинг Telegram Stars (из getChat)
   session_string text,                              -- Зашифрованный Pyrogram session string (Client API)
   settings      jsonb default '{}'                  -- Настройки пользователя (drafts_enabled, pro_model)

tests/test_database_users.py

@@ -354,6 +354,8 @@ async def test_creates_missing_user(self):
             is_bot=False,
             is_premium=False,
             language_code="ru",
+            phone_number=None,
+            bio=None,
         )
         assert result["user_id"] == 123
         assert result["username"] == "alice"

tests/test_handlers.py

@@ -553,6 +553,7 @@ async def test_voice_message_transcribes_and_generates_draft(self):
             mock_pc.read_chat_history = AsyncMock(return_value=voice_history)
             mock_pc.set_draft = AsyncMock(return_value=True)
             mock_pc.get_draft = AsyncMock(return_value=None)
+            mock_pc.get_chat_bio = AsyncMock(return_value=None)
             mock_gen.return_value = "Всё отлично!"
 
             await on_pyrogram_message(123, MagicMock(), message)
@@ -614,6 +615,7 @@ async def test_generates_and_sets_draft(self):
             ])
             mock_pc.set_draft = AsyncMock(return_value=True)
             mock_pc.get_draft = AsyncMock(return_value=None)
+            mock_pc.get_chat_bio = AsyncMock(return_value=None)
             mock_gen.return_value = "Hi there!"
 
             await on_pyrogram_message(123, MagicMock(), message)
@@ -647,6 +649,7 @@ async def test_invalid_auto_reply_is_treated_as_off(self):
             ])
             mock_pc.set_draft = AsyncMock(return_value=True)
             mock_pc.get_draft = AsyncMock(return_value=None)
+            mock_pc.get_chat_bio = AsyncMock(return_value=None)
             mock_gen.return_value = "Hi there!"
 
             await on_pyrogram_message(123, MagicMock(), message)
@@ -675,6 +678,7 @@ async def test_second_message_during_lock_is_queued(self):
              patch("handlers.pyrogram_handlers.get_user", new_callable=AsyncMock, return_value={"language_code": "en", "settings": {}}):
             mock_pc.set_draft = AsyncMock()
             mock_pc.get_draft = AsyncMock(return_value=None)
+            mock_pc.get_chat_bio = AsyncMock(return_value=None)
 
             await on_pyrogram_message(123, MagicMock(), message)
 
@@ -711,6 +715,7 @@ async def test_pending_message_triggers_regeneration_after_lock_release(self):
             ])
             mock_pc.set_draft = AsyncMock(return_value=True)
             mock_pc.get_draft = AsyncMock(return_value=None)
+            mock_pc.get_chat_bio = AsyncMock(return_value=None)
             mock_gen.return_value = "Hi there!"
 
             await on_pyrogram_message(123, MagicMock(), message)
@@ -746,6 +751,7 @@ async def test_no_regeneration_without_pending(self):
             ])
             mock_pc.set_draft = AsyncMock(return_value=True)
             mock_pc.get_draft = AsyncMock(return_value=None)
+            mock_pc.get_chat_bio = AsyncMock(return_value=None)
             mock_gen.return_value = "Hi there!"
 
             await on_pyrogram_message(123, MagicMock(), message)
@@ -781,6 +787,7 @@ async def test_duplicate_message_is_skipped(self):
             ])
             mock_pc.set_draft = AsyncMock(return_value=True)
             mock_pc.get_draft = AsyncMock(return_value=None)
+            mock_pc.get_chat_bio = AsyncMock(return_value=None)
             mock_gen.return_value = "Hi there!"
             await on_pyrogram_message(123, MagicMock(), message)
             assert mock_gen.call_count == 1
@@ -1768,6 +1775,7 @@ async def test_on_pyrogram_message_uses_pro_model_by_default(self):
             ])
             mock_pc.set_draft = AsyncMock(return_value=True)
             mock_pc.get_draft = AsyncMock(return_value=None)
+            mock_pc.get_chat_bio = AsyncMock(return_value=None)
             mock_gen.return_value = "Hi there!"
 
             await on_pyrogram_message(123, MagicMock(), message)
@@ -1789,6 +1797,7 @@ async def test_regenerate_reply_uses_pro_model_by_default(self):
             ])
             mock_pc.set_draft = AsyncMock(return_value=True)
             mock_pc.get_draft = AsyncMock(return_value=None)
+            mock_pc.get_chat_bio = AsyncMock(return_value=None)
             mock_gen.return_value = "Hi there!"
 
             await _regenerate_reply(123, 456)

tests/test_pyrogram_client.py

@@ -573,3 +573,63 @@ async def test_returns_false_on_error(self):
         assert result is False
 
         pyrogram_client._active_clients.pop(123, None)
+
+
+class TestGetChatBio:
+    """Тесты для get_chat_bio()."""
+
+    @pytest.mark.asyncio
+    async def test_returns_bio_string(self):
+        """Возвращает bio при успешном get_chat."""
+        mock_client = AsyncMock()
+        mock_chat = MagicMock()
+        mock_chat.bio = "Дизайнер из Москвы"
+        mock_client.get_chat = AsyncMock(return_value=mock_chat)
+
+        pyrogram_client._active_clients[123] = mock_client
+
+        result = await pyrogram_client.get_chat_bio(123, 456)
+
+        assert result == "Дизайнер из Москвы"
+        mock_client.get_chat.assert_called_once_with(456)
+
+        pyrogram_client._active_clients.pop(123, None)
+
+    @pytest.mark.asyncio
+    async def test_returns_none_when_no_client(self):
+        """Без активного клиента → None."""
+        pyrogram_client._active_clients.pop(123, None)
+
+        result = await pyrogram_client.get_chat_bio(123, 456)
+
+        assert result is None
+
+    @pytest.mark.asyncio
+    async def test_returns_none_on_exception(self):
+        """При ошибке get_chat → None (graceful)."""
+        mock_client = AsyncMock()
+        mock_client.get_chat = AsyncMock(side_effect=Exception("API error"))
+
+        pyrogram_client._active_clients[123] = mock_client
+
+        result = await pyrogram_client.get_chat_bio(123, 456)
+
+        assert result is None
+
+        pyrogram_client._active_clients.pop(123, None)
+
+    @pytest.mark.asyncio
+    async def test_returns_none_when_bio_is_empty(self):
+        """Пустая строка bio → None."""
+        mock_client = AsyncMock()
+        mock_chat = MagicMock()
+        mock_chat.bio = ""
+        mock_client.get_chat = AsyncMock(return_value=mock_chat)
+
+        pyrogram_client._active_clients[123] = mock_client
+
+        result = await pyrogram_client.get_chat_bio(123, 456)
+
+        assert result is None
+
+        pyrogram_client._active_clients.pop(123, None)

tests/test_user_helpers.py

@@ -11,21 +11,14 @@ class TestEnsureEffectiveUser:
     """Тесты для ensure_effective_user()."""
 
     @pytest.mark.asyncio
-    async def test_passes_effective_user_fields(self, mock_update):
+    async def test_always_upserts_and_returns_user(self, mock_update):
         expected_user = {"user_id": mock_update.effective_user.id, "settings": {}}
 
-        with patch("utils.telegram_user.ensure_user_exists", new_callable=AsyncMock, return_value=expected_user) as mock_ensure:
+        with patch("utils.telegram_user.get_user", new_callable=AsyncMock, return_value=expected_user), \
+             patch("utils.telegram_user.upsert_effective_user", new_callable=AsyncMock) as mock_upsert:
             result = await ensure_effective_user(mock_update)
 
-        mock_ensure.assert_called_once_with(
-            user_id=mock_update.effective_user.id,
-            username=mock_update.effective_user.username,
-            first_name=mock_update.effective_user.first_name,
-            last_name=mock_update.effective_user.last_name,
-            is_bot=mock_update.effective_user.is_bot,
-            is_premium=bool(mock_update.effective_user.is_premium),
-            language_code=mock_update.effective_user.language_code,
-        )
+        mock_upsert.assert_called_once_with(mock_update)
         assert result == expected_user
 
 
@@ -34,7 +27,8 @@ class TestUpsertEffectiveUser:
 
     @pytest.mark.asyncio
     async def test_passes_effective_user_fields(self, mock_update):
-        with patch("utils.telegram_user.upsert_user", new_callable=AsyncMock, return_value=True) as mock_upsert:
+        with patch("utils.telegram_user.upsert_user", new_callable=AsyncMock, return_value=True) as mock_upsert, \
+             patch("utils.telegram_user._fetch_bio", new_callable=AsyncMock, return_value=None):
             result = await upsert_effective_user(mock_update)
 
         mock_upsert.assert_called_once_with(
@@ -45,5 +39,7 @@ async def test_passes_effective_user_fields(self, mock_update):
             is_bot=mock_update.effective_user.is_bot,
             is_premium=bool(mock_update.effective_user.is_premium),
             language_code=mock_update.effective_user.language_code,
+            phone_number=None,
+            bio=None,
         )
         assert result is True

tests/test_utils.py

@@ -192,3 +192,34 @@ def test_tz_offset_zero_unchanged(self):
         result = format_chat_history(history, tz_offset=0)
         assert "[2026-03-14 14:30]" in result
 
+    def test_bio_shown_in_participants(self):
+        """Bio оппонента отображается в PARTICIPANTS блоке."""
+        history = [{"role": "other", "text": "Привет", "name": "Марина"}]
+        opponent_info = {"first_name": "Марина", "bio": "Дизайнер из Москвы"}
+
+        result = format_chat_history(history, None, opponent_info)
+        assert "Them bio: Дизайнер из Москвы" in result
+
+    def test_bio_absent_when_none(self):
+        """Строка bio отсутствует когда bio = None."""
+        history = [{"role": "other", "text": "Привет", "name": "Марина"}]
+        opponent_info = {"first_name": "Марина", "bio": None}
+
+        result = format_chat_history(history, None, opponent_info)
+        assert "bio" not in result
+
+    def test_bio_absent_when_missing(self):
+        """Строка bio отсутствует когда ключ bio отсутствует."""
+        history = [{"role": "other", "text": "Привет", "name": "Марина"}]
+        opponent_info = {"first_name": "Марина"}
+
+        result = format_chat_history(history, None, opponent_info)
+        assert "bio" not in result
+
+    def test_you_bio_shown(self):
+        """Bio владельца отображается в PARTICIPANTS блоке."""
+        history = [{"role": "other", "text": "Привет", "name": "Марина"}]
+        user_info = {"first_name": "Алексей", "bio": "Программист из РФ"}
+
+        result = format_chat_history(history, user_info, None)
+        assert "You bio: Программист из РФ" in result