feat (RAG): Implement configurable search mode in RAGQueryConfig

Signed-off-by: Varsha Prasad Narsing <varshaprasad96@gmail.com>
2025-12-29 01:28:46 +00:00 · 2025-04-14 16:53:17 -07:00 · 2025-04-14 16:53:17 -07:00 · e2a7022d3c
commit e2a7022d3c
parent 85b5f3172b
14 changed files with 210 additions and 43 deletions
--- a/llama_stack/providers/inline/tool_runtime/rag/memory.py
+++ b/llama_stack/providers/inline/tool_runtime/rag/memory.py
@ -122,6 +122,7 @@ class MemoryToolRuntimeImpl(ToolsProtocolPrivate, ToolRuntime, RAGToolRuntime):
                query=query,
                params={
                    "max_chunks": query_config.max_chunks,
+                    "mode": query_config.mode,
                },
            )
            for vector_db_id in vector_db_ids