Made Prompt dataclass immutable and got rid out of add_message method. openai_client\models.py's test run would remain broken for now
This commit is contained in:
parent
7056315d5f
commit
9209cc568d
@ -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))
|
||||
}
|
||||
|
||||
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user