phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

chore: update return type to Optional[str] (#982)

Browse source
This commit is contained in:
Sébastien Han 2025-02-12 07:10:28 +01:00 • committed by GitHub
parent 66d7e15c93
commit bf11cc0450
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
1 changed files with 4 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

14
llama_stack/providers/utils/inference/model_registry.py
View file

@ -57,17 +57,11 @@ class ModelRegistryHelper(ModelsProtocolPrivate):
self.alias_to_provider_id_map[alias_obj.llama_model] = alias_obj.provider_model_id self.alias_to_provider_id_map[alias_obj.llama_model] = alias_obj.provider_model_id
self.provider_id_to_llama_model_map[alias_obj.provider_model_id] = alias_obj.llama_model self.provider_id_to_llama_model_map[alias_obj.provider_model_id] = alias_obj.llama_model
def get_provider_model_id(self, identifier: str) -> str: def get_provider_model_id(self, identifier: str) -> Optional[str]:
if identifier in self.alias_to_provider_id_map: return self.alias_to_provider_id_map.get(identifier, None)
return self.alias_to_provider_id_map[identifier]
else:
return None
def get_llama_model(self, provider_model_id: str) -> str: def get_llama_model(self, provider_model_id: str) -> Optional[str]:
if provider_model_id in self.provider_id_to_llama_model_map: return self.provider_id_to_llama_model_map.get(provider_model_id, None)
return self.provider_id_to_llama_model_map[provider_model_id]
else:
return None
async def register_model(self, model: Model) -> Model: async def register_model(self, model: Model) -> Model:
if model.model_type == ModelType.embedding: if model.model_type == ModelType.embedding: