import llama_index

class rag_worker():
    def __init__(self) -> None:
        pass

    def assign_embedding_model(self):
        pass

    def save_to_checkpoint(self):
        pass

    def load_from_checkpoint(self):
        pass

    def add_documents_to_vector_store(self, documents):
        pass

    def add_text_to_vector_store(self, documents):
        pass

    def inspect_vector_store(self):
        pass

    def retrieve_from_store_with_query(self, query):
        pass

    @staticmethod
    def build_prompt(self):
        pass