mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-15 12:42:42 +00:00
access control unit test
This commit is contained in:
parent
ea0d342c5d
commit
54354257dc
2 changed files with 25 additions and 30 deletions
|
|
@ -435,7 +435,8 @@ class OpenAIMixin(NeedsRequestProviderData, ABC, BaseModel):
|
|||
"""
|
||||
# First check if the model is pre-registered in the model store
|
||||
if hasattr(self, "model_store") and self.model_store:
|
||||
if await self.model_store.has_model(model):
|
||||
qualified_model = f"{self.__provider_id__}/{model}"
|
||||
if await self.model_store.has_model(qualified_model):
|
||||
return True
|
||||
|
||||
# Then check the provider's dynamic model cache
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue