From dc21e14f641755311ddf328c6bd5ab4111d099c0 Mon Sep 17 00:00:00 2001 From: Dinesh Yeduguru Date: Thu, 19 Dec 2024 07:42:30 -0800 Subject: [PATCH] make api_key optional in config --- .../inline/tool_runtime/brave_search/config.py | 14 +++++++++++--- 1 file changed, 11 insertions(+), 3 deletions(-) diff --git a/llama_stack/providers/inline/tool_runtime/brave_search/config.py b/llama_stack/providers/inline/tool_runtime/brave_search/config.py index e8fbaaec9..565d428f7 100644 --- a/llama_stack/providers/inline/tool_runtime/brave_search/config.py +++ b/llama_stack/providers/inline/tool_runtime/brave_search/config.py @@ -4,9 +4,17 @@ # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. -from pydantic import BaseModel +from typing import Optional + +from pydantic import BaseModel, Field class BraveSearchToolConfig(BaseModel): - api_key: str - max_results: int = 3 + api_key: Optional[str] = Field( + default=None, + description="The Brave Search API Key", + ) + max_results: int = Field( + default=3, + description="The maximum number of results to return", + )