Skip to content

Commit e99a258

Browse files
committed
format
1 parent b4da39d commit e99a258

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

eval_protocol/models.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -256,6 +256,7 @@ def __iter__(self):
256256

257257
class Message(BaseModel):
258258
"""Chat message model with trajectory evaluation support."""
259+
259260
role: str # assistant, user, system, tool
260261
content: Optional[Union[str, List[ChatCompletionContentPartTextParam]]] = Field(
261262
default="", description="The content of the message."
@@ -273,7 +274,7 @@ class Message(BaseModel):
273274
def dump_mdoel_for_chat_completion_request(self):
274275
"""Only keep chat completion accepted fields"""
275276
return self.model_dump(exclude_none=True, exclude={"control_plane_step", "weight", "reasoning_content"})
276-
277+
277278
@classmethod
278279
def model_validate(cls, obj, *args, **kwargs):
279280
if isinstance(obj, dict):

0 commit comments

Comments
 (0)