add test for RAGChunks

Signed-off-by: Anxhela Coba <acoba@redhat.com>

Author: Anxhela21

pyproject.toml

@@ -65,6 +65,7 @@ dependencies = [
     "peft>=0.15.2",
     "trl>=0.18.2",
     "sentence-transformers>=5.1.0",
+    "greenlet>=3.2.4",
 ]
 
 

src/utils/types.py

@@ -7,7 +7,7 @@
 from llama_stack_client.types.shared.completion_message import CompletionMessage
 from llama_stack_client.types.shared.tool_call import ToolCall
 from llama_stack_client.types.tool_execution_step import ToolExecutionStep
-from pydantic.main import BaseModel, Field
+from pydantic import BaseModel, Field
 
 
 class Singleton(type):

tests/unit/app/endpoints/test_streaming_query.py

@@ -45,7 +45,7 @@
 from models.requests import QueryRequest, Attachment
 from models.config import ModelContextProtocolServer, Action
 from authorization.resolvers import NoopRolesResolver
-from utils.types import ToolCallSummary, TurnSummary
+from utils.types import ToolCallSummary, TurnSummary, RAGChunkData
 
 MOCK_AUTH = ("mock_user_id", "mock_username", False, "mock_token")
 
@@ -343,6 +343,13 @@ async def _test_streaming_query_endpoint_handler(mocker, store_transcript=False)
                         response=" ".join(SAMPLE_KNOWLEDGE_SEARCH_RESULTS),
                     )
                 ],
+                rag_chunks=[
+                    RAGChunkData(
+                        content=" ".join(SAMPLE_KNOWLEDGE_SEARCH_RESULTS),
+                        source="knowledge_search",
+                        score=None,
+                    )
+                ],
             ),
             attachments=[],
             rag_chunks=[],