phoenix-oss/llama-stack-mirror
0
0
Fork 1
mirror of https://github.com/meta-llama/llama-stack.git synced 2025-12-11 19:56:03 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

lint fix and rename file search

Browse source 
Signed-off-by: Francisco Javier Arceo <farceo@redhat.com>
This commit is contained in:
Francisco Javier Arceo 2025-10-24 20:24:03 -04:00
parent 2d9163529a
commit bb34e942d1
7 changed files with 26 additions and 25 deletions
Download patch file Download diff file Expand all files Collapse all files

2
llama_stack/core/routers/tool_runtime.py
View file

@ -41,7 +41,7 @@ class ToolRuntimeRouter(ToolRuntime):
query_config: FileSearchConfig | None = None,
) -> FileSearchResult:
logger.debug(f"ToolRuntimeRouter.FileSearchToolImpl.query: {vector_store_ids}")
provider = await self.routing_table.get_provider_impl("knowledge_search")
provider = await self.routing_table.get_provider_impl("file_search")
return await provider.query(content, vector_store_ids, query_config)
async def insert(

2
llama_stack/core/routing_tables/toolgroups.py
View file

@ -18,7 +18,7 @@ logger = get_logger(name=__name__, category="core::routing_tables")
def parse_toolgroup_from_toolgroup_name_pair(toolgroup_name_with_maybe_tool_name: str) -> str | None:
# handle the funny case like "builtin::file_search/knowledge_search"
# handle the funny case like "builtin::file_search/file_search"
parts = toolgroup_name_with_maybe_tool_name.split("/")
if len(parts) == 2:
return parts[0]