33import uuid
44from enum import Enum
55from typing import Any , Optional , Literal
6- from dataclasses import dataclass
6+ from dataclasses import dataclass , field
77from pydantic import BaseModel , Field
8+ from datetime import datetime , timezone
89
910
1011@dataclass
@@ -15,8 +16,8 @@ class MockFileBase:
1516 storage_path : str
1617 size : int = - 1
1718 storage_type : str = "unknown"
18- create_time : str = "xxxyyy"
19- update_time : str = "xxxyyy"
19+ create_time : str = field ( default_factory = lambda : datetime . now ( timezone . utc ). isoformat ())
20+ update_time : str = field ( default_factory = lambda : datetime . now ( timezone . utc ). isoformat ())
2021 user_id : uuid .UUID = uuid .uuid4 ()
2122
2223
@@ -62,12 +63,13 @@ class UserMessage(BaseMessage):
6263 name : str = "User"
6364
6465
66+ @dataclass
6567class MockMessage :
66- id : uuid .UUID = uuid .uuid4 ( )
68+ id : uuid .UUID = field ( default_factory = uuid .uuid4 )
6769 message : Optional [dict ] = None
68- files : list [Any ] = []
69- create_time : str = "xxxyyy"
70- update_time : str = "xxxyyy"
70+ files : list [Any ] = field ( default_factory = list )
71+ create_time : str = field ( default_factory = lambda : datetime . now ( timezone . utc ). isoformat ())
72+ update_time : str = field ( default_factory = lambda : datetime . now ( timezone . utc ). isoformat ())
7173
7274
7375class SubTaskToPrint (BaseModel ):
0 commit comments