llama-stack/llama_stack/distribution/ui/page/distribution/shields.py
Angel Nunez Mencias d851df055a
All checks were successful
Build and Push playground container / build-playground (push) Successful in 1m55s
Build and Push container / build (push) Successful in 4m18s
fix security update
2025-06-03 20:02:04 +02:00

19 lines
575 B
Python

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the terms described in the LICENSE file in
# the root directory of this source tree.
import streamlit as st
from llama_stack.distribution.ui.modules.api import LlamaStackApi
def shields():
# Shields Section
st.header("Shields")
shields_info = {s.identifier: s.to_dict() for s in LlamaStackApi().client.shields.list()}
selected_shield = st.selectbox("Select a shield", list(shields_info.keys()))
st.json(shields_info[selected_shield])