config rename

This commit is contained in:
Xi Yan 2024-11-11 22:41:15 -05:00
parent 8dee5faf0d
commit ecf1932424
4 changed files with 7 additions and 7 deletions

View file

@ -7,11 +7,11 @@ from typing import Dict
from llama_stack.distribution.datatypes import Api, ProviderSpec
from .config import MetaReferenceScoringConfig
from .config import BasicScoringConfig
async def get_provider_impl(
config: MetaReferenceScoringConfig,
config: BasicScoringConfig,
deps: Dict[Api, ProviderSpec],
):
from .scoring import MetaReferenceScoringImpl

View file

@ -3,7 +3,7 @@
#
# 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.scoring import * # noqa: F401, F403
from pydantic import BaseModel
class MetaReferenceScoringConfig(BaseModel): ...
class BasicScoringConfig(BaseModel): ...

View file

@ -14,7 +14,7 @@ from llama_stack.apis.datasets import * # noqa: F403
from llama_stack.apis.inference.inference import Inference
from llama_stack.providers.datatypes import ScoringFunctionsProtocolPrivate
from .config import MetaReferenceScoringConfig
from .config import BasicScoringConfig
from .scoring_fn.equality_scoring_fn import EqualityScoringFn
from .scoring_fn.llm_as_judge_scoring_fn import LlmAsJudgeScoringFn
from .scoring_fn.regex_parser_scoring_fn import RegexParserScoringFn
@ -28,7 +28,7 @@ LLM_JUDGE_FNS = [LlmAsJudgeScoringFn]
class MetaReferenceScoringImpl(Scoring, ScoringFunctionsProtocolPrivate):
def __init__(
self,
config: MetaReferenceScoringConfig,
config: BasicScoringConfig,
datasetio_api: DatasetIO,
datasets_api: Datasets,
inference_api: Inference,

View file

@ -16,7 +16,7 @@ def available_providers() -> List[ProviderSpec]:
provider_type="meta-reference",
pip_packages=[],
module="llama_stack.providers.inline.scoring.basic",
config_class="llama_stack.providers.inline.scoring.basic.MetaReferenceScoringConfig",
config_class="llama_stack.providers.inline.scoring.basic.BasicScoringConfig",
api_dependencies=[
Api.datasetio,
Api.datasets,