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",
+]

api/schemas/doi.py

@@ -1,29 +1,7 @@
-from pydantic import BaseModel
+from hub_core.schemas.doi import DoICriterion, DoIReport, DoISummaryEntry
 
-
-class DoICriterion(BaseModel):
-    id: str
-    label: str
-    tier: str
-    status: str   # pass | fail | warn | skip
-    detail: str = ""
-
-
-class DoIReport(BaseModel):
-    repo_slug: str
-    tier: str     # none | core | standard | full
-    core_pass: bool
-    standard_pass: bool
-    full_pass: bool
-    criteria: list[DoICriterion] = []
-    checked_at: str
-
-
-class DoISummaryEntry(BaseModel):
-    repo_slug: str
-    domain_slug: str | None
-    tier: str
-    core_pass: bool
-    standard_pass: bool
-    full_pass: bool
-    checked_at: str
+__all__ = [
+    "DoICriterion",
+    "DoIReport",
+    "DoISummaryEntry",
+]