From 9a28f681cf1db0ea91be5a30a039e44d75ea10ad Mon Sep 17 00:00:00 2001 From: Chihurumnaya Ibiam Date: Tue, 22 Oct 2024 18:37:40 +0100 Subject: [PATCH] Remove redundant docstrings Signed-off-by: Chihurumnaya Ibiam --- rag_agent.py | 7 ------- 1 file changed, 7 deletions(-) diff --git a/rag_agent.py b/rag_agent.py index e20fa9b..1e008fe 100644 --- a/rag_agent.py +++ b/rag_agent.py @@ -36,19 +36,14 @@ class RAG_Agent: def __init__(self, model="llama3.1"): - """ - Initialize 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): - """To Set a new model if needed.""" self.model = OllamaLLM(model=model) def get_model(self): - """Return the LLM model.""" return self.model # Loading the docs for retrieval in Vector Database @@ -105,8 +100,6 @@ def get_relevant_document(self, query, threshold=0.5): def run(self): - """Run the main logic of the RAG agent.""" - # Format documents for context def format_docs(docs): return "\n\n".join(doc.page_content for doc in docs)