Merge pull request #46 from joerunde/purge-logs

🔊 add debug logs for prompt unload
caikit · Nov 30, 2023 · 94f63ba · 94f63ba
2 parents 51e9cac + d015b0e
commit 94f63ba
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/caikit_tgis_backend/tgis_backend.py b/caikit_tgis_backend/tgis_backend.py
@@ -250,6 +250,11 @@ def unload_prompt_artifacts(self, model_id: str, *prompt_ids: str):
         error.value_check(
             "<TGB99822514E>", conn is not None, "Unknown model {}", model_id
         )
+        log.debug3(
+            "Unloading prompt artifacts for model: %s, prompts: %s",
+            model_id,
+            prompt_ids,
+        )
         conn.unload_prompt_artifacts(*prompt_ids)
 
     @property

diff --git a/caikit_tgis_backend/tgis_connection.py b/caikit_tgis_backend/tgis_connection.py
@@ -250,6 +250,9 @@ def unload_prompt_artifacts(self, *prompt_ids: str):
         )
         for prompt_id in prompt_ids:
             prompt_id_dir = os.path.join(self.prompt_dir, prompt_id)
+            log.debug3(
+                "Unloading prompt artifact %s at path %s", prompt_id, prompt_id_dir
+            )
             shutil.rmtree(prompt_id_dir, ignore_errors=True)
 
     def get_client(self) -> generation_pb2_grpc.GenerationServiceStub: