diff --git a/services/rag/llamaindex/server.py b/services/rag/llamaindex/server.py index 6a2526e..51530a0 100644 --- a/services/rag/llamaindex/server.py +++ b/services/rag/llamaindex/server.py @@ -68,6 +68,7 @@ class TestQueryResponse(BaseModel): response: str sources: list[SourceItem] error: bool + success: bool @app.get("/health") @@ -104,7 +105,11 @@ 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, error=False + query=query, + response=response_text, + sources=sources, + error=False, + success=True, ) except HTTPException: