File tree Expand file tree Collapse file tree 2 files changed +19
-1
lines changed Expand file tree Collapse file tree 2 files changed +19
-1
lines changed Original file line number Diff line number Diff 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
174176class AquaModelLicenseHandler (AquaAPIhandler ):
Original file line number Diff line number Diff 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 (
You can’t perform that action at this time.
0 commit comments