Update validator.py

SLR722 · Jan 13, 2025 · 5a9a056 · 5a9a056
1 parent 9b4d330
commit 5a9a056
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/llama_stack/providers/inline/post_training/common/validator.py b/llama_stack/providers/inline/post_training/common/validator.py
@@ -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 @@
     ],
     "dialog": [
         {
-            ColumnName.dialog.value: StringType(),
+            ColumnName.dialog.value: DialogType(),
         }
     ],
 }