phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-08-01 16:24:44 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

config rename

Browse source
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
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/providers/inline/scoring/basic/__init__.py
View file

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

4
llama_stack/providers/inline/scoring/basic/config.py
View file

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

4
llama_stack/providers/inline/scoring/basic/scoring.py
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.apis.inference.inference import Inference
from llama_stack.providers.datatypes import ScoringFunctionsProtocolPrivate 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.equality_scoring_fn import EqualityScoringFn
from .scoring_fn.llm_as_judge_scoring_fn import LlmAsJudgeScoringFn from .scoring_fn.llm_as_judge_scoring_fn import LlmAsJudgeScoringFn
from .scoring_fn.regex_parser_scoring_fn import RegexParserScoringFn from .scoring_fn.regex_parser_scoring_fn import RegexParserScoringFn
@ -28,7 +28,7 @@ LLM_JUDGE_FNS = [LlmAsJudgeScoringFn]
class MetaReferenceScoringImpl(Scoring, ScoringFunctionsProtocolPrivate): class MetaReferenceScoringImpl(Scoring, ScoringFunctionsProtocolPrivate):
def __init__( def __init__(
self, self,
config: MetaReferenceScoringConfig, config: BasicScoringConfig,
datasetio_api: DatasetIO, datasetio_api: DatasetIO,
datasets_api: Datasets, datasets_api: Datasets,
inference_api: Inference, inference_api: Inference,

2
llama_stack/providers/registry/scoring.py
View file

@ -16,7 +16,7 @@ def available_providers() -> List[ProviderSpec]:
provider_type="meta-reference", provider_type="meta-reference",
pip_packages=[], pip_packages=[],
module="llama_stack.providers.inline.scoring.basic", 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_dependencies=[
Api.datasetio, Api.datasetio,
Api.datasets, Api.datasets,