File size: 1,145 Bytes
b2b6341 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 | import sys
import os
sys.path.append(os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
from backend.ingestion.embedder import embed_texts, embed_query
from backend.vectorstore.faiss_store import add_vectors, search_vectors, get_total_vectors
# Step 1 - Embed some test chunks
print("Testing embedder...")
chunks = [
"Machine learning is a subset of artificial intelligence.",
"Python is a popular programming language for data science.",
"Neural networks are inspired by the human brain.",
]
vectors = embed_texts(chunks)
print(f"[SUCCESS] Embedded {len(vectors)} chunks, vector size: {len(vectors[0])}")
# Step 2 - Add to FAISS
print("\nTesting FAISS storage...")
fake_ids = ["chunk-001", "chunk-002", "chunk-003"]
add_vectors(fake_ids, vectors)
print(f"[SUCCESS] Total vectors in index: {get_total_vectors()}")
# Step 3 - Search
print("\nTesting search...")
query_vec = embed_query("What is artificial intelligence?")
results = search_vectors(query_vec, top_k=2)
print("[SUCCESS] Top 2 results:")
for r in results:
idx = fake_ids.index(r["chunk_id"])
print(f" Score: {r['score']:.4f} | Text: {chunks[idx]}") |