MrSimple07 commited on
Commit
c3088ad
·
1 Parent(s): 823efe5

new prompt for query enhancement + 4500, 30

Browse files
Files changed (2) hide show
  1. config.py +9 -7
  2. utils.py +2 -1
config.py CHANGED
@@ -55,16 +55,18 @@ MAX_CHARS_TABLE = 4500
55
  MAX_ROWS_TABLE = 30
56
 
57
 
58
- QUERY_EXPANSION_PROMPT = """Ты помощник для расширения поисковых запросов.
59
-
60
  Пользователь задал вопрос: "{original_query}"
61
 
62
- Сгенерируй 5 альтернативных формулировок этого же вопроса, которые помогут найти релевантную информацию в технической документации. Формулировки должны быть:
63
- - Близкими по смыслу к оригинальному вопросу
64
- - Использовать технические термины и синонимы
65
- - Быть краткими (5-10 слов)
 
 
66
 
67
- Верни ТОЛЬКО 5 вопросов, каждый с новой строки, без нумерации и объяснений."""
68
 
69
  CUSTOM_PROMPT = """
70
  Вы являетесь высокоспециализированным Ассистентом для анализа нормативных документов (AIEXP). Ваша цель - предоставлять точные, корректные и контекстно релевантные ответы исключительно на основе предоставленного контекста из нормативной документации.
 
55
  MAX_ROWS_TABLE = 30
56
 
57
 
58
+ QUERY_EXPANSION_PROMPT = """Ты — интеллектуальный помощник для расширения поисковых запросов по стандартам ГОСТ, ТУ, ISO, EN и другой технической документации.
59
+ Твоя цель — помочь системе найти все возможные формулировки вопроса, включая те, где встречаются редкие или неочевидные термины.
60
  Пользователь задал вопрос: "{original_query}"
61
 
62
+ Сгенерируй 5 вариантов запроса, которые:
63
+ Сохраняют смысл исходного вопроса
64
+ Используют синонимы и технические термины (например: "сталь" → "сплав", "марка", "материал")
65
+ Добавляют возможные контекстные уточнения (например: "ГОСТ", "ТУ", "марка", "лист", "труба", "прокат", "применение", "химический состав")
66
+ Могут охватывать как частотные, так и редкие термины
67
+ Краткие — не более 10 слов каждая
68
 
69
+ Верни только 5 запросов, каждый с новой строки, без нумерации и пояснений."""
70
 
71
  CUSTOM_PROMPT = """
72
  Вы являетесь высокоспециализированным Ассистентом для анализа нормативных документов (AIEXP). Ваша цель - предоставлять точные, корректные и контекстно релевантные ответы исключительно на основе предоставленного контекста из нормативной документации.
utils.py CHANGED
@@ -205,7 +205,8 @@ STEEL_PRODUCT_EXPANSIONS = {
205
  "12X18H10T": ["Листы", "Поковки", "Сортовой прокат"],
206
  "10X17H13M2T": ["Трубы", "Арматура", "Поковки", "Фланцы"],
207
  "20X23H18": ["Листы", "Сортовой прокат", "Поковки"],
208
- "03X17H14M3": ["Трубы", "Листы", "Проволока"]
 
209
  }
210
 
211
 
 
205
  "12X18H10T": ["Листы", "Поковки", "Сортовой прокат"],
206
  "10X17H13M2T": ["Трубы", "Арматура", "Поковки", "Фланцы"],
207
  "20X23H18": ["Листы", "Сортовой прокат", "Поковки"],
208
+ "03X17H14M3": ["Трубы", "Листы", "Проволока"],
209
+ "59023.6": ["Режимы термической обработки стали 59023.6"],
210
  }
211
 
212