Skip to content

Commit daef898

Browse files
Adding cache delete method for model details (#989)
2 parents 8316571 + 838d543 commit daef898

File tree

2 files changed

+19
-1
lines changed

2 files changed

+19
-1
lines changed

ads/aqua/extension/model_handler.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -164,11 +164,13 @@ def put(self, id):
164164

165165
enable_finetuning = input_data.get("enable_finetuning")
166166
task = input_data.get("task")
167+
app=AquaModelApp()
167168
self.finish(
168-
AquaModelApp().edit_registered_model(
169+
app.edit_registered_model(
169170
id, inference_container, enable_finetuning, task
170171
)
171172
)
173+
app.clear_model_details_cache(model_id=id)
172174

173175

174176
class AquaModelLicenseHandler(AquaAPIhandler):

ads/aqua/model/model.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -718,6 +718,22 @@ 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(f"Clearing _service_model_details_cache for {model_id}")
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 = {"key": {"model_id": model_id}, "cache_deleted": True}
734+
735+
return res
736+
721737
@staticmethod
722738
def list_valid_inference_containers():
723739
containers = list(

0 commit comments

Comments
 (0)