format

Author: mayinghan

eval_protocol/models.py

@@ -257,7 +257,7 @@ def __iter__(self):
 
 class Message(BaseModel):
     """Chat message model with trajectory evaluation support."""
-    
+
     model_config = ConfigDict(extra="allow")
     role: str  # assistant, user, system, tool
     content: Optional[Union[str, List[ChatCompletionContentPartTextParam]]] = Field(
@@ -271,9 +271,12 @@ class Message(BaseModel):
     tool_calls: Optional[List[ChatCompletionMessageToolCall]] = None
     function_call: Optional[FunctionCall] = None
     control_plane_step: Optional[Dict[str, Any]] = None
+
     def dump_mdoel_for_chat_completion_request(self):
         """Only keep chat completion accepted fields"""
-        exclude_fields = {"control_plane_step", "reasoning_content"} | set(self.model_extra.keys()) if self.model_extra else set()
+        exclude_fields = (
+            {"control_plane_step", "reasoning_content"} | set(self.model_extra.keys()) if self.model_extra else set()
+        )
         return self.model_dump(exclude_none=True, exclude=exclude_fields)
 
     @classmethod