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

more scoring function for rag

Browse source
This commit is contained in:
Xi Yan 2024-12-19 16:53:39 -08:00
parent b94ab8d013
commit 9aa4a405ca
8 changed files with 132 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/providers/utils/common/data_schema_validator_mixin.py
View file

@ -34,11 +34,11 @@ class DataSchemaValidatorMixin:
dataset_schema, self.get_expected_schema_for_eval()
)
def validate_row_schema_for_scoring(self, row_schema: Dict[str, Any]):
self.validate_row_schema(row_schema, self.get_expected_schema_for_scoring())
def validate_row_schema_for_scoring(self, input_row: Dict[str, Any]):
self.validate_row_schema(input_row, self.get_expected_schema_for_scoring())
def validate_row_schema_for_eval(self, row_schema: Dict[str, Any]):
self.validate_row_schema(row_schema, self.get_expected_schema_for_eval())
def validate_row_schema_for_eval(self, input_row: Dict[str, Any]):
self.validate_row_schema(input_row, self.get_expected_schema_for_eval())
def get_expected_schema_for_scoring(self):
return [