File size: 409 Bytes
7248d39
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
"""Smoke test for FAISS vector store."""

import sys
from pathlib import Path

sys.path.insert(0, str(Path(__file__).resolve().parents[1]))

from db.faiss_client import FaissDB


def main():
    print("Opening FAISS store...")
    db = FaissDB()
    docs = db.list_documents()
    print(f"Documents indexed: {len(docs)}")
    db.close()
    print("FAISS test passed.")


if __name__ == "__main__":
    main()