import json
from pathlib import Path

import faiss
import numpy as np
from sentence_transformers import SentenceTransformer


corpus = [
    {
        "id": "doc-001",
        "text": "Sentence Transformers converts text into dense embeddings.",
    },
    {
        "id": "doc-002",
        "text": "FAISS stores vectors and searches nearest neighbors locally.",
    },
    {
        "id": "doc-003",
        "text": "Cross-encoders rerank a small set of retrieved passages.",
    },
    {
        "id": "doc-004",
        "text": "Qdrant stores vectors behind a database service API.",
    },
]