This commit is contained in:
Ashwin Bharambe 2025-10-16 06:43:17 -07:00
parent 54354257dc
commit bbd40d8cc4
3 changed files with 3 additions and 6 deletions

View file

@ -102,9 +102,6 @@ class DiskDistributionRegistry(DistributionRegistry):
"Unregister it first if you want to replace it."
)
if "sentence-transformers/sentence-transformers" in obj.identifier:
raise Exception("OMG")
await self.kvstore.set(
KEY_FORMAT.format(type=obj.type, identifier=obj.identifier),
obj.model_dump_json(),

View file

@ -435,7 +435,7 @@ 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:
qualified_model = f"{self.__provider_id__}/{model}"
qualified_model = f"{self.__provider_id__}/{model}" # type: ignore[attr-defined]
if await self.model_store.has_model(qualified_model):
return True