phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2026-01-02 17:14:30 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

address_comment

Browse source
This commit is contained in:
Botao Chen 2025-01-12 22:52:53 -08:00
parent a3b9bc29c8
commit af1ba00227
2 changed files with 11 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

8
llama_stack/providers/inline/post_training/common/validator.py
View file

@ -9,7 +9,11 @@
#
# 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 ChatCompletionInputType, StringType
from llama_stack.apis.common.type_system import (
ChatCompletionInputType,
DialogType,
StringType,
)
from llama_stack.apis.datasets import Datasets
from llama_stack.providers.utils.common.data_schema_validator import (
ColumnName,
@ -25,7 +29,7 @@ EXPECTED_DATASET_SCHEMA = {
],
"dialog": [
{
ColumnName.dialog.value: StringType(),
ColumnName.dialog.value: DialogType(),
}
],
}