phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-07-29 07:14:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 1

more commits

Browse source
This commit is contained in:
anigasan 2025-07-06 18:44:27 -07:00
parent ef332c296e
commit e679cd0261
34 changed files with 68 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
docs/source/providers/tool_runtime/remote_tavily-search.md
View file

@ -10,12 +10,16 @@ Tavily Search tool for AI-optimized web search with structured results.
|-------|------|----------|---------|-------------|
| `api_key` | `str \| None` | No | | The Tavily Search API Key |
| `max_results` | `<class 'int'>` | No | 3 | The maximum number of results to return |
| `timeout` | `<class 'float'>` | No | 30.0 | HTTP request timeout for the API |
| `connect_timeout` | `<class 'float'>` | No | 10.0 | HTTP connection timeout in seconds for the API |
## Sample Configuration
```yaml
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
```

2
llama_stack/providers/remote/tool_runtime/tavily_search/config.py
View file

@ -32,6 +32,6 @@ class TavilySearchToolConfig(BaseModel):
return {
"api_key": "${env.TAVILY_SEARCH_API_KEY:=}",
"max_results": 3,
"timeout:" 30.0,
"timeout": 30.0,
"connect_timeout": 10.0,
}

2
llama_stack/providers/remote/tool_runtime/tavily_search/tavily_search.py
View file

@ -70,7 +70,7 @@ class TavilySearchToolRuntimeImpl(ToolGroupsProtocolPrivate, ToolRuntime, NeedsR
async def invoke_tool(self, tool_name: str, kwargs: dict[str, Any]) -> ToolInvocationResult:
api_key = self._get_api_key()
timeout = https.Timeout(timeout=self.config.timeout, connect=self.config.connect_timeout)
httpx.Timeout(timeout=self.config.timeout, connect=self.config.connect_timeout)
async with httpx.AsyncClient() as client:
response = await client.post(

2
llama_stack/templates/bedrock/run.yaml
View file

@ -85,6 +85,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/cerebras/run.yaml
View file

@ -91,6 +91,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/ci-tests/run.yaml
View file

@ -89,6 +89,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/dell/run-with-safety.yaml
View file

@ -92,6 +92,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/dell/run.yaml
View file

@ -88,6 +88,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/fireworks/run-with-safety.yaml
View file

@ -105,6 +105,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: wolfram-alpha
provider_type: remote::wolfram-alpha
config:

2
llama_stack/templates/fireworks/run.yaml
View file

@ -100,6 +100,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: wolfram-alpha
provider_type: remote::wolfram-alpha
config:

2
llama_stack/templates/groq/run.yaml
View file

@ -91,6 +91,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/hf-endpoint/run-with-safety.yaml
View file

@ -96,6 +96,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/hf-endpoint/run.yaml
View file

@ -91,6 +91,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/hf-serverless/run-with-safety.yaml
View file

@ -96,6 +96,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/hf-serverless/run.yaml
View file

@ -91,6 +91,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/llama_api/run.yaml
View file

@ -101,6 +101,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/meta-reference-gpu/run-with-safety.yaml
View file

@ -106,6 +106,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/meta-reference-gpu/run.yaml
View file

@ -96,6 +96,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/ollama/run-with-safety.yaml
View file

@ -107,6 +107,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/ollama/run.yaml
View file

@ -105,6 +105,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/open-benchmark/run.yaml
View file

@ -115,6 +115,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/passthrough/run-with-safety.yaml
View file

@ -96,6 +96,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: wolfram-alpha
provider_type: remote::wolfram-alpha
config:

2
llama_stack/templates/passthrough/run.yaml
View file

@ -91,6 +91,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: wolfram-alpha
provider_type: remote::wolfram-alpha
config:

2
llama_stack/templates/postgres-demo/run.yaml
View file

@ -65,6 +65,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/remote-vllm/run-with-safety.yaml
View file

@ -100,6 +100,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/remote-vllm/run.yaml
View file

@ -93,6 +93,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/sambanova/run.yaml
View file

@ -70,6 +70,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/starter/run.yaml
View file

@ -162,6 +162,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/tgi/run-with-safety.yaml
View file

@ -91,6 +91,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/tgi/run.yaml
View file

@ -90,6 +90,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/together/run-with-safety.yaml
View file

@ -96,6 +96,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/together/run.yaml
View file

@ -91,6 +91,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/vllm-gpu/run.yaml
View file

@ -95,6 +95,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}

2
llama_stack/templates/watsonx/run.yaml
View file

@ -92,6 +92,8 @@ providers:
config:
api_key: ${env.TAVILY_SEARCH_API_KEY:=}
max_results: 3
timeout: 30.0
connect_timeout: 10.0
- provider_id: rag-runtime
provider_type: inline::rag-runtime
config: {}