vgecbot / tests /unit /test_unit.py
harsh-dev's picture
docker deployment
4225666
from app.api.dependencies import get_rag_service
req = {
"tests": [
{
"question": "List all faculties in Computer Engineering Department",
"document": "_department_php_dept_3.md"
},
{
"question": "Who is Dr. Arun B. Nandurbarkar?",
"document": "_department_php_dept_15.md"
},
{
"question": "Give faculty names from Information Technology Department",
"document": "_department_php_dept_7.md"
}
],
"k": 5,
"threshold": 0.4
}
RAG_SERVICE = get_rag_service()
print(RAG_SERVICE.test_queries(tests=req))