Update app.py

app.py

@@ -183,6 +183,9 @@ def getRAGChain(customerName, customerDistrict, custDetailsPresent, vectordb,llm
     )
     return chain
 
+def getRetriever(vectordb):
+    return vectordb.as_retriever(search_type="mmr", search_kwargs={'k': 2})
+
 def createVectorDB(documents,embeddingModelID):
     text_splitter = RecursiveCharacterTextSplitter(chunk_size=1500, chunk_overlap=150)
     texts = []