Skip to content

Commit 149f20f

Browse files
authored
Add default types for Log() object (#125)
1 parent b6e5f4d commit 149f20f

File tree

1 file changed

+7
-7
lines changed

1 file changed

+7
-7
lines changed

quotientai/resources/logs.py

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,7 @@
77
import uuid
88

99
from collections import deque
10-
from dataclasses import dataclass
10+
from dataclasses import dataclass, field
1111
from datetime import datetime, timezone
1212
from threading import Thread, Event
1313
from typing import Any, Dict, List, Optional, Union
@@ -48,12 +48,12 @@ class Log:
4848
hallucination_detection: bool
4949
inconsistency_detection: bool
5050
user_query: str
51-
model_output: str
52-
documents: Optional[List[Union[str, LogDocument]]]
53-
message_history: Optional[List[Dict[str, Any]]]
54-
instructions: Optional[List[str]]
55-
tags: Dict[str, Any]
56-
created_at: datetime
51+
model_output: Optional[str] = None
52+
documents: Optional[List[Union[str, LogDocument]]] = field(default_factory=list)
53+
message_history: Optional[List[Dict[str, Any]]] = field(default_factory=list)
54+
instructions: Optional[List[str]] = field(default_factory=list)
55+
tags: Dict[str, Any] = field(default_factory=dict)
56+
created_at: datetime = field(default_factory=lambda: datetime.now(timezone.utc))
5757
status: Optional[LogStatus] = None
5858
updated_at: Optional[datetime] = None
5959
has_hallucination: Optional[bool] = None

0 commit comments

Comments
 (0)