33import uuid
44from enum import Enum
55from typing import Any , Optional , Literal
6- from dataclasses import dataclass
6+ from dataclasses import dataclass , field
7+ from datetime import datetime , timezone
78from pydantic import BaseModel , Field
89
910
11+ def _get_utc_now_iso ():
12+ return datetime .now (timezone .utc ).isoformat ()
13+
14+
1015@dataclass
1116class MockFileBase :
1217 filename : str
@@ -15,8 +20,8 @@ class MockFileBase:
1520 storage_path : str
1621 size : int = - 1
1722 storage_type : str = "unknown"
18- create_time : str = "xxxyyy"
19- update_time : str = "xxxyyy"
23+ create_time : str = field ( default_factory = _get_utc_now_iso )
24+ update_time : str = field ( default_factory = _get_utc_now_iso )
2025 user_id : uuid .UUID = uuid .uuid4 ()
2126
2227
@@ -62,12 +67,13 @@ class UserMessage(BaseMessage):
6267 name : str = "User"
6368
6469
70+ @dataclass
6571class MockMessage :
66- id : uuid .UUID = uuid .uuid4 ( )
72+ id : uuid .UUID = field ( default_factory = uuid .uuid4 )
6773 message : Optional [dict ] = None
68- files : list [Any ] = []
69- create_time : str = "xxxyyy"
70- update_time : str = "xxxyyy"
74+ files : list [Any ] = field ( default_factory = list )
75+ create_time : str = field ( default_factory = _get_utc_now_iso )
76+ update_time : str = field ( default_factory = _get_utc_now_iso )
7177
7278
7379class SubTaskToPrint (BaseModel ):
0 commit comments