From c6d3fc6fb617c3ed5772fd397835d6d71279869d Mon Sep 17 00:00:00 2001 From: Xi Yan Date: Thu, 26 Dec 2024 17:00:56 -0800 Subject: [PATCH] datatypes --- llama_stack/distribution/datatypes.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/llama_stack/distribution/datatypes.py b/llama_stack/distribution/datatypes.py index f2dea6012..dec62bfae 100644 --- a/llama_stack/distribution/datatypes.py +++ b/llama_stack/distribution/datatypes.py @@ -4,24 +4,24 @@ # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. -from typing import Dict, List, Optional, Union +from typing import Annotated, Any, Dict, List, Optional, Union from pydantic import BaseModel, Field from llama_stack.apis.datasetio import DatasetIO -from llama_stack.apis.datasets import * # noqa: F403 +from llama_stack.apis.datasets import Dataset, DatasetInput from llama_stack.apis.eval import Eval -from llama_stack.apis.eval_tasks import EvalTaskInput +from llama_stack.apis.eval_tasks import EvalTask, EvalTaskInput from llama_stack.apis.inference import Inference from llama_stack.apis.memory import Memory -from llama_stack.apis.memory_banks import * # noqa: F403 -from llama_stack.apis.models import * # noqa: F403 +from llama_stack.apis.memory_banks import MemoryBank, MemoryBankInput +from llama_stack.apis.models import Model, ModelInput from llama_stack.apis.safety import Safety from llama_stack.apis.scoring import Scoring -from llama_stack.apis.scoring_functions import * # noqa: F403 -from llama_stack.apis.shields import * # noqa: F403 +from llama_stack.apis.scoring_functions import ScoringFn, ScoringFnInput +from llama_stack.apis.shields import Shield, ShieldInput from llama_stack.apis.tools import Tool, ToolGroup, ToolRuntime -from llama_stack.providers.datatypes import * # noqa: F403 +from llama_stack.providers.datatypes import Api, ProviderSpec from llama_stack.providers.utils.kvstore.config import KVStoreConfig LLAMA_STACK_BUILD_CONFIG_VERSION = "2"