feat: import core schemas from hub-core

api/schemas/agent_message.py

@@ -1,30 +1,7 @@
-import uuid
-from datetime import datetime
+from hub_core.schemas.agent_message import MessageCreate, MessageRead, MessageReply
 
-from pydantic import BaseModel, ConfigDict
-
-
-class MessageCreate(BaseModel):
-    from_agent: str
-    to_agent: str
-    subject: str
-    body: str
-    thread_id: uuid.UUID | None = None
-
-
-class MessageReply(BaseModel):
-    from_agent: str
-    body: str
-
-
-class MessageRead(BaseModel):
-    model_config = ConfigDict(from_attributes=True)
-    id: uuid.UUID
-    from_agent: str
-    to_agent: str
-    subject: str
-    body: str
-    thread_id: uuid.UUID | None = None
-    read_at: datetime | None = None
-    archived_at: datetime | None = None
-    created_at: datetime
+__all__ = [
+    "MessageCreate",
+    "MessageRead",
+    "MessageReply",
+]