26 lines
531 B
Python
26 lines
531 B
Python
from __future__ import annotations
|
|
|
|
from uuid import UUID
|
|
|
|
from sqlmodel.ext.asyncio.session import AsyncSession
|
|
|
|
from app.models.activity_events import ActivityEvent
|
|
|
|
|
|
def record_activity(
|
|
session: AsyncSession,
|
|
*,
|
|
event_type: str,
|
|
message: str,
|
|
agent_id: UUID | None = None,
|
|
task_id: UUID | None = None,
|
|
) -> ActivityEvent:
|
|
event = ActivityEvent(
|
|
event_type=event_type,
|
|
message=message,
|
|
agent_id=agent_id,
|
|
task_id=task_id,
|
|
)
|
|
session.add(event)
|
|
return event
|