Skip to content

Commit f8a9548

Browse files
committed
initial the create_time to actual timestamp
1 parent 400c1e7 commit f8a9548

File tree

1 file changed

+9
-7
lines changed

1 file changed

+9
-7
lines changed

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

Lines changed: 9 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,9 @@
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
77
from 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
6567
class 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

7375
class SubTaskToPrint(BaseModel):

0 commit comments

Comments
 (0)