feat: solution for 'Повторный экзамен: FAQ-бот — ChromaDB + один MCP-tool'

tests/test_ingest.py

@@ -0,0 +1,44 @@
+import os
+import tempfile
+import shutil
+import json
+from pathlib import Path
+
+import chromadb
+from chromadb.config import Settings
+
+from src.ingest import ingest_faq
+
+def test_ingest_faq(tmp_path):
+    # Create a temporary FAQ file
+    faq_content = """Q: What is Python?
+A: Python is a programming language.
+
+Q: What is ChromaDB?
+A: ChromaDB is a vector database."""
+    faq_file = tmp_path / "faq.txt"
+    faq_file.write_text(faq_content, encoding="utf-8")
+
+    # Initialize a temporary ChromaDB client
+    db_dir = tmp_path / "chromadb"
+    client = chromadb.Client(Settings(
+        chroma_db_impl="duckdb+parquet",
+        persist_directory=str(db_dir)
+    ))
+
+    collection_name = "test_collection"
+
+    # Ingest
+    ingest_faq(faq_file, client, collection_name)
+
+    # Verify collection exists and has documents
+    collection = client.get_collection(name=collection_name)
+    assert collection.count() == 2
+
+    # Verify metadata
+    docs = collection.get(ids=["0", "1"])
+    assert docs["metadatas"][0]["question"] == "What is Python?"
+    assert docs["metadatas"][1]["answer"] == "ChromaDB is a vector database."
+
+    # Clean up
+    shutil.rmtree(db_dir)

tests/test_retrieval.py

@@ -0,0 +1,51 @@
+import os
+import tempfile
+import shutil
+import json
+from pathlib import Path
+
+import chromadb
+from chromadb.config import Settings
+
+from src.ingest import ingest_faq
+from src.retriever import get_answer
+
+def test_retrieval(tmp_path, monkeypatch):
+    # Mock OpenAI API key
+    monkeypatch.setenv("OPENAI_API_KEY", "test_key")
+
+    # Create a temporary FAQ file
+    faq_content = """Q: What is Python?
+A: Python is a programming language.
+
+Q: What is ChromaDB?
+A: ChromaDB is a vector database."""
+    faq_file = tmp_path / "faq.txt"
+    faq_file.write_text(faq_content, encoding="utf-8")
+
+    # Initialize a temporary ChromaDB client
+    db_dir = tmp_path / "chromadb"
+    client = chromadb.Client(Settings(
+        chroma_db_impl="duckdb+parquet",
+        persist_directory=str(db_dir)
+    ))
+
+    collection_name = "test_collection"
+
+    # Ingest
+    ingest_faq(faq_file, client, collection_name)
+
+    # Mock OpenAIChat to avoid real API calls
+    class DummyLLM:
+        def __call__(self, *args, **kwargs):
+            return "Dummy answer"
+
+    # Patch the LLM in retriever
+    monkeypatch.setattr("src.retriever.OpenAIChat", DummyLLM)
+
+    # Retrieve answer
+    answer = get_answer("What is Python?", client, collection_name)
+    assert answer == "Dummy answer"
+
+    # Clean up
+    shutil.rmtree(db_dir)