diff --git a/llama_stack/core/prompts/prompts.py b/llama_stack/core/prompts/prompts.py index 31e8bfcc9..f69c81f8b 100644 --- a/llama_stack/core/prompts/prompts.py +++ b/llama_stack/core/prompts/prompts.py @@ -43,7 +43,7 @@ class PromptServiceImpl(Prompts): def _get_prompt_key(self, prompt_id: str, version: str | None = None) -> str: if version: - return f"prompts:v1:{prompt_id}:{version}" + return self._get_version_key(prompt_id, version) return f"prompts:v1:{prompt_id}:default" def _get_version_key(self, prompt_id: str, version: str) -> str: