diff --git a/templates/components/vectordbs/python/mongo/index.py b/templates/components/vectordbs/python/mongo/index.py index d9b14dd22e4e0f1273612a8d986150e865a2bf5c..a80590b5c159bb0e2849a009c28930ecbd43a3d9 100644 --- a/templates/components/vectordbs/python/mongo/index.py +++ b/templates/components/vectordbs/python/mongo/index.py @@ -20,4 +20,4 @@ def get_chat_engine(): ) index = VectorStoreIndex.from_vector_store(store, service_context) logger.info("Finished connecting to index from MongoDB.") - return index.as_chat_engine(similarity_top_k=5) + return index.as_chat_engine(similarity_top_k=5, chat_mode="condense_plus_context") diff --git a/templates/components/vectordbs/python/none/index.py b/templates/components/vectordbs/python/none/index.py index 0170d6e83b1acbbc8c49475fcc4e3e376554db94..4404c66e28cfcada4f19b01420b7ee63a981bf86 100644 --- a/templates/components/vectordbs/python/none/index.py +++ b/templates/components/vectordbs/python/none/index.py @@ -22,4 +22,4 @@ def get_chat_engine(): storage_context = StorageContext.from_defaults(persist_dir=STORAGE_DIR) index = load_index_from_storage(storage_context, service_context=service_context) logger.info(f"Finished loading index from {STORAGE_DIR}") - return index.as_chat_engine() + return index.as_chat_engine(similarity_top_k=5, chat_mode="condense_plus_context") diff --git a/templates/components/vectordbs/python/pg/index.py b/templates/components/vectordbs/python/pg/index.py index 5c902772a682c6744a25e83609bb38e4faa339c9..510c21a538e176373522d323808547a160d1c146 100644 --- a/templates/components/vectordbs/python/pg/index.py +++ b/templates/components/vectordbs/python/pg/index.py @@ -13,4 +13,4 @@ def get_chat_engine(): store = init_pg_vector_store_from_env() index = VectorStoreIndex.from_vector_store(store, service_context) logger.info("Finished connecting to index from PGVector.") - return index.as_chat_engine(similarity_top_k=5) + return index.as_chat_engine(similarity_top_k=5, chat_mode="condense_plus_context")