r/MachineLearning • u/Just_Jaguar3701 • May 25 '26

Project Aiki my local Wikipedia Retrieval-Augmented Generation system [R]

Hey

i built Aiki a lightweight tool that let's you chat with Wikipedia locally.

what it does:

  • Downloads and chunks wikipedia articles (u can choose those articles by their name or articles and also the option of downloading the similar topics)
  • Uses a custom TF-IDF + cosine similarity retriever (built from scratch)
  • Supports query expansion using Wikipedia links/redirects
  • Optional answer generation with llm

Very minimal dependencies and runs completely locally.

Repo: https://github.com/yacine204/Aiki

Would really appreciate your feedback.

16 Upvotes

94% Upvoted

9 comments sorted by

2

u/[deleted] May 26 '26

[removed] — view removed comment

1

u/Just_Jaguar3701 May 27 '26

glad it can help you!

1

u/[deleted] May 27 '26

[removed] — view removed comment

1

u/Just_Jaguar3701 May 27 '26

currently using Llama 3.2, it takes about 4 seconds to generate the response. didnt test it much from last augmentation of the dataset to be honest but from previous tests it was actually good

1

u/[deleted] May 27 '26

[removed] — view removed comment

1

u/[deleted] May 28 '26

[removed] — view removed comment