phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-10-05 12:21:52 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

update prompt variables to list for validation

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-09-07 14:10:19 -04:00
parent 60361b910c
commit 1390660dcf
5 changed files with 65 additions and 37 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -76,7 +76,7 @@ class PromptServiceImpl(Prompts):
"prompt_id": prompt.prompt_id,
"prompt": prompt.prompt,
"version": prompt.version,
"variables": prompt.variables or {},
"variables": prompt.variables or [],
}
)
@ -84,7 +84,7 @@ class PromptServiceImpl(Prompts):
"""Deserialize a prompt from JSON string."""
obj = json.loads(data)
return Prompt(
prompt_id=obj["prompt_id"], prompt=obj["prompt"], version=obj["version"], variables=obj.get("variables", {})
prompt_id=obj["prompt_id"], prompt=obj["prompt"], version=obj["version"], variables=obj.get("variables", [])
)
async def list_prompts(self) -> ListPromptsResponse:
@ -121,11 +121,11 @@ class PromptServiceImpl(Prompts):
async def create_prompt(
self,
prompt: str,
variables: dict[str, str] | None = None,
variables: list[str] | None = None,
) -> Prompt:
"""Create a new prompt."""
if variables is None:
variables = {}
variables = []
prompt_obj = Prompt(prompt_id=Prompt.generate_prompt_id(), prompt=prompt, version="1", variables=variables)
@ -142,12 +142,12 @@ class PromptServiceImpl(Prompts):
self,
prompt_id: str,
prompt: str,
variables: dict[str, str] | None = None,
variables: list[str] | None = None,
version: str | None = None,
) -> Prompt:
"""Update an existing prompt (increments version)."""
if variables is None:
variables = {}
variables = []
current_prompt = await self.get_prompt(prompt_id)
if version and current_prompt.version != version: