add

Author: mayinghan

eval_protocol/models.py

@@ -256,7 +256,6 @@ def __iter__(self):
 
 class Message(BaseModel):
     """Chat message model with trajectory evaluation support."""
-
     role: str  # assistant, user, system, tool
     content: Optional[Union[str, List[ChatCompletionContentPartTextParam]]] = Field(
         default="", description="The content of the message."
@@ -269,7 +268,12 @@ class Message(BaseModel):
     tool_calls: Optional[List[ChatCompletionMessageToolCall]] = None
     function_call: Optional[FunctionCall] = None
     control_plane_step: Optional[Dict[str, Any]] = None
+    weight: Optional[int] = None
 
+    def dump_mdoel_for_chat_completion_request(self):
+        """Only keep chat completion accepted fields"""
+        return self.model_dump(exclude_none=True, exclude={"control_plane_step", "weight", "reasoning_content"})
+        
     @classmethod
     def model_validate(cls, obj, *args, **kwargs):
         if isinstance(obj, dict):