registry refactor

This commit is contained in:
Xi Yan 2024-10-14 00:12:46 -07:00
parent 78cb88c3c4
commit 18fe966e96
9 changed files with 60 additions and 106 deletions

View file

@ -0,0 +1,12 @@
# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# 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.evals import * # noqa: F403
from ..registry import Registry
class GeneratorProcessorRegistry(Registry[BaseGeneratorProcessor]):
_REGISTRY: Dict[str, BaseGeneratorProcessor] = {}