@@ -22,7 +22,7 @@ def create_prompt_lineage_helper(name: str, description: Optional[str] = None):
2222 def process_response (response ):
2323 prompt = response ["data" ]["createPromptLineage" ]
2424 if prompt and prompt .get ("deletedAt" ):
25- logger .warn (
25+ logger .warning (
2626 f"Prompt { name } was deleted - please update any references to use an active prompt in production"
2727 )
2828 return prompt
@@ -38,7 +38,7 @@ def get_prompt_lineage_helper(name: str):
3838 def process_response (response ):
3939 prompt = response ["data" ]["promptLineage" ]
4040 if prompt and prompt .get ("deletedAt" ):
41- logger .warn (
41+ logger .warning (
4242 f"Prompt { name } was deleted - please update any references to use an active prompt in production"
4343 )
4444 return prompt
@@ -67,8 +67,8 @@ def process_response(response):
6767 prompt_lineage = prompt .get ("lineage" )
6868
6969 if prompt_lineage and prompt_lineage .get ("deletedAt" ):
70- logger .warn (
71- f"Prompt { name } was deleted - please update any references to use an active prompt in production"
70+ logger .warning (
71+ f"Prompt { prompt_lineage . name } was deleted - please update any references to use an active prompt in production"
7272 )
7373 return Prompt .from_dict (api , prompt ) if prompt else None
7474
@@ -77,8 +77,10 @@ def process_response(response):
7777 return gql .CREATE_PROMPT_VERSION , description , variables , process_response
7878
7979
80- def get_prompt_cache_key (id : Optional [str ], name : Optional [str ], version : Optional [int ]) -> str :
81- if id :
80+ def get_prompt_cache_key (
81+ id : Optional [str ], name : Optional [str ], version : Optional [int ]
82+ ) -> str :
83+ if id :
8284 return id
8385 elif name and version :
8486 return f"{ name } -{ version } "
@@ -111,7 +113,7 @@ def process_response(response):
111113 prompt_lineage = prompt_version .get ("lineage" )
112114
113115 if prompt_lineage and prompt_lineage .get ("deletedAt" ):
114- logger .warn (
116+ logger .warning (
115117 f"Prompt { name } was deleted - please update any references to use an active prompt in production"
116118 )
117119 prompt = Prompt .from_dict (api , prompt_version ) if prompt_version else None
@@ -121,7 +123,14 @@ def process_response(response):
121123
122124 description = "get prompt"
123125
124- return gql .GET_PROMPT_VERSION , description , variables , process_response , timeout , cached_prompt
126+ return (
127+ gql .GET_PROMPT_VERSION ,
128+ description ,
129+ variables ,
130+ process_response ,
131+ timeout ,
132+ cached_prompt ,
133+ )
125134
126135
127136def create_prompt_variant_helper (
0 commit comments