Skip to content

Commit df0776c

Browse files
authored
initial the create_time to actual timestamp (#113)
1 parent 400c1e7 commit df0776c

File tree

1 file changed

+13
-7
lines changed

1 file changed

+13
-7
lines changed

alias/src/alias/agent/mock/mock_message_models.py

Lines changed: 13 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -3,10 +3,15 @@
33
import uuid
44
from enum import Enum
55
from typing import Any, Optional, Literal
6-
from dataclasses import dataclass
6+
from dataclasses import dataclass, field
7+
from datetime import datetime, timezone
78
from pydantic import BaseModel, Field
89

910

11+
def _get_utc_now_iso():
12+
return datetime.now(timezone.utc).isoformat()
13+
14+
1015
@dataclass
1116
class 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
6571
class 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

7379
class SubTaskToPrint(BaseModel):

0 commit comments

Comments
 (0)