phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-27 18:50:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

feat: support filters in file search (#2472)

Browse source 
# What does this PR do?
Move to use vector_stores.search for file search tool in Responses,
which supports filters.

closes #2435 

## Test Plan
Added e2e test with fitlers.
myenv ❯ llama stack run llama_stack/templates/fireworks/run.yaml

pytest -sv tests/verifications/openai_api/test_responses.py \
  -k 'file_search and filters' \
  --base-url=http://localhost:8321/v1/openai/v1 \
  --model=meta-llama/Llama-3.3-70B-Instruct
This commit is contained in:
ehhuang 2025-06-18 21:50:55 -07:00 • committed by GitHub
parent fd37a50e6a
commit db2cd9e8f3
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
13 changed files with 449 additions and 63 deletions
Download patch file Download diff file Expand all files Collapse all files

32
docs/_static/llama-stack-spec.html vendored
View file

@ -7296,7 +7296,7 @@
}
},
"additionalProperties": false,
"title": "FileSearchRankingOptions"
"title": "SearchRankingOptions"
}
},
"additionalProperties": false,
@ -13478,28 +13478,16 @@
},
"ranking_options": {
"type": "object",
"additionalProperties": {
"oneOf": [
{
"type": "null"
},
{
"type": "boolean"
},
{
"type": "number"
},
{
"type": "string"
},
{
"type": "array"
},
{
"type": "object"
}
]
"properties": {
"ranker": {
"type": "string"
},
"score_threshold": {
"type": "number",
"default": 0.0
}
},
"additionalProperties": false,
"description": "Ranking options for fine-tuning the search results."
},
"rewrite_query": {

17
docs/_static/llama-stack-spec.yaml vendored
View file

@ -5179,7 +5179,7 @@ components:
type: number
default: 0.0
additionalProperties: false
title: FileSearchRankingOptions
title: SearchRankingOptions
additionalProperties: false
required:
- type
@ -9408,14 +9408,13 @@ components:
Maximum number of results to return (1 to 50 inclusive, default 10).
ranking_options:
type: object
additionalProperties:
oneOf:
- type: 'null'
- type: boolean
- type: number
- type: string
- type: array
- type: object
properties:
ranker:
type: string
score_threshold:
type: number
default: 0.0
additionalProperties: false
description: >-
Ranking options for fine-tuning the search results.
rewrite_query: