twkeed-rag

pip install torch sentence-transformers FlagEmbedding faiss-cpu

from twkeed_rag import TwkeedRAG, Document

# Initialize
rag = TwkeedRAG()

# Add Arabic documents
rag.add_documents([
    Document(id="1", text="المملكة العربية السعودية هي أكبر دولة في شبه الجزيرة العربية"),
    Document(id="2", text="الرياض هي عاصمة المملكة العربية السعودية"),
    Document(id="3", text="مكة المكرمة هي أقدس مدينة في الإسلام"),
])

# Search
results = rag.search("ما هي عاصمة السعودية؟", top_k=3)

for r in results:
    print(f"Score: {r.rerank_score:.4f} - {r.document.text}")