forked from phoenix-oss/llama-stack-mirror
2aca7265b3
# What does this PR do? Change validation to TODO same as was done [here](https://github.com/meta-llama/llama-stack/blob/main/llama_stack/providers/inline/eval/meta_reference/eval.py#L87) until validation can be implemented Closes #1849 ## Test Plan Signed-off-by: Kevin <kpostlet@redhat.com>
36 lines
975 B
Python
36 lines
975 B
Python
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
# Copyright (c) Meta Platforms, IAny, nc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the terms described in the LICENSE file in
|
|
# the root directory of this source tree.
|
|
|
|
from typing import Any
|
|
|
|
from llama_stack.apis.common.type_system import (
|
|
ChatCompletionInputType,
|
|
DialogType,
|
|
StringType,
|
|
)
|
|
from llama_stack.providers.utils.common.data_schema_validator import (
|
|
ColumnName,
|
|
)
|
|
|
|
EXPECTED_DATASET_SCHEMA: dict[str, list[dict[str, Any]]] = {
|
|
"instruct": [
|
|
{
|
|
ColumnName.chat_completion_input.value: ChatCompletionInputType(),
|
|
ColumnName.expected_answer.value: StringType(),
|
|
}
|
|
],
|
|
"dialog": [
|
|
{
|
|
ColumnName.dialog.value: DialogType(),
|
|
}
|
|
],
|
|
}
|