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

basic scoring function works

Browse source
This commit is contained in:
Xi Yan 2024-10-23 14:42:28 -07:00
parent 38e31ab525
commit 70c08e694d
5 changed files with 164 additions and 6 deletions
Download patch file Download diff file Expand all files Collapse all files

9
llama_stack/apis/scoring_functions/scoring_functions.py
View file

@ -34,8 +34,8 @@ class Parameter(BaseModel):
@json_schema_type
class CommonDef(BaseModel):
name: str
class CommonFunctionDef(BaseModel):
identifier: str
description: Optional[str] = None
metadata: Dict[str, Any] = Field(
default_factory=dict,
@ -46,10 +46,11 @@ class CommonDef(BaseModel):
@json_schema_type
class DeterministicFunctionDef(CommonDef):
class DeterministicFunctionDef(CommonFunctionDef):
type: Literal["deterministic"] = "deterministic"
parameters: List[Parameter] = Field(
description="List of parameters for the deterministic function",
default_factory=list,
)
return_type: ParamType = Field(
description="The return type of the deterministic function",
@ -58,7 +59,7 @@ class DeterministicFunctionDef(CommonDef):
@json_schema_type
class LLMJudgeFunctionDef(CommonDef):
class LLMJudgeFunctionDef(CommonFunctionDef):
type: Literal["judge"] = "judge"
model: str = Field(
description="The LLM model to use for the judge function",