mirror of
https://github.com/meta-llama/llama-stack.git
synced 2026-01-01 22:10:00 +00:00
refine
This commit is contained in:
parent
1e915d87fb
commit
a3b9bc29c8
3 changed files with 6 additions and 2 deletions
|
|
@ -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(),
|
||||
}
|
||||
],
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue