phoenix-oss/llama-stack
1
0
Fork 0
forked from phoenix-oss/llama-stack-mirror
Code Pull requests Releases Packages 2 Activity Actions

fix security update
All checks were successful
Build and Push playground container / build-playground (push) Successful in 1m57s
Details
Build and Push container / build (push) Successful in 4m33s
Details

Browse source
This commit is contained in:
Angel Nunez Mencias 2025-06-03 20:07:06 +02:00
parent 8943b283e9
commit b174effe05
Signed by: angel.nunez
SSH key fingerprint: SHA256:z1nFAg1v1AfbhEHrgBetByUJUwziv2R2f4VyN75opcg
14 changed files with 45 additions and 41 deletions
Download patch file Download diff file Expand all files Collapse all files

6
llama_stack/distribution/ui/page/playground/chat.py
View file

@ -6,12 +6,12 @@
import streamlit as st
from llama_stack.distribution.ui.modules.api import llama_stack_api
from llama_stack.distribution.ui.modules.api import LlamaStackApi
# Sidebar configurations
with st.sidebar:
st.header("Configuration")
available_models = llama_stack_api.client.models.list()
available_models = LlamaStackApi().client.models.list()
available_models = [model.identifier for model in available_models if model.model_type == "llm"]
selected_model = st.selectbox(
"Choose a model",
@ -103,7 +103,7 @@ if prompt := st.chat_input("Example: What is Llama Stack?"):
else:
strategy = {"type": "greedy"}
response = llama_stack_api.client.inference.chat_completion(
response = LlamaStackApi().client.inference.chat_completion(
messages=[
{"role": "system", "content": system_prompt},
{"role": "user", "content": prompt},