phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-04 04:04:14 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

reward scoring model enum

Browse source
This commit is contained in:
Raghotham Murthy 2024-07-10 21:59:01 -07:00
parent ebb59aa35f
commit 6ec7c47938
4 changed files with 24 additions and 17 deletions
Download patch file Download diff file Expand all files Collapse all files

5
source/api_definitions.py
View file

@ -29,6 +29,7 @@ from model_types import (
InstructModel,
Message,
PretrainedModel,
RewardModel,
SamplingParams,
ShieldConfig,
StopReason,
@ -245,9 +246,7 @@ class RewardScoringRequest:
"""Request to score a reward function. A list of prompts and a list of responses per prompt."""
prompt_generations: List[KPromptGenerations]
# TODO(ragho): create a RewardModel enum tye
model: str
model: RewardModel
@json_schema_type