feat: improve wording for deleted prompt lineage

Chainlit · Dec 16, 2024 · accecf0 · accecf0
1 parent ffa6053
commit accecf0
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/literalai/api/helpers/prompt_helpers.py b/literalai/api/helpers/prompt_helpers.py
@@ -22,7 +22,7 @@ def create_prompt_lineage_helper(name: str, description: Optional[str] = None):
     def process_response(response):
         prompt = response["data"]["createPromptLineage"]
         if prompt and prompt.get("deletedAt"):
-            logger.warn("This prompt lineage is part of a deleted lineage")
+            logger.warn(f"Prompt lineage {name} was deleted")
         return prompt
 
     description = "create prompt lineage"
@@ -36,7 +36,7 @@ def get_prompt_lineage_helper(name: str):
     def process_response(response):
         prompt = response["data"]["promptLineage"]
         if prompt and prompt.get("deletedAt"):
-            logger.warn("This prompt lineage is part of a deleted lineage")
+            logger.warn(f"Prompt lineage {name} was deleted")
         return prompt
 
     description = "get prompt lineage"
@@ -63,7 +63,7 @@ def process_response(response):
         prompt_lineage = prompt.get("lineage")
 
         if prompt_lineage and prompt_lineage.get("deletedAt"):
-            logger.warn("This prompt version is part of a deleted lineage")
+            logger.warn(f"Prompt version is part of a deleted lineage {prompt_lineage.get('name')}")
         return Prompt.from_dict(api, prompt) if prompt else None
 
     description = "create prompt version"
@@ -105,7 +105,7 @@ def process_response(response):
         prompt_lineage = prompt_version.get("lineage")
 
         if prompt_lineage and prompt_lineage.get("deletedAt"):
-            logger.warn("This prompt version is part of a deleted lineage")
+            logger.warn(f"Prompt version is part of a deleted lineage {prompt_lineage.get('name')}")
         prompt = Prompt.from_dict(api, prompt_version) if prompt_version else None
         if cache and prompt:
             put_prompt(cache, prompt)