Merge pull request #156 from Chainlit/matt/eng-2178-allow-prompt-dele…

…tion feat: add warning messages when getting deleted promp
Chainlit · Jan 9, 2025 · 9608e6c · 9608e6c
2 parents 843de0c + c9f7514
commit 9608e6c
Show file tree

Hide file tree

Showing 2 changed files with 40 additions and 3 deletions.
diff --git a/literalai/api/helpers/gql.py b/literalai/api/helpers/gql.py
@@ -989,6 +989,7 @@
     ) {
       id
       name
+      deletedAt
     }
   }"""
 
@@ -999,6 +1000,7 @@
         name: $name
     ) {
         id
+        deletedAt
     }
 }"""
 
@@ -1023,6 +1025,7 @@
       id
       lineage {
         name
+        deletedAt
       }
       version
       createdAt
@@ -1083,6 +1086,7 @@
         version
         lineage {
             name
+            deletedAt
         }
     }
 }

diff --git a/literalai/api/helpers/prompt_helpers.py b/literalai/api/helpers/prompt_helpers.py
@@ -1,3 +1,4 @@
+import logging
 from typing import TYPE_CHECKING, Optional, TypedDict, Callable
 
 from literalai.observability.generation import GenerationMessage
@@ -12,11 +13,18 @@
 from literalai.api.helpers import gql
 
 
+logger = logging.getLogger(__name__)
+
+
 def create_prompt_lineage_helper(name: str, description: Optional[str] = None):
     variables = {"name": name, "description": description}
 
     def process_response(response):
         prompt = response["data"]["createPromptLineage"]
+        if prompt and prompt.get("deletedAt"):
+            logger.warning(
+                f"Prompt {name} was deleted - please update any references to use an active prompt in production"
+            )
         return prompt
 
     description = "create prompt lineage"
@@ -29,6 +37,10 @@ def get_prompt_lineage_helper(name: str):
 
     def process_response(response):
         prompt = response["data"]["promptLineage"]
+        if prompt and prompt.get("deletedAt"):
+            logger.warning(
+                f"Prompt {name} was deleted - please update any references to use an active prompt in production"
+            )
         return prompt
 
     description = "get prompt lineage"
@@ -52,15 +64,23 @@ def create_prompt_helper(
 
     def process_response(response):
         prompt = response["data"]["createPromptVersion"]
+        prompt_lineage = prompt.get("lineage")
+
+        if prompt_lineage and prompt_lineage.get("deletedAt"):
+            logger.warning(
+                f"Prompt {prompt_lineage.get('name')} was deleted - please update any references to use an active prompt in production"
+            )
         return Prompt.from_dict(api, prompt) if prompt else None
 
     description = "create prompt version"
 
     return gql.CREATE_PROMPT_VERSION, description, variables, process_response
 
 
-def get_prompt_cache_key(id: Optional[str], name: Optional[str], version: Optional[int]) -> str:
-    if id:  
+def get_prompt_cache_key(
+    id: Optional[str], name: Optional[str], version: Optional[int]
+) -> str:
+    if id:
         return id
     elif name and version:
         return f"{name}-{version}"
@@ -90,14 +110,27 @@ def get_prompt_helper(
 
     def process_response(response):
         prompt_version = response["data"]["promptVersion"]
+        prompt_lineage = prompt_version.get("lineage")
+
+        if prompt_lineage and prompt_lineage.get("deletedAt"):
+            logger.warning(
+                f"Prompt {name} was deleted - please update any references to use an active prompt in production"
+            )
         prompt = Prompt.from_dict(api, prompt_version) if prompt_version else None
         if cache and prompt:
             put_prompt(cache, prompt)
         return prompt
 
     description = "get prompt"
 
-    return gql.GET_PROMPT_VERSION, description, variables, process_response, timeout, cached_prompt
+    return (
+        gql.GET_PROMPT_VERSION,
+        description,
+        variables,
+        process_response,
+        timeout,
+        cached_prompt,
+    )
 
 
 def create_prompt_variant_helper(