Extracts weld type from query

utils.py

@@ -201,25 +201,33 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
     
     try:
         start_time = time.time()
-        retrieved_nodes = query_engine.retriever.retrieve(question)
-        log_message(f"user query: {question}")
-
         
+        # Use enhanced retrieval if available
+        if hasattr(query_engine, 'retrieve'):
+            # Use the enhanced retrieval that's aware of weld types
+            retrieved_nodes = query_engine.retrieve(question)
+        else:
+            # Fallback to standard retrieval
+            retrieved_nodes = query_engine.retriever.retrieve(question)
+        
+        log_message(f"user query: {question}")
         log_message(f"RETRIEVED: {len(retrieved_nodes)} nodes")
         
         unique_retrieved = deduplicate_nodes(retrieved_nodes)
-
-        # DEBUG: Log what was retrieved
         log_message(f"RETRIEVED: unique {len(unique_retrieved)} nodes")
-        for i, node in enumerate(unique_retrieved):  # All debug
+        
+        for i, node in enumerate(unique_retrieved[:15]):  # Log first 15
             table_num = node.metadata.get('table_number', 'N/A')
             table_title = node.metadata.get('table_title', 'N/A')
             doc_id = node.metadata.get('document_id', 'N/A')
             log_message(f"  [{i+1}] {doc_id} - Table {table_num}: {table_title[:50]}")
-        log_message(f"UNIQUE NODES: {len(unique_retrieved)} nodes")
         
-        # Simple reranking
-        reranked_nodes = rerank_nodes(question, unique_retrieved, reranker, top_k=20)
+        # Rerank only if we have nodes
+        if unique_retrieved:
+            reranked_nodes = rerank_nodes(question, unique_retrieved, reranker, top_k=20)
+        else:
+            reranked_nodes = []
+            log_message("WARNING: No nodes to rerank!")
         
         # Direct query without formatting
         response = query_engine.query(question)
@@ -261,5 +269,7 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
         
     except Exception as e:
         log_message(f"Ошибка: {str(e)}")
+        import traceback
+        log_message(traceback.format_exc())
         error_msg = f"<div style='background-color: #e53e3e; color: white; padding: 20px; border-radius: 10px;'>Ошибка: {str(e)}</div>"
         return error_msg, "", ""