Add Hugging Face token validation and improve error handling in chat history saving

app.py

@@ -5,7 +5,10 @@ from config.constants import DEFAULT_SYSTEM_MESSAGE
 from config.settings import DEFAULT_MODEL, HF_TOKEN
 from src.knowledge_base.vector_store import create_vector_store, load_vector_store
 
-# Создаем клиент для инференса с токеном
+if not HF_TOKEN:
+    raise ValueError("HUGGINGFACE_TOKEN not found in environment variables")
+
+# Initialize HF client with token
 client = InferenceClient(
     DEFAULT_MODEL,
     token=HF_TOKEN
@@ -92,12 +95,17 @@ def respond(
             messages.append({"role": "assistant", "content": response})
             try:
                 from src.knowledge_base.dataset import DatasetManager
-                dataset = DatasetManager()
+                from config.settings import HF_TOKEN
+                
+                dataset = DatasetManager(token=HF_TOKEN)  # Explicitly pass the token
                 success, msg = dataset.save_chat_history(conversation_id, messages)
+                print(f"Chat history save attempt: {success}, Message: {msg}")  # Add debug log
                 if not success:
-                    print(f"Ошибка при сохранении истории чата: {msg}")
+                    print(f"Failed to save chat history: {msg}")
             except Exception as e:
-                print(f"Ошибка при сохранении истории чата: {str(e)}")
+                import traceback
+                print(f"Exception while saving chat history: {str(e)}")
+                print(traceback.format_exc())  # Print full traceback for debugging
             
     except Exception as e:
         print(f"Ошибка при генерации ответа: {str(e)}")

src/knowledge_base/dataset.py

@@ -8,23 +8,23 @@ import tempfile
 from typing import Tuple, List, Dict, Any, Optional
 from datetime import datetime
 from huggingface_hub import HfApi, HfFolder
-from config.settings import VECTOR_STORE_PATH
+from config.settings import VECTOR_STORE_PATH, HF_TOKEN
 
 class DatasetManager:
     def __init__(self, dataset_name="Rulga/status-law-knowledge-base", token: Optional[str] = None):
         """
-        Инициализация менеджера датасетов
+        Initialize dataset manager
         
         Args:
-            dataset_name: Имя датасета на Hugging Face Hub
-            token: Токен доступа к Hugging Face Hub (если не задан, берется из ~/.huggingface/token)
+            dataset_name: Hugging Face Hub dataset name
+            token: Hugging Face access token (if None, will use HF_TOKEN from settings)
         """
-        self.token = token if token else HfFolder.get_token()
+        self.token = token if token else HF_TOKEN
         if not self.token:
-            raise ValueError("Не найден токен Hugging Face. Установите переменную окружения HUGGINGFACE_TOKEN")
+            raise ValueError("Hugging Face token not found. Please set HUGGINGFACE_TOKEN environment variable")
             
-        self.api = HfApi(token=self.token)
         self.dataset_name = dataset_name
+        self.api = HfApi(token=self.token)
         
         # Проверяем/создаем репозиторий при инициализации
         try: