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

fix: search mode validation for rag query

Browse source
This commit is contained in:
Bobbins228 2025-07-22 16:36:21 +01:00
parent b5a6ecc331
commit 3d9f83ae87
No known key found for this signature in database
GPG key ID: 89FB8102B093BA9E
1 changed files with 2 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

4
llama_stack/apis/tools/rag_tool.py
View file

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