added type hints

app.py

@@ -248,18 +248,17 @@ def main_answer_question(question):
                 "<div style='color: black;'>Источники недоступны из-за ошибки</div>",
                 "<div style='color: black;'>Чанки недоступны из-за ошибки</div>")
 
-def retrieve_chunks(question, top_k=20):
+def retrieve_chunks(question: str, top_k: int = 20) -> list:
     from index_retriever import rerank_nodes
     global query_engine, reranker
-    
+
     if query_engine is None:
-        return "Система не инициализирована"
-    
+        return []
+
     try:
         retrieved_nodes = query_engine.retriever.retrieve(question)
         log_message(f"Получено {len(retrieved_nodes)} узлов")
-        
-        # Rerank nodes
+
         reranked_nodes = rerank_nodes(
             question,
             retrieved_nodes,
@@ -267,11 +266,10 @@ def retrieve_chunks(question, top_k=20):
             top_k=top_k,
             min_score_threshold=0.5
         )
-        
+
         chunks_data = []
         for i, node in enumerate(reranked_nodes):
             metadata = node.metadata if hasattr(node, 'metadata') else {}
-            
             chunk = {
                 'rank': i + 1,
                 'document_id': metadata.get('document_id', 'unknown'),
@@ -284,14 +282,15 @@ def retrieve_chunks(question, top_k=20):
                 'text': node.text
             }
             chunks_data.append(chunk)
-        
+
         log_message(f"Возвращено {len(chunks_data)} чанков")
         return chunks_data
-        
+
     except Exception as e:
         log_message(f"Ошибка получения чанков: {str(e)}")
-        return f"Ошибка: {str(e)}"
+        return []
 
+        
 def create_demo_interface(answer_question_func, switch_model_func, current_model, chunk_info=None):
     with gr.Blocks(title="AIEXP - AI Expert для нормативной документации", theme=gr.themes.Soft()) as demo:
         gr.api(retrieve_chunks, api_name="retrieve_chunks")