Skip to content

Commit 80c94bf

Browse files
committed
fix tox typecheck errors for utils
1 parent 109625c commit 80c94bf

File tree

1 file changed

+19
-3
lines changed
  • util/opentelemetry-util-genai/src/opentelemetry/util/genai

1 file changed

+19
-3
lines changed

util/opentelemetry-util-genai/src/opentelemetry/util/genai/types.py

Lines changed: 19 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,18 @@ class OutputMessage:
8181
finish_reason: Union[str, FinishReason]
8282

8383

84+
def _new_input_messages() -> List[InputMessage]:
85+
return []
86+
87+
88+
def _new_output_messages() -> List[OutputMessage]:
89+
return []
90+
91+
92+
def _new_str_any_dict() -> Dict[str, Any]:
93+
return {}
94+
95+
8496
@dataclass
8597
class LLMInvocation:
8698
"""
@@ -94,14 +106,18 @@ class LLMInvocation:
94106
span: Optional[Span] = None
95107
start_time: float = field(default_factory=time.time)
96108
end_time: Optional[float] = None
97-
input_messages: List[InputMessage] = field(default_factory=list)
98-
output_messages: List[OutputMessage] = field(default_factory=list)
109+
input_messages: List[InputMessage] = field(
110+
default_factory=_new_input_messages
111+
)
112+
output_messages: List[OutputMessage] = field(
113+
default_factory=_new_output_messages
114+
)
99115
provider: Optional[str] = None
100116
response_model_name: Optional[str] = None
101117
response_id: Optional[str] = None
102118
input_tokens: Optional[AttributeValue] = None
103119
output_tokens: Optional[AttributeValue] = None
104-
attributes: Dict[str, Any] = field(default_factory=dict)
120+
attributes: Dict[str, Any] = field(default_factory=_new_str_any_dict)
105121

106122

107123
@dataclass

0 commit comments

Comments
 (0)