mirror of
https://github.com/meta-llama/llama-stack.git
synced 2025-12-03 09:53:45 +00:00
Add web_search_2025_08_26 type support
Signed-off-by: Aakanksha Duggal <aduggal@redhat.com>
This commit is contained in:
parent
b68a25d377
commit
4d3b89903e
2 changed files with 10 additions and 2 deletions
|
|
@ -403,7 +403,7 @@ class OpenAIResponseText(BaseModel):
|
||||||
|
|
||||||
|
|
||||||
# Must match type Literals of OpenAIResponseInputToolWebSearch below
|
# 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
|
@json_schema_type
|
||||||
|
|
@ -415,7 +415,7 @@ class OpenAIResponseInputToolWebSearch(BaseModel):
|
||||||
"""
|
"""
|
||||||
|
|
||||||
# Must match values of WebSearchToolTypes above
|
# 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"
|
"web_search"
|
||||||
)
|
)
|
||||||
# TODO: actually use search_context_size somewhere...
|
# TODO: actually use search_context_size somewhere...
|
||||||
|
|
|
||||||
|
|
@ -95,6 +95,14 @@ web_search_test_cases = [
|
||||||
),
|
),
|
||||||
id="llama_experts",
|
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
|
# File search test cases
|
||||||
|
|
|
||||||
Loading…
Add table
Add a link
Reference in a new issue