initial the create_time to actual timestamp (#113)

This commit is contained in:
Osier-Yi
2026-01-30 17:32:26 +08:00
committed by GitHub
parent 400c1e77bf
commit df0776cac4

View File

@@ -3,10 +3,15 @@
import uuid import uuid
from enum import Enum from enum import Enum
from typing import Any, Optional, Literal from typing import Any, Optional, Literal
from dataclasses import dataclass from dataclasses import dataclass, field
from datetime import datetime, timezone
from pydantic import BaseModel, Field from pydantic import BaseModel, Field
def _get_utc_now_iso():
return datetime.now(timezone.utc).isoformat()
@dataclass @dataclass
class MockFileBase: class MockFileBase:
filename: str filename: str
@@ -15,8 +20,8 @@ class MockFileBase:
storage_path: str storage_path: str
size: int = -1 size: int = -1
storage_type: str = "unknown" storage_type: str = "unknown"
create_time: str = "xxxyyy" create_time: str = field(default_factory=_get_utc_now_iso)
update_time: str = "xxxyyy" update_time: str = field(default_factory=_get_utc_now_iso)
user_id: uuid.UUID = uuid.uuid4() user_id: uuid.UUID = uuid.uuid4()
@@ -62,12 +67,13 @@ class UserMessage(BaseMessage):
name: str = "User" name: str = "User"
@dataclass
class MockMessage: class MockMessage:
id: uuid.UUID = uuid.uuid4() id: uuid.UUID = field(default_factory=uuid.uuid4)
message: Optional[dict] = None message: Optional[dict] = None
files: list[Any] = [] files: list[Any] = field(default_factory=list)
create_time: str = "xxxyyy" create_time: str = field(default_factory=_get_utc_now_iso)
update_time: str = "xxxyyy" update_time: str = field(default_factory=_get_utc_now_iso)
class SubTaskToPrint(BaseModel): class SubTaskToPrint(BaseModel):