diff --git a/services/rag/llamaindex/server.py b/services/rag/llamaindex/server.py index eeb7ea4..6a2526e 100644 --- a/services/rag/llamaindex/server.py +++ b/services/rag/llamaindex/server.py @@ -67,6 +67,7 @@ class TestQueryResponse(BaseModel): query: str response: str sources: list[SourceItem] + error: bool @app.get("/health") @@ -102,7 +103,9 @@ def test_query(payload: TestQueryRequest) -> TestQueryResponse: response_text = str(result) logger.info(f"/api/test-query completed successfully (sources={len(sources)})") - return TestQueryResponse(query=query, response=response_text, sources=sources) + return TestQueryResponse( + query=query, response=response_text, sources=sources, error=False + ) except HTTPException: raise