From 5c010e234a13b064803884e5e9c1fd9ce47f3741 Mon Sep 17 00:00:00 2001 From: Michael Clifford Date: Wed, 9 Apr 2025 09:56:41 -0400 Subject: [PATCH] fix: add tavily_search option to playground api (#1909) # What does this PR do? This PR adds the "TAVILY_SEARCH_API_KEY" option to the playground to enable the use of the websearch tool. [//]: # (If resolving an issue, uncomment and update the line below) [//]: # (Closes #[issue-number]) ## Test Plan ``` export TAVILY_SEARCH_API_KEY=*** streamlit run llama_stack/distribution/ui/app.py ``` Without this change the builtin websearch tool will fail due to missing API key. [//]: # (## Documentation) Related to #1902 Signed-off-by: Michael Clifford --- llama_stack/distribution/ui/modules/api.py | 1 + 1 file changed, 1 insertion(+) diff --git a/llama_stack/distribution/ui/modules/api.py b/llama_stack/distribution/ui/modules/api.py index 40caccda0..d5395c5b9 100644 --- a/llama_stack/distribution/ui/modules/api.py +++ b/llama_stack/distribution/ui/modules/api.py @@ -19,6 +19,7 @@ class LlamaStackApi: "together_api_key": os.environ.get("TOGETHER_API_KEY", ""), "sambanova_api_key": os.environ.get("SAMBANOVA_API_KEY", ""), "openai_api_key": os.environ.get("OPENAI_API_KEY", ""), + "tavily_search_api_key": os.environ.get("TAVILY_SEARCH_API_KEY", ""), }, )