ValidationError: get_chats() падает на закреплённом пересланном сообщении (LinkedMessage.message обязателен при type="forward")

[iamptic]

Описание

bot.get_chats() (и любой парсинг ответа GET /chats через модель Chats) падает с ValidationError, если хотя бы в одном чате закреплено пересланное сообщение.

Для пересланного закрепа API MAX возвращает pinned_message.link с type: "forward" и без поля message (поле message присутствует только у reply-линков, у forward его в payload нет). Однако в модели LinkedMessage поле message объявлено обязательным:

# maxapi/types/message.py
class LinkedMessage(BaseModel):
    type: MessageLinkType
    sender: User | None = None
    chat_id: int | None = None
    message: MessageBody          # <-- required, падает на type="forward"

Так как Chats валидирует всю страницу атомарно (chats: list[Chat]), один «отравленный» чат рушит весь ответ get_chats() — список чатов становится пустым у бота, даже если он админ во всех чатах. На проде это привело к тому, что у пользователей пропал список их чатов.

Точный текст ошибки

1 validation error for Chats
chats.0.pinned_message.link.message
  Field required [type=missing, input_value={'type': 'forward', 'chat_id': ...}, input_type=dict]
    For further information visit https://errors.pydantic.dev/2.x/v/missing

Минимальный repro

from maxapi.types.chats import Chats

payload = {
  "chats": [
    {
      "chat_id": 70000000000001,
      "type": "chat",
      "status": "active",
      "title": "Test chat",
      "last_event_time": 1700000000000,
      "participants_count": 3,
      "is_public": False,
      "pinned_message": {
        "recipient": {"chat_id": 70000000000001, "chat_type": "chat"},
        "timestamp": 1700000000000,
        "link": {
          "type": "forward",
          "chat_id": 70000000000002
          # поля "message" нет — так приходит для пересланного закрепа
        },
        "body": {"mid": "mid1", "seq": 1, "text": "forwarded"}
      }
    }
  ],
  "marker": None
}

Chats.model_validate(payload)   # -> ValidationError: ...link.message Field required

Затронутые версии

Воспроизводится на 0.9.16 и на актуальной 1.1.0 (проверено в чистом venv, Python 3.12, pydantic 2.x). В обеих LinkedMessage.message остаётся обязательным.

Предлагаемое исправление

Сделать message необязательным, по аналогии с уже исправленным sender (#11):

class LinkedMessage(BaseModel):
    type: MessageLinkType
    sender: User | None = None
    chat_id: int | None = None
    message: MessageBody | None = None   # forward-линк приходит без message

Места, использующие self.link.message.mid (например Message.edit()), стоит защитить проверкой на None.

См. также #11 — там аналогичная проблема с link.sender.

[K1rL3s]

Я не могу воспроизвести. Можно больше деталей?

Бот админ в канале и чате, пересланное сообщение закреплено. На bot.get_chats() чаты успешно получаются, и в Chat.pinned_message.link.message есть сообщение.

Код:

import asyncio
import logging
import os

from maxapi import Bot

async def main() -> None:
    logging.basicConfig(level=logging.DEBUG)
    token = os.environ["TOKEN"]

    maxapi_bot = Bot(token=token)
    chats = await maxapi_bot.get_chats(count=100)
    for chat in chats.chats:
        print(chat)
    await maxapi_bot.close_session()

if __name__ == "__main__":
    asyncio.run(main())

Вывод:

chat_id=-75626443272901 type=<ChatType.CHAT: 'chat'> status=<ChatStatus.ACTIVE: 'active'> title='тест чат' icon=None last_event_time=1780782268340 participants_count=3 owner_id=6033605 participants=None is_public=False link='https://max.ru/join/---' description=None dialog_with_user=None messages_count=4 chat_message_id=None pinned_message=Message(sender=User(user_id=6033605, first_name='КирЛес', last_name='', username=None, is_bot=False, last_activity_time=1780782391000, description=None, avatar_url=None, full_avatar_url=None, commands=None), recipient=Recipient(user_id=None, chat_id=-75626443272901, chat_type=<ChatType.CHAT: 'chat'>), timestamp=1780782256710, link=LinkedMessage(type=<MessageLinkType.FORWARD: 'forward'>, sender=User(user_id=6033605, first_name='КирЛес', last_name='', username=None, is_bot=False, last_activity_time=1780782391000, description=None, avatar_url=None, full_avatar_url=None, commands=None), chat_id=0, message=MessageBody(mid='mid.0000000000000000019e9e9272dc7273', seq=116704992187675251, text='Мб мб', attachments=[], markup=[])), body=MessageBody(mid='mid.ffffbb37d8b68d3b019e9ee4d2467fcf', seq=116705345975779279, text='', attachments=[], markup=[]), stat=None, url_api=None, bot=Bot(token='***'))

chat_id=-70923932175045 type=<ChatType.CHANNEL: 'channel'> status=<ChatStatus.ACTIVE: 'active'> title='test' icon=None last_event_time=1780781947341 participants_count=4 owner_id=6033605 participants=None is_public=False link='https://max.ru/join/---' description='channel' dialog_with_user=None messages_count=61 chat_message_id=None pinned_message=Message(sender=None, recipient=Recipient(user_id=None, chat_id=-70923932175045, chat_type=<ChatType.CHANNEL: 'channel'>), timestamp=1780781947341, link=LinkedMessage(type=<MessageLinkType.FORWARD: 'forward'>, sender=User(user_id=6033605, first_name='КирЛес', last_name='', username=None, is_bot=False, last_activity_time=1780782391000, description=None, avatar_url=None, full_avatar_url=None, commands=None), chat_id=0, message=MessageBody(mid='mid.0000000000000000019e9e9272dc7273', seq=116704992187675251, text='Мб мб', attachments=[], markup=[])), body=MessageBody(mid='mid.ffffbf7ebc37753b019e9ee019cd7fcf', seq=116705325700972495, text='', attachments=[], markup=[]), stat=None, url_api=None, bot=Bot(token='***'))

Как выглядят чаты:

[love-apples]

@iamptic