From 9209cc568d1340820ac4d9b654b4f44a042a3129 Mon Sep 17 00:00:00 2001 From: T-BENZIN Date: Fri, 18 Jul 2025 14:28:23 +0500 Subject: [PATCH] Made Prompt dataclass immutable and got rid out of add_message method. openai_client\models.py's test run would remain broken for now --- reginaldCog/openai_client/models.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/reginaldCog/openai_client/models.py b/reginaldCog/openai_client/models.py index 200b98f..3ec01e3 100644 --- a/reginaldCog/openai_client/models.py +++ b/reginaldCog/openai_client/models.py @@ -60,25 +60,22 @@ class Message: content: list[Content] -@dataclass +@dataclass(frozen=True) class Prompt: model: str input: list[Message] - def add_message(self, message: Message): - self.input.append(message) - @dataclass(frozen=True) class Response: output: list[Message] -def filter_fields(data: dict, filter_by_class: dataclass) -> dict: +def filter_fields(data: dict, dataclass_type: dataclass) -> dict: return { i_key: i_value for i_key, i_value in data.items() - if i_key in tuple(f.name for f in fields(filter_by_class)) + if i_key in tuple(i_field.name for i_field in fields(dataclass_type)) }