This commit is contained in:
Botao Chen 2025-01-10 14:32:32 -08:00
parent 1e915d87fb
commit a3b9bc29c8
3 changed files with 6 additions and 2 deletions

View file

@ -9,7 +9,7 @@
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
from llama_stack.apis.common.type_system import StringType
from llama_stack.apis.common.type_system import ChatCompletionInputType, StringType
from llama_stack.apis.datasets import Datasets
from llama_stack.providers.utils.common.data_schema_validator import (
ColumnName,
@ -19,7 +19,7 @@ from llama_stack.providers.utils.common.data_schema_validator import (
EXPECTED_DATASET_SCHEMA = {
"instruct": [
{
ColumnName.chat_completion_input.value: StringType(),
ColumnName.chat_completion_input.value: ChatCompletionInputType(),
ColumnName.expected_answer.value: StringType(),
}
],