diff --git a/llama_stack/apis/tools/rag_tool.py b/llama_stack/apis/tools/rag_tool.py index cfaa49488..1d5e7b6cb 100644 --- a/llama_stack/apis/tools/rag_tool.py +++ b/llama_stack/apis/tools/rag_tool.py @@ -4,7 +4,7 @@ # This source code is licensed under the terms described in the LICENSE file in # the root directory of this source tree. -from enum import Enum +from enum import Enum, StrEnum from typing import Annotated, Any, Literal, Protocol from pydantic import BaseModel, Field, field_validator @@ -88,7 +88,7 @@ class RAGQueryGenerator(Enum): @json_schema_type -class RAGSearchMode(Enum): +class RAGSearchMode(StrEnum): """ Search modes for RAG query retrieval: - VECTOR: Uses vector similarity search for semantic matching