MrSimple01 commited on
Commit
80b469e
·
verified ·
1 Parent(s): abb0a7b

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +3 -1
utils.py CHANGED
@@ -201,7 +201,7 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
201
 
202
  normalized_question = normalize_text(question)
203
  log_message(f"Normalized question: {normalized_question}")
204
- normalized_question_2 = normalize_steel_designations(normalized_question)
205
  log_message(f"After steel normalization: {normalized_question_2}")
206
 
207
  if query_engine is None:
@@ -213,6 +213,8 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
213
  log_message(f"user query: {question}")
214
  log_message(f"normalized query: {normalized_question}")
215
  log_message(f"after steel normalization: {normalized_question_2}")
 
 
216
 
217
  log_message(f"RETRIEVED: {len(retrieved_nodes)} nodes")
218
 
 
201
 
202
  normalized_question = normalize_text(question)
203
  log_message(f"Normalized question: {normalized_question}")
204
+ normalized_question_2, query_changes = normalize_steel_designations(question)
205
  log_message(f"After steel normalization: {normalized_question_2}")
206
 
207
  if query_engine is None:
 
213
  log_message(f"user query: {question}")
214
  log_message(f"normalized query: {normalized_question}")
215
  log_message(f"after steel normalization: {normalized_question_2}")
216
+ log_message(f"Steel grades normalized in query: {query_changes}")
217
+
218
 
219
  log_message(f"RETRIEVED: {len(retrieved_nodes)} nodes")
220