Skip to content
This repository was archived by the owner on Aug 5, 2025. It is now read-only.

Commit 95061a7

Browse files
committed
feat: change warn to warning + fix ci
1 parent 79dd258 commit 95061a7

File tree

1 file changed

+17
-8
lines changed

1 file changed

+17
-8
lines changed

literalai/api/helpers/prompt_helpers.py

Lines changed: 17 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -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

127136
def create_prompt_variant_helper(

0 commit comments

Comments
 (0)