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/tools.py
View file

@ -13,7 +13,7 @@ from llama_stack_client import Agent
from llama_stack_client.lib.agents.react.agent import ReActAgent
from llama_stack_client.lib.agents.react.tool_parser import ReActOutput
from llama_stack.distribution.ui.modules.api import llama_stack_api
from llama_stack.distribution.ui.modules.api import LlamaStackApi
class AgentType(enum.Enum):
@ -24,7 +24,7 @@ class AgentType(enum.Enum):
def tool_chat_page():
st.title("🛠 Tools")
client = llama_stack_api.client
client = LlamaStackApi().client
models = client.models.list()
model_list = [model.identifier for model in models if model.api_model_type == "llm"]
@ -55,7 +55,7 @@ def tool_chat_page():
)
if "builtin::rag" in toolgroup_selection:
vector_dbs = llama_stack_api.client.vector_dbs.list() or []
vector_dbs = LlamaStackApi().client.vector_dbs.list() or []
if not vector_dbs:
st.info("No vector databases available for selection.")
vector_dbs = [vector_db.identifier for vector_db in vector_dbs]