support adding alias for models without hf repo/sku entry

This commit is contained in:
Dinesh Yeduguru 2024-11-18 23:31:04 -08:00
parent fcc2132e6f
commit 8bd0a33206
2 changed files with 18 additions and 7 deletions

View file

@ -36,6 +36,16 @@ def build_model_alias(provider_model_id: str, model_descriptor: str) -> ModelAli
)
def build_model_alias_with_just_llama_model(
provider_model_id: str, model_descriptor: str
) -> ModelAlias:
return ModelAlias(
provider_model_id=provider_model_id,
aliases=[],
llama_model=model_descriptor,
)
class ModelRegistryHelper(ModelsProtocolPrivate):
def __init__(self, model_aliases: List[ModelAlias]):
self.alias_to_provider_id_map = {}