enhanced the prompt with metadata

utils.py

@@ -43,6 +43,48 @@ def get_embedding_model(model_name="sentence-transformers/paraphrase-multilingua
 def get_reranker_model(model_name='cross-encoder/ms-marco-MiniLM-L-12-v2'):
     return CrossEncoder(model_name)
 
+def format_context_for_llm(nodes):
+    context_parts = []
+    
+    for node in nodes:
+        metadata = node.metadata if hasattr(node, 'metadata') else {}
+        doc_id = metadata.get('document_id', 'Неизвестный документ')
+        
+        section_info = ""
+        if metadata.get('section'):
+            section_info = f"пункт {metadata['section']}"
+        elif metadata.get('paragraph'):
+            section_info = f"пункт {metadata['paragraph']}"
+        elif metadata.get('clause'):
+            section_info = f"пункт {metadata['clause']}"
+        elif metadata.get('item'):
+            section_info = f"пункт {metadata['item']}"
+        elif metadata.get('page'):
+            section_info = f"страница {metadata['page']}"
+        
+        if metadata.get('type') == 'table' and metadata.get('table_number'):
+            table_num = metadata['table_number']
+            if not str(table_num).startswith('№'):
+                table_num = f"№{table_num}"
+            section_info = f"таблица {table_num}"
+        
+        if metadata.get('type') == 'image' and metadata.get('image_number'):
+            image_num = metadata['image_number']
+            if not str(image_num).startswith('№'):
+                image_num = f"№{image_num}"
+            section_info = f"рисунок {image_num}"
+        
+        context_text = node.text if hasattr(node, 'text') else str(node)
+        
+        if section_info:
+            formatted_context = f"[ИСТОЧНИК: {section_info} документа {doc_id}]\n{context_text}\n"
+        else:
+            formatted_context = f"[ИСТОЧНИК: документ {doc_id}]\n{context_text}\n"
+        
+        context_parts.append(formatted_context)
+    
+    return "\n".join(context_parts)
+
 def generate_sources_html(nodes, chunks_df=None):
     html = "<div style='background-color: #2d3748; color: white; padding: 20px; border-radius: 10px; max-height: 400px; overflow-y: auto;'>"
     html += "<h3 style='color: #63b3ed; margin-top: 0;'>Источники:</h3>"
@@ -92,7 +134,6 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
     try:
         log_message(f"Получен вопрос: {question}")
         log_message(f"Используется модель: {current_model}")
-        log_message(f"Используется промпт: {PROMPT_SIMPLE_POISK[:200]}...")
         start_time = time.time()
         
         log_message("Извлекаю релевантные узлы")
@@ -102,8 +143,19 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
         log_message("Применяю переранжировку")
         reranked_nodes = rerank_nodes(question, retrieved_nodes, reranker, top_k=10)
         
+        formatted_context = format_context_for_llm(reranked_nodes)
+        
+        enhanced_question = f"""
+Контекст из базы данных:
+{formatted_context}
+
+Вопрос пользователя: {question}
+
+Инструкции: Отвечай строго на основе предоставленного контекста. Обязательно указывай источники в формате "по пункту X документа Y" или "согласно разделу X документа Y". Если информация из нескольких источников, перечисли их все.
+"""
+        
         log_message(f"Отправляю запрос в LLM с {len(reranked_nodes)} узлами")
-        response = query_engine.query(question)
+        response = query_engine._llm.complete(enhanced_question)
         
         end_time = time.time()
         processing_time = end_time - start_time
@@ -114,7 +166,7 @@ def answer_question(question, query_engine, reranker, current_model, chunks_df=N
         
         answer_with_time = f"""<div style='background-color: #2d3748; color: white; padding: 20px; border-radius: 10px; margin-bottom: 10px;'>
         <h3 style='color: #63b3ed; margin-top: 0;'>Ответ (Модель: {current_model}):</h3>
-        <div style='line-height: 1.6; font-size: 16px;'>{response.response}</div>
+        <div style='line-height: 1.6; font-size: 16px;'>{response.text}</div>
         <div style='margin-top: 15px; padding-top: 10px; border-top: 1px solid #4a5568; font-size: 14px; color: #a0aec0;'>
         Время обработки: {processing_time:.2f} секунд
         </div>