Migrated from RetrievalQA, removed print statements

[kshitijdshah99]

I have added paths of all the docs and modified the code as per our requirements. This Fixes #4

[chimosky]

You've set an initial value for self.model in a640e38, but haven't removed the kwarg you passed to set_model.

Your initial value means that set_model doesn't need to be called if you're setting it to the same value as the one that's been set.

[chimosky]

Just looked at c45998c, I didn't say you needed to remove the function, it's not redundant.

I said "You've set an initial value for self.model in a640e38, but haven't removed the kwarg you passed to set_model."

Please let me know if you don't understand what I meant.

[kshitijdshah99]

ok so you want me to do this changes?

def __init__(self, model="llama3.1"):
        """
        Initializing the RAG agent with a default model like Llama3.1.
        """
        self.model = OllamaLLM(model=model)
        self.retriever = None
        self.prompt = ChatPromptTemplate.from_template(PROMPT_TEMPLATE)

    def set_model(self, model):
        self.model = OllamaLLM(model=model)

[chimosky]

Yes, that's what I want.

The init docstring is stating the obvious, if that's all it does then I don't think we need it.

[chimosky]

Fixed the issues with redundant comments in 3ca1ff7.