From 6f4e891bcd20c418ccdf2ae91435562ac60d3aac Mon Sep 17 00:00:00 2001 From: anigasan Date: Wed, 16 Jul 2025 21:12:01 -0700 Subject: [PATCH] tavily changes to ymls --- docs/source/providers/tool_runtime/remote_tavily-search.md | 2 ++ llama_stack/templates/meta-reference-gpu/run-with-safety.yaml | 1 + llama_stack/templates/meta-reference-gpu/run.yaml | 1 + llama_stack/templates/open-benchmark/run.yaml | 1 + llama_stack/templates/postgres-demo/run.yaml | 1 + llama_stack/templates/starter/run.yaml | 1 + llama_stack/templates/vllm-gpu/run.yaml | 1 + 7 files changed, 8 insertions(+) diff --git a/docs/source/providers/tool_runtime/remote_tavily-search.md b/docs/source/providers/tool_runtime/remote_tavily-search.md index 3dc31534d..94769bebe 100644 --- a/docs/source/providers/tool_runtime/remote_tavily-search.md +++ b/docs/source/providers/tool_runtime/remote_tavily-search.md @@ -10,12 +10,14 @@ Tavily Search tool for AI-optimized web search with structured results. |-------|------|----------|---------|-------------| | `api_key` | `str \| None` | No | | The Tavily Search API Key | | `max_results` | `` | No | 3 | The maximum number of results to return | +| `timeout` | `` | No | 30.0 | HTTP request timeout for the API | ## Sample Configuration ```yaml api_key: ${env.TAVILY_SEARCH_API_KEY:=} max_results: 3 +timeout: 30.0 ``` diff --git a/llama_stack/templates/meta-reference-gpu/run-with-safety.yaml b/llama_stack/templates/meta-reference-gpu/run-with-safety.yaml index 49657a680..b0c7a56ec 100644 --- a/llama_stack/templates/meta-reference-gpu/run-with-safety.yaml +++ b/llama_stack/templates/meta-reference-gpu/run-with-safety.yaml @@ -107,6 +107,7 @@ providers: config: api_key: ${env.TAVILY_SEARCH_API_KEY:=} max_results: 3 + timeout: 30.0 - provider_id: rag-runtime provider_type: inline::rag-runtime config: {} diff --git a/llama_stack/templates/meta-reference-gpu/run.yaml b/llama_stack/templates/meta-reference-gpu/run.yaml index 2923b5faf..e280f514d 100644 --- a/llama_stack/templates/meta-reference-gpu/run.yaml +++ b/llama_stack/templates/meta-reference-gpu/run.yaml @@ -97,6 +97,7 @@ providers: config: api_key: ${env.TAVILY_SEARCH_API_KEY:=} max_results: 3 + timeout: 30.0 - provider_id: rag-runtime provider_type: inline::rag-runtime config: {} diff --git a/llama_stack/templates/open-benchmark/run.yaml b/llama_stack/templates/open-benchmark/run.yaml index 0b368ebc9..803a0a26d 100644 --- a/llama_stack/templates/open-benchmark/run.yaml +++ b/llama_stack/templates/open-benchmark/run.yaml @@ -119,6 +119,7 @@ providers: config: api_key: ${env.TAVILY_SEARCH_API_KEY:=} max_results: 3 + timeout: 30.0 - provider_id: rag-runtime provider_type: inline::rag-runtime config: {} diff --git a/llama_stack/templates/postgres-demo/run.yaml b/llama_stack/templates/postgres-demo/run.yaml index 2b6b1a64f..c6b674fa0 100644 --- a/llama_stack/templates/postgres-demo/run.yaml +++ b/llama_stack/templates/postgres-demo/run.yaml @@ -65,6 +65,7 @@ providers: config: api_key: ${env.TAVILY_SEARCH_API_KEY:=} max_results: 3 + timeout: 30.0 - provider_id: rag-runtime provider_type: inline::rag-runtime config: {} diff --git a/llama_stack/templates/starter/run.yaml b/llama_stack/templates/starter/run.yaml index 888a2c3bf..e6b54850e 100644 --- a/llama_stack/templates/starter/run.yaml +++ b/llama_stack/templates/starter/run.yaml @@ -246,6 +246,7 @@ providers: config: api_key: ${env.TAVILY_SEARCH_API_KEY:=} max_results: 3 + timeout: 30.0 - provider_id: rag-runtime provider_type: inline::rag-runtime config: {} diff --git a/llama_stack/templates/vllm-gpu/run.yaml b/llama_stack/templates/vllm-gpu/run.yaml index 4241569a4..576742df6 100644 --- a/llama_stack/templates/vllm-gpu/run.yaml +++ b/llama_stack/templates/vllm-gpu/run.yaml @@ -96,6 +96,7 @@ providers: config: api_key: ${env.TAVILY_SEARCH_API_KEY:=} max_results: 3 + timeout: 30.0 - provider_id: rag-runtime provider_type: inline::rag-runtime config: {}