Structured system messages: actor + mentions as objects
Some checks failed
Deploy Tracker / deploy (push) Failing after 5s

- Message model: added actor_id FK (who initiated the action)
- MessageOut: mentions as MemberBrief[], actor as MemberBrief
- _system_message: accepts mentioned_members list, stores actor_id
- WS filtering: checks mentions array (not just text content)
- broadcast_message: handles both string and dict mentions
This commit is contained in:
markov 2026-02-27 08:33:39 +01:00
parent f20da3685d
commit 5642f53e11
6 changed files with 65 additions and 19 deletions

View File

@ -14,6 +14,15 @@ from .schemas import (
)
def _resolve_mentions(m: Message) -> list[MemberBrief]:
"""Convert stored mention slugs to MemberBrief objects.
Uses pre-loaded relationships if available, otherwise returns slug-only briefs."""
if not m.mentions:
return []
# For now: return slug-only briefs (id will be empty until we store mention IDs)
return [MemberBrief(id="", slug=slug, name=slug) for slug in m.mentions]
def member_brief(m: Member | None) -> MemberBrief | None:
if not m:
return None
@ -75,7 +84,8 @@ def message_out(m: Message) -> MessageOut:
author=member_brief(m.author) if m.author else None,
content=m.content,
thinking=m.thinking,
mentions=m.mentions or [],
mentions=_resolve_mentions(m),
actor=member_brief(m.actor) if hasattr(m, 'actor') and m.actor else None,
voice_url=m.voice_url,
attachments=[attachment_out(a) for a in (m.attachments or [])],
created_at=m.created_at.isoformat() if m.created_at else "",

View File

@ -61,7 +61,8 @@ class MessageOut(BaseModel):
author: MemberBrief | None = None
content: str
thinking: str | None = None
mentions: list[str] = []
mentions: list[MemberBrief] = []
actor: MemberBrief | None = None
voice_url: str | None = None
attachments: list[AttachmentOut] = []
created_at: str

View File

