Update app.py
Browse files
app.py
CHANGED
@@ -667,7 +667,7 @@ def automated_testing(file, query, test_params, expected_result=None):
|
|
667 |
all_stats = []
|
668 |
|
669 |
param_grid = ParameterGrid(test_params)
|
670 |
-
|
671 |
for params in tqdm(param_grid, desc="Running tests"):
|
672 |
chunks, embedding_model, num_tokens = process_files(
|
673 |
file.name if file else None,
|
@@ -690,7 +690,7 @@ def automated_testing(file, query, test_params, expected_result=None):
|
|
690 |
chunks = optimized_chunks
|
691 |
|
692 |
if params['use_query_optimization']:
|
693 |
-
optimized_queries = optimize_query(query, params['query_optimization_model'],
|
694 |
|
695 |
#optimized_queries = optimize_query(query, )
|
696 |
query = " ".join(optimized_queries)
|
|
|
667 |
all_stats = []
|
668 |
|
669 |
param_grid = ParameterGrid(test_params)
|
670 |
+
print(param_grid)
|
671 |
for params in tqdm(param_grid, desc="Running tests"):
|
672 |
chunks, embedding_model, num_tokens = process_files(
|
673 |
file.name if file else None,
|
|
|
690 |
chunks = optimized_chunks
|
691 |
|
692 |
if params['use_query_optimization']:
|
693 |
+
optimized_queries = optimize_query(query, params['query_optimization_model'], chunks , embedding_model , params['vector_store_type'] , params['search_type'] , params['top_k'] )
|
694 |
|
695 |
#optimized_queries = optimize_query(query, )
|
696 |
query = " ".join(optimized_queries)
|