Update app.py

app.py

@@ -32,7 +32,7 @@ from scipy.stats import spearmanr
 from functools import lru_cache
 from langchain.retrievers import MultiQueryRetriever
 from langchain_huggingface.llms import HuggingFacePipeline
-from langchain.schema import BaseRunManager, NoOpRunManager
+from langchain.schema import NoOpRunManager
 from transformers import pipeline
 from sklearn.model_selection import ParameterGrid
 from tqdm import tqdm
@@ -199,9 +199,8 @@ def optimize_query(query, llm_model, chunks, embedding_model, vector_store_type,
         retriever=temp_retriever,
         llm=llm
     ) 
-        # Use a NoOpRunManager as the run_manager
-    run_manager = NoOpRunManager()
-    optimized_queries = multi_query_retriever.generate_queries(query, run_manager=run_manager)
+        # Use a NoOpRunManager as the run_manager 
+    optimized_queries = multi_query_retriever.invoke(query)
     return optimized_queries 
     
 
@@ -1184,7 +1183,7 @@ def optimize_query(query, llm):
         retriever=get_retriever(vector_store, search_type, search_kwargs),
         llm=llm
     )
-    optimized_queries = multi_query_retriever.generate_queries(query)
+    optimized_queries = multi_query_retriever.invoke(query)
     return optimized_queries
 ````