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

Merge branch 'main' into rm-faiss-file

Browse source
This commit is contained in:
raghotham 2025-07-14 10:57:14 -07:00 • committed by GitHub
commit c30e343fbb
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
9 changed files with 102 additions and 4 deletions
Download patch file Download diff file Expand all files Collapse all files

12
tests/unit/rag/test_rag_query.py
View file

@ -8,6 +8,7 @@ from unittest.mock import AsyncMock, MagicMock
import pytest
from llama_stack.apis.tools.rag_tool import RAGQueryConfig
from llama_stack.apis.vector_io import (
Chunk,
ChunkMetadata,
@ -58,3 +59,14 @@ class TestRagQuery:
)
assert expected_metadata_string in result.content[1].text
assert result.content is not None
async def test_query_raises_incorrect_mode(self):
with pytest.raises(ValueError):
RAGQueryConfig(mode="invalid_mode")
@pytest.mark.asyncio
async def test_query_accepts_valid_modes(self):
RAGQueryConfig() # Test default (vector)
RAGQueryConfig(mode="vector") # Test vector
RAGQueryConfig(mode="keyword") # Test keyword
RAGQueryConfig(mode="hybrid") # Test hybrid