Updating k

app.py

@@ -19,12 +19,12 @@ service_context = ServiceContext.from_defaults(llm=llm)
 
 arv_index = load_index_from_storage(StorageContext.from_defaults(persist_dir = "./arv/"))
 # arv_summary_index = load_index_from_storage(StorageContext.from_defaults(persist_dir = "./arv_summary/"))
-arv_vector_query_engine = arv_index.as_query_engine(similarity_top_k = 3)
+arv_vector_query_engine = arv_index.as_query_engine(similarity_top_k = 2)
 # arv_summary_query_engine = arv_summary_index.as_query_engine()
 
 nishauri_index = load_index_from_storage(StorageContext.from_defaults(persist_dir = "./nishauri/"))
 # nishauri_summary_index = load_index_from_storage(StorageContext.from_defaults(persist_dir = "./nishauri_summary/"))
-nishauri_vector_query_engine = nishauri_index.as_query_engine(similarity_top_k = 3)
+nishauri_vector_query_engine = nishauri_index.as_query_engine(similarity_top_k = 2)
 # nishauri_summary_query_engine = nishauri_summary_index.as_query_engine()
 
 from llama_index.agent import OpenAIAgent
@@ -120,7 +120,7 @@ nodes.append(node)
 
 # define top-level retriever
 vector_index = VectorStoreIndex(nodes)
-vector_retriever = vector_index.as_retriever(similarity_top_k=1)
+vector_retriever = vector_index.as_retriever(similarity_top_k=2)
 
 # define recursive retriever
 from llama_index.retrievers import RecursiveRetriever