new r + stee product expansion

utils.py

@@ -195,20 +195,57 @@ def debug_search_tables(vector_index, search_term="С-25"):
     
     return matching
 
+GENERIC_STEEL_CONTEXT = "стандарт ГОСТ технические условия марка материал применение сварка"
+
+
 from documents_prep import normalize_text, normalize_steel_designations
 
+STEEL_PRODUCT_EXPANSIONS = {
+    "08X18H10T": ["Листы", "Трубы", "Поковки", "Крепежные изделия", "Сортовой прокат", "Отливки"],
+    "12X18H10T": ["Листы", "Поковки", "Сортовой прокат"],
+    "10X17H13M2T": ["Трубы", "Арматура", "Поковки", "Фланцы"],
+    "20X23H18": ["Листы", "Сортовой прокат", "Поковки"],
+    "03X17H14M3": ["Трубы", "Листы", "Проволока"]
+}
+
+
 def enhance_query_for_steel_grades(query):
-    """Expand query with related terms for better steel grade retrieval"""
-    import re
-    
-    # Detect if query contains steel grades
-    steel_pattern = r'\b\d{1,3}[XHТCВKMAPХНТСВКМАР]\d*[XHТCВKMAPХНТСВКМАР\d]*\b'
+    """Expand query with steel grade specific context"""
+    import re    
+    # Detect steel grades in query
+    steel_pattern = r'\b[СC][ВB]-\d{1,3}(?:[A-ZА-ЯЁ]\d*)+\b'
     matches = re.findall(steel_pattern, query, re.IGNORECASE)
+
+    if not matches:
+        return query
     
-    if matches:
-        # Add contextual terms
-        enhanced = query + " стандарт материал марка стали применение"
-        log_message(f"Enhanced query with steel context: {enhanced}")
+    # Collect context expansions
+    added_context = []
+    grades_found = []
+    
+    for match in matches:
+        match_upper = match.upper()
+        grades_found.append(match_upper)
+        
+        # Check if we have specific context for this grade
+        if match_upper in STEEL_PRODUCT_EXPANSIONS:
+            context = STEEL_PRODUCT_EXPANSIONS[match_upper]
+            added_context.append(context)
+            log_message(f"  Found specific context for {match_upper}")
+        else:
+            # Use generic context for unknown grades
+            added_context.append(GENERIC_STEEL_CONTEXT)
+            log_message(f"  Using generic context for {match_upper}")
+    
+    # Build enhanced query
+    if added_context:
+        # Remove duplicates from context
+        unique_context = ' '.join(set(' '.join(added_context).split()))
+        enhanced = f"{query} {unique_context}"
+        
+        log_message(f"Enhanced query for steel grades: {', '.join(grades_found)}")
+        log_message(f"Added context: {unique_context[:100]}...")
+        
         return enhanced
     
     return query