phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-22 16:23:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update PromptServiceImpl to use persistence resolver

Browse source
This commit is contained in:
Ashwin Bharambe 2025-10-05 13:13:37 -07:00
parent b03e6c32eb
commit 2b088e41fb
1 changed files with 10 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

12
llama_stack/core/prompts/prompts.py
View file

@ -11,6 +11,7 @@ from pydantic import BaseModel
from llama_stack.apis.prompts import ListPromptsResponse, Prompt, Prompts
from llama_stack.core.datatypes import StackRunConfig
from llama_stack.core.persistence_resolver import resolve_metadata_store_config
from llama_stack.core.utils.config_dirs import DISTRIBS_BASE_DIR
from llama_stack.providers.utils.kvstore import KVStore, kvstore_impl
from llama_stack.providers.utils.kvstore.config import SqliteKVStoreConfig
@ -41,9 +42,16 @@ class PromptServiceImpl(Prompts):
self.kvstore: KVStore
async def initialize(self) -> None:
kvstore_config = SqliteKVStoreConfig(
db_path=(DISTRIBS_BASE_DIR / self.config.run_config.image_name / "prompts.db").as_posix()
# Use metadata store backend with prompts namespace
kvstore_config = resolve_metadata_store_config(
self.config.run_config.persistence,
self.config.run_config.image_name,
)
# Override namespace for prompts if using a shared backend
if kvstore_config.namespace is None:
config_dict = kvstore_config.model_dump()
config_dict["namespace"] = "prompts"
kvstore_config = type(kvstore_config)(**config_dict)
self.kvstore = await kvstore_impl(kvstore_config)
def _get_default_key(self, prompt_id: str) -> str: