Skip to content

Commit 74c9bf3

Browse files
Adding delete model details cache feature
1 parent 8316571 commit 74c9bf3

File tree

2 files changed

+22
-0
lines changed

2 files changed

+22
-0
lines changed

ads/aqua/extension/model_handler.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -169,6 +169,7 @@ def put(self, id):
169169
id, inference_container, enable_finetuning, task
170170
)
171171
)
172+
AquaModelApp().clear_model_details_cache(model_id=id)
172173

173174

174175
class AquaModelLicenseHandler(AquaAPIhandler):

ads/aqua/model/model.py

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -718,6 +718,27 @@ def clear_model_list_cache(
718718
}
719719
return res
720720

721+
def clear_model_details_cache(self,model_id):
722+
"""
723+
Allows user to clear model details cache item
724+
Returns
725+
-------
726+
dict with the key used, and True if cache has the key that needs to be deleted.
727+
"""
728+
res={}
729+
logger.info("Clearing _service_model_details_cache")
730+
with self._cache_lock:
731+
if model_id in self._service_model_details_cache:
732+
self._service_model_details_cache.pop(key=model_id)
733+
res={
734+
"key":{
735+
"model_id":model_id
736+
},
737+
"cache_deleted":True
738+
}
739+
740+
return res
741+
721742
@staticmethod
722743
def list_valid_inference_containers():
723744
containers = list(

0 commit comments

Comments
 (0)