| import os |
|
|
| os.environ.setdefault("LLM_PROVIDER", "mock") |
| os.environ.setdefault("EMBEDDING_BACKEND", "tfidf") |
|
|
| from app.core.config import get_settings |
| from app.crag.retriever import ( |
| infer_sql_query, |
| looks_numeric_question, |
| retrieve_internal_docs, |
| ) |
| from app.ingestion.indexer import get_indexer, reset_indexer |
|
|
| get_settings.cache_clear() |
| reset_indexer() |
|
|
|
|
| def test_indexer_loads_documents(): |
| indexer = get_indexer() |
| assert len(indexer.documents) >= 6 |
| assert len(indexer.chunks) > 0 |
| doc_ids = {d.doc_id for d in indexer.documents} |
| assert "refund_policy" in doc_ids |
| assert "sales_q1_2026" in doc_ids |
|
|
|
|
| def test_retrieval_returns_relevant_refund_policy(): |
| results = retrieve_internal_docs( |
| "Quel est le délai de remboursement ?", top_k=3 |
| ) |
| assert results |
| doc_ids = [r["doc_id"] for r in results] |
| assert "refund_policy" in doc_ids |
|
|
|
|
| def test_numeric_question_detection(): |
| assert looks_numeric_question( |
| "Quel client a le meilleur chiffre d'affaires en juin ?" |
| ) |
| assert not looks_numeric_question( |
| "Quelle est la politique de remboursement ?" |
| ) |
|
|
|
|
| def test_infer_sql_query_extracts_month_and_product(): |
| params = infer_sql_query( |
| "Quel est le chiffre d'affaires total du Produit X en avril ?" |
| ) |
| assert params is not None |
| assert params["filters"].get("mois") == "Avril" |
| assert params["filters"].get("produit") == "Produit X" |
|
|
