Skip to content

Commit 9c681fd

Browse files
committed
Update type hint to reflect that function can return None
1 parent f1bfe80 commit 9c681fd

File tree

1 file changed

+1
-6
lines changed

1 file changed

+1
-6
lines changed

packages/slackBotFunction/app/services/prompt_loader.py

Lines changed: 1 addition & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -13,11 +13,6 @@ def load_prompt(prompt_name: str, prompt_version: str = None) -> str:
1313
1414
Resolves prompt name to ID, then loads the specified version.
1515
Supports both DRAFT and numbered versions.
16-
17-
Args:
18-
prompt_name: The human-readable name of the prompt
19-
prompt_version: Version to load - "DRAFT" for latest draft, number for published version,
20-
None for default behavior (loads DRAFT)
2116
"""
2217
try:
2318
client = boto3.client("bedrock-agent", region_name=os.environ["AWS_REGION"])
@@ -71,7 +66,7 @@ def load_prompt(prompt_name: str, prompt_version: str = None) -> str:
7166
raise PromptLoadError(f"Unexpected error loading prompt '{prompt_name}': {e}")
7267

7368

74-
def get_prompt_id_from_name(client, prompt_name: str) -> str:
69+
def get_prompt_id_from_name(client, prompt_name: str) -> str | None:
7570
"""
7671
Get the 10-character prompt ID from the prompt name using ListPrompts.
7772
"""

0 commit comments

Comments
 (0)