phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-05 18:22:41 +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

5
llama_stack/apis/common/type_system.py
View file

@ -54,6 +54,11 @@ class AgentTurnInputType(BaseModel):
type: Literal["agent_turn_input"] = "agent_turn_input" type: Literal["agent_turn_input"] = "agent_turn_input"
class DialogType(BaseModel):
# expects List[Message] for messages
type: Literal["dialog"] = "dialog"
ParamType = register_schema( ParamType = register_schema(
Annotated[ Annotated[
Union[ Union[

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 # This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree. # 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.apis.datasets import Datasets
from llama_stack.providers.utils.common.data_schema_validator import ( from llama_stack.providers.utils.common.data_schema_validator import (
ColumnName, ColumnName,
@ -25,7 +29,7 @@ EXPECTED_DATASET_SCHEMA = {
], ],
"dialog": [ "dialog": [
{ {
ColumnName.dialog.value: StringType(), ColumnName.dialog.value: DialogType(),
} }
], ],
} }