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

new cols in the explicit schema too

Browse source
This commit is contained in:
Yang Yang 2025-03-12 17:18:01 -07:00
parent 1085aaedb8
commit 0c07811f8e
1 changed files with 18 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

18
llama_stack/providers/utils/common/data_schema_validator.py
View file

@ -41,6 +41,15 @@ VALID_SCHEMAS_FOR_SCORING = [
ColumnName.generated_answer.value: StringType(), ColumnName.generated_answer.value: StringType(),
ColumnName.context.value: StringType(), ColumnName.context.value: StringType(),
}, },
{
ColumnName.input_query.value: StringType(),
ColumnName.expected_answer.value: StringType(),
ColumnName.generated_answer.value: StringType(),
ColumnName.function.value: StringType(),
ColumnName.language.value: StringType(),
ColumnName.id.value: StringType(),
ColumnName.ground_truth.value: StringType(),
},
] ]
VALID_SCHEMAS_FOR_EVAL = [ VALID_SCHEMAS_FOR_EVAL = [
@ -54,6 +63,15 @@ VALID_SCHEMAS_FOR_EVAL = [
ColumnName.expected_answer.value: StringType(), ColumnName.expected_answer.value: StringType(),
ColumnName.completion_input.value: CompletionInputType(), ColumnName.completion_input.value: CompletionInputType(),
}, },
{
ColumnName.input_query.value: StringType(),
ColumnName.expected_answer.value: StringType(),
ColumnName.generated_answer.value: StringType(),
ColumnName.function.value: StringType(),
ColumnName.language.value: StringType(),
ColumnName.id.value: StringType(),
ColumnName.ground_truth.value: StringType(),
},
] ]