extract score regex to llm context

This commit is contained in:
Xi Yan 2024-10-25 17:03:01 -07:00
parent 247a53d393
commit 9b410a87bf
3 changed files with 13 additions and 7 deletions

View file

@ -26,6 +26,10 @@ class Parameter(BaseModel):
class LLMAsJudgeContext(BaseModel):
judge_model: str
prompt_template: Optional[str] = None
judge_score_regex: Optional[List[str]] = Field(
description="Regex to extract the score from the judge response",
default=None,
)
@json_schema_type