@ -16,7 +16,7 @@ from ..enums import (
)
from ..models import Task, Step, Project, Member, Message, Chat, TaskAction
from .auth import get_current_member
from .schemas import TaskOut, MessageOut
from .schemas import TaskOut, MessageOut, MemberBrief
from .converters import task_out, message_out, member_brief
router = APIRouter(tags=["tasks"])
@ -94,15 +94,24 @@ async def _system_message(
task_text: str | None = None,
project_slug: str = "",
actor: Member | None = None,
mentioned_members: list[Member] | None = None,
):
"""Create system messages: one in project chat + one in task comments.
Uses MessageOut schema for WS broadcasts."""
actor: who initiated the action (stored as actor_id)
mentioned_members: members referenced in the message (stored in mentions array)
"""
from ..ws.manager import manager
prefix = project_slug[:2].upper() if project_slug else "XX"
key = f"{prefix}-{task.number}"
task_text = task_text or chat_text
actor_slug = actor.slug if actor else "system"
actor_brief = MemberBrief(id=str(actor.id), slug=actor.slug, name=actor.name) if actor else None
# Mentioned members as slugs (DB storage) and briefs (broadcast)
mention_slugs = [m.slug for m in (mentioned_members or [])]
mention_briefs = [MemberBrief(id=str(m.id), slug=m.slug, name=m.name) for m in (mentioned_members or [])]
chat_id = await _get_project_chat_id(db, task.project_id)
@ -111,8 +120,9 @@ async def _system_message(
task_id=task.id,
author_type=AuthorType.SYSTEM,
author_id=None,
actor_id=actor.id if actor else None,
content=task_text,
mentions=[],
mentions=mention_slugs, # still stored as slugs in DB for compat
)
db.add(task_msg)
@ -123,26 +133,29 @@ async def _system_message(
chat_id=chat_id,
author_type=AuthorType.SYSTEM,
author_id=None,
actor_id=actor.id if actor else None,
content=chat_text,
mentions=[],
mentions=mention_slugs,
)
db.add(chat_msg)
await db.flush()
# Build MessageOut for task comment broadcast (use broadcast_message, not broadcast_task_event)
now_iso = datetime.datetime.now(datetime.timezone.utc).isoformat()
# Build MessageOut for task comment broadcast
task_msg_out = MessageOut(
id=str(task_msg.id),
task_id=str(task.id),
author_type=AuthorType.SYSTEM,
author_id=None,
author=None,
actor=actor_brief,
content=task_text,
mentions=[],
mentions=mention_briefs,
attachments=[],
created_at=task_msg.created_at.isoformat() if task_msg.created_at else datetime.datetime.now(datetime.timezone.utc).isoformat(),
created_at=task_msg.created_at.isoformat() if task_msg.created_at else now_iso,
)
# Broadcast task comment via broadcast_message (applies agent filtering: @mention check)
project_id = str(task.project_id)
await manager.broadcast_message(
project_id,
@ -158,12 +171,13 @@ async def _system_message(
author_type=AuthorType.SYSTEM,
author_id=None,
author=None,
actor=actor_brief,
content=chat_text,
mentions=[],
mentions=mention_briefs,
attachments=[],
created_at=chat_msg.created_at.isoformat() if chat_msg.created_at else datetime.datetime.now(datetime.timezone.utc).isoformat(),
created_at=chat_msg.created_at.isoformat() if chat_msg.created_at else now_iso,
)
await manager.broadcast_message(str(task.project_id), chat_msg_out.model_dump(), author_slug="system")
await manager.broadcast_message(project_id, chat_msg_out.model_dump(), author_slug="system")
async def _get_task(task_id: str, db: AsyncSession) -> Task:
@ -284,11 +298,14 @@ async def create_task(
# System message
chat_text = f"{key} создана: {task.title}"
task_text = f"Задача создана: {task.title}"
mentioned = []
if task_full.assignee:
chat_text += f" → @{task_full.assignee.slug}"
task_text += f". Назначена на @{task_full.assignee.slug}"
mentioned.append(task_full.assignee)
await _system_message(db, task_full, chat_text=chat_text, task_text=task_text,
project_slug=project.slug, actor=current_member)
project_slug=project.slug, actor=current_member,
mentioned_members=mentioned)
await db.commit()
return task_out(task_full, project.slug)
@ -357,6 +374,7 @@ async def update_task(
chat_text=f"{key}: назначена на @{new_assignee.slug}",
task_text=f"{who} назначил задачу на @{new_assignee.slug} ({now_str})",
project_slug=slug, actor=current_member,
mentioned_members=[new_assignee],
)
if new_assignee_id not in (task.watcher_ids or []):
task.watcher_ids = (task.watcher_ids or []) + [new_assignee_id]
@ -518,6 +536,7 @@ async def assign_task(
chat_text=f"{key}: назначена на @{assignee.slug}",
task_text=f"@{current_member.slug} назначил задачу на @{assignee.slug}",
project_slug=proj_slug, actor=current_member,
mentioned_members=[assignee],
)
await db.commit()

View File

@ -44,6 +44,9 @@ class Message(Base):
author_type: Mapped[str] = mapped_column(String(20), nullable=False) # human | agent | system
author_id: Mapped[uuid.UUID | None] = mapped_column(ForeignKey("members.id"), nullable=True)
# Actor — who initiated the action (for system messages: who created/moved/assigned the task)
actor_id: Mapped[uuid.UUID | None] = mapped_column(ForeignKey("members.id"), nullable=True)
# Content
content: Mapped[str] = mapped_column(Text, nullable=False)
thinking: Mapped[str | None] = mapped_column(Text, nullable=True) # LLM reasoning/thinking block
@ -52,7 +55,8 @@ class Message(Base):
# Relationships
chat: Mapped["Chat | None"] = relationship(back_populates="messages", foreign_keys=[chat_id])
author: Mapped["Member"] = relationship()
author: Mapped["Member"] = relationship(foreign_keys=[author_id])
actor: Mapped["Member | None"] = relationship(foreign_keys=[actor_id])
attachments: Mapped[list["Attachment"]] = relationship(back_populates="message", cascade="all, delete-orphan")
replies: Mapped[list["Message"]] = relationship(back_populates="parent")
parent: Mapped["Message | None"] = relationship(back_populates="replies", remote_side="Message.id")

View File

@ -31,6 +31,8 @@ def _to_member_brief(member: Member) -> MemberBrief:
def _to_message_out(msg: Message, author: Member | None = None) -> MessageOut:
# Convert mention slugs to MemberBrief (slug-only for WS handler; full resolution in REST)
mention_briefs = [MemberBrief(id="", slug=s, name=s) for s in (msg.mentions or [])]
return MessageOut(
id=str(msg.id),
chat_id=str(msg.chat_id) if msg.chat_id else None,
@ -41,7 +43,8 @@ def _to_message_out(msg: Message, author: Member | None = None) -> MessageOut:
author=_to_member_brief(author) if author else None,
content=msg.content,
thinking=msg.thinking,
mentions=msg.mentions or [],
mentions=mention_briefs,
actor=_to_member_brief(msg.actor) if hasattr(msg, 'actor') and msg.actor else None,
voice_url=msg.voice_url,
attachments=[], # WS broadcasts don't include full attachments
created_at=msg.created_at.isoformat() if msg.created_at else "",

View File

@ -89,7 +89,14 @@ class ConnectionManager:
- If chat_listen == "mentions": only if @slug in mentions
- System messages: only if @slug mentioned in content
"""
mentions = message.get("mentions", [])
raw_mentions = message.get("mentions", [])
# mentions can be list of strings (slugs) or list of dicts (MemberBrief objects)
mentions: list[str] = []
for m in raw_mentions:
if isinstance(m, str):
mentions.append(m)
elif isinstance(m, dict):
mentions.append(m.get("slug", ""))
content = message.get("content", "")
author_type = message.get("author_type", "")
payload = {"type": event_type or WSEventType.MESSAGE_NEW, "data": message}
@ -109,9 +116,11 @@ class ConnectionManager:
continue
if client.chat_listen == ListenMode.NONE:
continue
# System messages: only if agent is mentioned in content
# System messages: only if agent is mentioned (by slug in mentions array, or @slug in content for compat)
if author_type == AuthorType.SYSTEM:
if f"@{client.member_slug}" not in content:
mentioned_in_mentions = client.member_slug in mentions
mentioned_in_content = f"@{client.member_slug}" in content
if not mentioned_in_mentions and not mentioned_in_content:
continue
await self.send_to_session(session_id, payload)
continue