llama-stack/llama_stack/distribution/ui

(Experimental) LLama Stack UI

Docker Setup

⚠️ This is a work in progress.

Developer Setup

1. Start up Llama Stack API server. More details here.

llama stack build --template together --image-type conda

llama stack run together

2. (Optional) Register datasets and eval tasks as resources. If you want to run pre-configured evaluation flows (e.g. Evaluations (Generation + Scoring) Page).

llama-stack-client datasets register \
--dataset-id "mmlu" \
--provider-id "huggingface" \
--url "https://huggingface.co/datasets/llamastack/evals" \
--metadata '{"path": "llamastack/evals", "name": "evals__mmlu__details", "split": "train"}' \
--schema '{"input_query": {"type": "string"}, "expected_answer": {"type": "string", "chat_completion_input": {"type": "string"}}}'

llama-stack-client benchmarks register \
--eval-task-id meta-reference-mmlu \
--provider-id meta-reference \
--dataset-id mmlu \
--scoring-functions basic::regex_parser_multiple_choice_answer

3. Start Streamlit UI

cd llama_stack/distribution/ui
pip install -r requirements.txt
streamlit run app.py

Environment Variables

Environment Variable	Description	Default Value
LLAMA_STACK_ENDPOINT	The endpoint for the Llama Stack	http://localhost:8321
FIREWORKS_API_KEY	API key for Fireworks provider	(empty string)
TOGETHER_API_KEY	API key for Together provider	(empty string)
SAMBANOVA_API_KEY	API key for SambaNova provider	(empty string)
OPENAI_API_KEY	API key for OpenAI provider	(empty string)