diff --git a/reginaldCog/llm_clients/openai_data_models.py b/reginaldCog/llm_clients/openai_data_models.py index 21ab7c8..171e7c4 100644 --- a/reginaldCog/llm_clients/openai_data_models.py +++ b/reginaldCog/llm_clients/openai_data_models.py @@ -23,8 +23,8 @@ class OutputText(Content): @dataclass(frozen=True) class InputImage(Content): type: str = field(init=False, default='input_image') - image_url: str = field(default=None) - file_id: str = field(default=None) + image_url: str | None = field(default=None) + file_id: str | None = field(default=None) def __post_init__(self): if self.image_url is None and self.file_id is None: