diff --git a/src/llama_stack/apis/agents/openai_responses.py b/src/llama_stack/apis/agents/openai_responses.py index 69e2b2012..553ac16c2 100644 --- a/src/llama_stack/apis/agents/openai_responses.py +++ b/src/llama_stack/apis/agents/openai_responses.py @@ -403,7 +403,7 @@ class OpenAIResponseText(BaseModel): # Must match type Literals of OpenAIResponseInputToolWebSearch below -WebSearchToolTypes = ["web_search", "web_search_preview", "web_search_preview_2025_03_11"] +WebSearchToolTypes = ["web_search", "web_search_preview", "web_search_preview_2025_03_11", "web_search_2025_08_26"] @json_schema_type @@ -415,7 +415,7 @@ class OpenAIResponseInputToolWebSearch(BaseModel): """ # Must match values of WebSearchToolTypes above - type: Literal["web_search"] | Literal["web_search_preview"] | Literal["web_search_preview_2025_03_11"] = ( + type: Literal["web_search"] | Literal["web_search_preview"] | Literal["web_search_preview_2025_03_11"] | Literal["web_search_2025_08_26"] = ( "web_search" ) # TODO: actually use search_context_size somewhere... diff --git a/tests/integration/responses/fixtures/test_cases.py b/tests/integration/responses/fixtures/test_cases.py index b90dde62f..7ad923c40 100644 --- a/tests/integration/responses/fixtures/test_cases.py +++ b/tests/integration/responses/fixtures/test_cases.py @@ -95,6 +95,14 @@ web_search_test_cases = [ ), id="llama_experts", ), + pytest.param( + ResponsesTestCase( + input="What is the latest version of Python?", + tools=[{"type": "web_search_2025_08_26"}], + expected="python", + ), + id="web_search_2025_08_26_type", + ), ] # File search test cases