mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-17 18:32:41 +00:00
82 lines
2.2 KiB
Python
82 lines
2.2 KiB
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
|
|
|
|
|
|
def main():
|
|
# Evaluation pages
|
|
application_evaluation_page = st.Page(
|
|
"page/evaluations/app_eval.py",
|
|
title="Application Evaluation",
|
|
icon="📊",
|
|
default=False,
|
|
)
|
|
native_evaluation_page = st.Page(
|
|
"page/evaluations/native_eval.py",
|
|
title="Native Evaluation",
|
|
icon="📊",
|
|
default=False,
|
|
)
|
|
|
|
# Playground pages
|
|
chat_page = st.Page(
|
|
"page/playground/chat.py", title="Chat", icon="💬", default=True
|
|
)
|
|
rag_page = st.Page("page/playground/rag.py", title="RAG", icon="💬", default=False)
|
|
|
|
# Distribution pages
|
|
provider_page = st.Page(
|
|
"page/distribution/providers.py", title="Provider", icon="🔍", default=False
|
|
)
|
|
model_page = st.Page(
|
|
"page/distribution/models.py", title="Models", icon="🔍", default=False
|
|
)
|
|
memory_bank_page = st.Page(
|
|
"page/distribution/memory_banks.py",
|
|
title="Memory Banks",
|
|
icon="🔍",
|
|
default=False,
|
|
)
|
|
shield_page = st.Page(
|
|
"page/distribution/shields.py", title="Shields", icon="🔍", default=False
|
|
)
|
|
scoring_function_page = st.Page(
|
|
"page/distribution/scoring_functions.py",
|
|
title="Scoring Functions",
|
|
icon="🔍",
|
|
default=False,
|
|
)
|
|
eval_task_page = st.Page(
|
|
"page/distribution/eval_tasks.py",
|
|
title="Eval Tasks",
|
|
icon="🔍",
|
|
default=False,
|
|
)
|
|
|
|
pg = st.navigation(
|
|
{
|
|
"Playground": [
|
|
chat_page,
|
|
rag_page,
|
|
application_evaluation_page,
|
|
native_evaluation_page,
|
|
],
|
|
"Inspect": [
|
|
provider_page,
|
|
model_page,
|
|
memory_bank_page,
|
|
shield_page,
|
|
scoring_function_page,
|
|
eval_task_page,
|
|
],
|
|
},
|
|
expanded=False,
|
|
)
|
|
pg.run()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|