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]
|
content: list[Content]
|
||||||
|
|
||||||
|
|
||||||
@dataclass
|
@dataclass(frozen=True)
|
||||||
class Prompt:
|
class Prompt:
|
||||||
model: str
|
model: str
|
||||||
input: list[Message]
|
input: list[Message]
|
||||||
|
|
||||||
def add_message(self, message: Message):
|
|
||||||
self.input.append(message)
|
|
||||||
|
|
||||||
|
|
||||||
@dataclass(frozen=True)
|
@dataclass(frozen=True)
|
||||||
class Response:
|
class Response:
|
||||||
output: list[Message]
|
output: list[Message]
|
||||||
|
|
||||||
|
|
||||||
def filter_fields(data: dict, filter_by_class: dataclass) -> dict:
|
def filter_fields(data: dict, dataclass_type: dataclass) -> dict:
|
||||||
return {
|
return {
|
||||||
i_key: i_value
|
i_key: i_value
|
||||||
for i_key, i_value in data.items()
|
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