fixed query def

index_retriever.py

@@ -79,62 +79,64 @@ class IndexRetriever:
             return False
 
     def create_query_engine(self):
-            try:
-                log_message(f"Применяется промпт: {self.config.PROMPT_SIMPLE_POISK[:100]}...")
-                
-                bm25_retriever = BM25Retriever.from_defaults(
-                    docstore=self.vector_index.docstore, 
-                    similarity_top_k=15
-                )
-                
-                vector_retriever = VectorIndexRetriever(
-                    index=self.vector_index, 
-                    similarity_top_k=20,
-                    similarity_cutoff=0.5
-                )
-                
-                hybrid_retriever = QueryFusionRetriever(
-                    [vector_retriever, bm25_retriever],
-                    similarity_top_k=30,
-                    num_queries=1
-                )
-                
-                custom_prompt_template = PromptTemplate(self.config.PROMPT_SIMPLE_POISK)
-                response_synthesizer = get_response_synthesizer(
-                    response_mode=ResponseMode.TREE_SUMMARIZE,
-                    text_qa_template=custom_prompt_template
-                )
-                
-                self.query_engine = RetrieverQueryEngine(
-                    retriever=hybrid_retriever,
-                    response_synthesizer=response_synthesizer
-                )
-                
-                log_message("Query engine успешно создан с кастомным промптом")
-                
-            except Exception as e:
-                log_message(f"Ошибка создания query engine: {str(e)}")
-                raise
+        try:
+            log_message(f"Применяется промпт: {self.config.PROMPT_SIMPLE_POISK[:100]}...")
+            
+            bm25_retriever = BM25Retriever.from_defaults(
+                docstore=self.vector_index.docstore, 
+                similarity_top_k=15
+            )
+            
+            vector_retriever = VectorIndexRetriever(
+                index=self.vector_index, 
+                similarity_top_k=20,
+                similarity_cutoff=0.5
+            )
+            
+            hybrid_retriever = QueryFusionRetriever(
+                [vector_retriever, bm25_retriever],
+                similarity_top_k=30,
+                num_queries=1
+            )
+            
+            custom_prompt_template = PromptTemplate(self.config.PROMPT_SIMPLE_POISK)
+            response_synthesizer = get_response_synthesizer(
+                response_mode=ResponseMode.TREE_SUMMARIZE,
+                text_qa_template=custom_prompt_template
+            )
+            
+            self.query_engine = RetrieverQueryEngine(
+                retriever=hybrid_retriever,
+                response_synthesizer=response_synthesizer
+            )
+            
+            log_message("Query engine успешно создан с кастомным промптом")
+            
+        except Exception as e:
+            log_message(f"Ошибка создания query engine: {str(e)}")
+            raise
 
-            def query(self, question):
-                """Метод для выполнения запроса с применением промпта"""
-                if self.query_engine is None:
-                    log_message("❌ Query engine не инициализирован")
-                    return "❌ Система не инициализирована"
-                
-                try:
-                    log_message(f"Обрабатываю запрос: {question}")
-                    log_message(f"Используется промпт: {self.config.PROMPT_SIMPLE_POISK[:100]}...")
-                    
-                    response = self.query_engine.query(question)
-                    log_message(f"Ответ получен, длина: {len(str(response))}")
-                    
-                    return str(response)
-                    
-                except Exception as e:
-                    error_msg = f"Ошибка обработки запроса: {str(e)}"
-                    log_message(error_msg)
-                    return f"❌ {error_msg}"
+    def query(self, question):
+        """Метод для выполнения запроса с применением промпта"""
+        if self.query_engine is None:
+            log_message("❌ Query engine не инициализирован")
+            return "❌ Система не инициализирована"
+        
+        try:
+            log_message(f"Получен вопрос: {question}")
+            log_message(f"Используется модель: {self.current_model}")
+            log_message(f"Применяется промпт: {self.config.PROMPT_SIMPLE_POISK[:150]}...")
+            log_message(f"Обрабатываю запрос: {question}")
+            
+            response = self.query_engine.query(question)
+            log_message(f"Ответ по��учен, длина: {len(str(response))}")
+            
+            return str(response)
+            
+        except Exception as e:
+            error_msg = f"Ошибка обработки запроса: {str(e)}"
+            log_message(error_msg)
+            return f"❌ {error_msg}"
 
     def switch_model(self, model_name):
         try: