rag.py

class RAGIndex:
    def __init__(self):
        self.docs=[]

    def add(self,chunks):
        self.docs.extend(chunks)

    def retrieve(self,q,k=5):
        return [(d,1.0) for d in self.docs[:k]]