refactor: исправил вызов api в боте

Author: oleg444556

answer/handlers/ask_bot.py

@@ -44,7 +44,7 @@ async def handle_any_message(message: Message):
 
         search_message = await message.answer("🔍 Ищу информацию и готовлю развернутый ответ...")
 
-        api_result = bot_service.generate_response(
+        api_result = await bot_service.generate_response(
             text=validated_question, chat_id=str(message.chat.id), generate_ai_response=True
         )
 

answer/routes/base.py

@@ -175,6 +175,7 @@ async def init_resources():
         "vector_store": app.state.vector_store,
         "ensemble_retriever": app.state.ensemble_retriever,
         "keywords_dict": app.state.keywords_dict,
+        "text_preprocessor": app.state.text_preprocessor,
     }
     search_service.set_app_state(app_state_dict)
 
@@ -191,27 +192,16 @@ async def shutdown_resources():
 async def generate_response(user_input: UserInput):
     if not user_input.text:
         raise HTTPException(status_code=400, detail="Text cannot be empty")
-    
-    if user_input.generate_ai_response:
-        ensemble_retriever = app.state.ensemble_retriever
-    else:
-        ensemble_retriever = app.state.filtered_ensemble_retriever
-    
-    processed_text = app.state.text_preprocessor.preprocess(user_input.text)
-        
-    results, combined_text = get_context(
-        query=processed_text,
-        key_words_dict=app.state.keywords_dict,
-        ensemble_retriever=ensemble_retriever,
-        vector_store=app.state.vector_store,
-        ensemble_k=settings.ensemble_k,
-        verbose=True,
+
+    response = await bot_service.generate_response(
+        text=user_input.text,
+        generate_ai_response=user_input.generate_ai_response,
     )
 
-    if results is None:
+    if response is None:
         raise HTTPException(status_code=500, detail="Ошибка генерации ответа")
 
-    return results
+    return response
 
 
 @app.post("/users", response_model=UserResponse)

answer/services/bot_service.py

@@ -80,7 +80,7 @@ def get_or_create_user(self, chat_id: str) -> Tuple[Optional[Dict], bool]:
             logger.error(f"Ошибка получения/создания пользователя: {e}", exc_info=True)
             return None, False
 
-    def generate_response(
+    async def generate_response(
         self, text: str, chat_id: str = "", generate_ai_response: bool = False
     ) -> Optional[Dict[str, Any]]:
         """
@@ -106,15 +106,21 @@ def generate_response(
                 else app_state.get("filtered_ensemble_retriever", app_state["ensemble_retriever"])
             )
 
+            processed_text = app_state["text_preprocessor"].preprocess(text)
+
             results, combined_text = get_context(
-                query=text,
+                query=processed_text,
                 key_words_dict=app_state["keywords_dict"],
                 ensemble_retriever=ensemble_retriever,
                 vector_store=app_state["vector_store"],
                 ensemble_k=settings.ensemble_k,
                 verbose=True,
             )
 
+            if results is None:
+                logger.error("Ошибка генерации ответа от get_context")
+                return None
+
             formatted_results = [
                 {
                     "topic": getattr(r, "topic", ""),