fix

2026-05-09 01:30:39 -06:00
parent 48e6ddeeac
commit 4660b51de3
4 changed files with 47 additions and 63 deletions
--- a/prompt/schemas.py
+++ b/prompt/schemas.py
@@ -47,6 +47,6 @@ class ConversationOut(Schema):
            "task_id": conv.task_id,
            "task_title": conv.task.title,
            "is_active": conv.is_active,
-            "message_count": conv.messages.count(),
+            "message_count": conv.msg_count if hasattr(conv, "msg_count") else conv.messages.count(),
            "created": conv.created.isoformat(),
        }