Skip to content

Commit 9a6fe7a

Browse files
author
Dylan Huang
committed
Add optional status field to StatusResponse model in remote rollout processor
- Introduced an optional status indicator in the StatusResponse model to differentiate between successful and failed rollouts. - Updated documentation to clarify the purpose of the new status field for better understanding in the eval-protocol context.
1 parent c90758d commit 9a6fe7a

1 file changed

Lines changed: 7 additions & 1 deletion

File tree

eval_protocol/types/remote_rollout_processor.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44

55
from typing import Any, Dict, List, Optional
66
from pydantic import BaseModel, Field
7-
from eval_protocol.models import Message
7+
from eval_protocol.models import Message, Status
88

99

1010
class RolloutMetadata(BaseModel):
@@ -40,6 +40,12 @@ class StatusResponse(BaseModel):
4040
terminated: bool
4141
info: Optional[Dict[str, Any]] = None
4242

43+
status: Optional[Status] = None
44+
"""
45+
Optional status indicator for the rollout to be used by eval-protocol. This
46+
is useful to distinguish between successful and failed rollouts.
47+
"""
48+
4349

4450
def create_langfuse_config_tags(init_request: InitRequest) -> List[str]:
4551
"""Create Langfuse tags from InitRequest metadata."""

0 commit comments

Comments
 (0)