From 37442a495b3586f128a43bf1a449be5884bbb853 Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Tue, 22 Oct 2024 08:25:47 -0700 Subject: [PATCH] fix --- llama_stack/apis/datasets/__init__.py | 2 +- llama_stack/apis/datasets/datasets.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/llama_stack/apis/datasets/__init__.py b/llama_stack/apis/datasets/__init__.py index 33557a0ab..102b9927f 100644 --- a/llama_stack/apis/datasets/__init__.py +++ b/llama_stack/apis/datasets/__init__.py @@ -4,4 +4,4 @@ # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. -from .dataset import * # noqa: F401 F403 +from .datasets import * # noqa: F401 F403 diff --git a/llama_stack/apis/datasets/datasets.py b/llama_stack/apis/datasets/datasets.py index addc59952..9bb7e6f7f 100644 --- a/llama_stack/apis/datasets/datasets.py +++ b/llama_stack/apis/datasets/datasets.py @@ -4,13 +4,13 @@ # 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, Dict, Optional, Protocol +from typing import Any, Dict, List, Optional, Protocol from llama_models.llama3.api.datatypes import URL from llama_models.schema_utils import json_schema_type, webmethod -from pydantic import BaseModel +from pydantic import BaseModel, Field from llama_stack.apis.common.type_system import ParamType @@ -25,7 +25,7 @@ class DatasetDef(BaseModel): identifier: str = Field( description="A unique name for the dataset", ) - schema: DatasetSchema = Field( + dataset_schema: DatasetSchema = Field( description="The schema definition for this dataset", ) url: URL