phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-06-28 10:54:19 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Responses file_search wire up additional params

Browse source 
This adds passing of max_num_results from the file_search tool call
into the knowledge_search tool, as well as logs warnings if the
filters or ranking_options params are used since those are not wired
up yet. And, it adds the API surface for filters and ranking options
so we don't have to generate clients again as we add that.

Signed-off-by: Ben Browning <bbrownin@redhat.com>
This commit is contained in:
Ben Browning 2025-06-12 07:05:58 -04:00
parent 788d34d8b4
commit 8a5ea57253
4 changed files with 53 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

3
llama_stack/apis/agents/openai_responses.py
View file

@ -409,8 +409,9 @@ class FileSearchRankingOptions(BaseModel):
class OpenAIResponseInputToolFileSearch(BaseModel):
type: Literal["file_search"] = "file_search"
vector_store_ids: list[str]
filters: dict[str, Any] | None = None
max_num_results: int | None = Field(default=10, ge=1, le=50)
ranking_options: FileSearchRankingOptions | None = None
# TODO: add filters, max_num_results
class ApprovalFilter(BaseModel):