feat: новая функция

backend/services/notifyOllamaReady.js

@@ -0,0 +1,166 @@
+/**
+ * Copyright (c) 2024-2025 Тарабанов Александр Викторович
+ * All rights reserved.
+ * 
+ * This software is proprietary and confidential.
+ * Unauthorized copying, modification, or distribution is prohibited.
+ * 
+ * For licensing inquiries: info@hb3-accelerator.com
+ * Website: https://hb3-accelerator.com
+ * GitHub: https://github.com/HB3-ACCELERATOR
+ */
+
+const axios = require('axios');
+const logger = require('../utils/logger');
+
+/**
+ * Скрипт для уведомления Ollama о готовности
+ * Используется для проверки доступности Ollama и прогрева моделей
+ */
+
+const OLLAMA_HOST = process.env.OLLAMA_HOST || 'http://ollama:11434';
+const MAX_RETRIES = 30;
+const RETRY_DELAY = 2000; // 2 секунды
+
+/**
+ * Проверить доступность Ollama
+ * @returns {Promise<boolean>}
+ */
+async function checkOllamaHealth() {
+  try {
+    const response = await axios.get(`${OLLAMA_HOST}/api/tags`, {
+      timeout: 5000
+    });
+    
+    return response.status === 200;
+  } catch (error) {
+    return false;
+  }
+}
+
+/**
+ * Дождаться готовности Ollama с retry
+ * @returns {Promise<boolean>}
+ */
+async function waitForOllama() {
+  logger.info('[NotifyOllamaReady] Ожидание готовности Ollama...');
+  
+  for (let i = 0; i < MAX_RETRIES; i++) {
+    const isReady = await checkOllamaHealth();
+    
+    if (isReady) {
+      logger.info(`[NotifyOllamaReady] ✅ Ollama готов! (попытка ${i + 1}/${MAX_RETRIES})`);
+      return true;
+    }
+    
+    logger.info(`[NotifyOllamaReady] Ollama не готов, повтор ${i + 1}/${MAX_RETRIES}...`);
+    await new Promise(resolve => setTimeout(resolve, RETRY_DELAY));
+  }
+  
+  logger.error('[NotifyOllamaReady] ❌ Ollama не стал доступен после всех попыток');
+  return false;
+}
+
+/**
+ * Получить список доступных моделей
+ * @returns {Promise<Array>}
+ */
+async function getAvailableModels() {
+  try {
+    const response = await axios.get(`${OLLAMA_HOST}/api/tags`, {
+      timeout: 5000
+    });
+    
+    return response.data.models || [];
+  } catch (error) {
+    logger.error('[NotifyOllamaReady] Ошибка получения моделей:', error.message);
+    return [];
+  }
+}
+
+/**
+ * Прогреть модель (загрузить в память)
+ * @param {string} modelName - Название модели
+ * @returns {Promise<boolean>}
+ */
+async function warmupModel(modelName) {
+  try {
+    logger.info(`[NotifyOllamaReady] Прогрев модели: ${modelName}`);
+    
+    const response = await axios.post(`${OLLAMA_HOST}/api/generate`, {
+      model: modelName,
+      prompt: 'Hello',
+      stream: false
+    }, {
+      timeout: 30000
+    });
+    
+    if (response.status === 200) {
+      logger.info(`[NotifyOllamaReady] ✅ Модель ${modelName} прогрета`);
+      return true;
+    }
+    
+    return false;
+  } catch (error) {
+    logger.error(`[NotifyOllamaReady] Ошибка прогрева модели ${modelName}:`, error.message);
+    return false;
+  }
+}
+
+/**
+ * Основная функция инициализации
+ */
+async function initialize() {
+  try {
+    logger.info('[NotifyOllamaReady] 🚀 Начало инициализации Ollama...');
+    
+    // Ждем готовности Ollama
+    const isReady = await waitForOllama();
+    
+    if (!isReady) {
+      logger.error('[NotifyOllamaReady] Не удалось дождаться готовности Ollama');
+      return false;
+    }
+    
+    // Получаем список моделей
+    const models = await getAvailableModels();
+    logger.info(`[NotifyOllamaReady] Найдено моделей: ${models.length}`);
+    
+    if (models.length > 0) {
+      logger.info('[NotifyOllamaReady] Доступные модели:', models.map(m => m.name).join(', '));
+      
+      // Прогреваем первую модель (опционально)
+      if (process.env.WARMUP_MODEL === 'true' && models[0]) {
+        await warmupModel(models[0].name);
+      }
+    }
+    
+    logger.info('[NotifyOllamaReady] ✅ Инициализация завершена');
+    return true;
+    
+  } catch (error) {
+    logger.error('[NotifyOllamaReady] Ошибка инициализации:', error);
+    return false;
+  }
+}
+
+// Если запущен напрямую как скрипт
+if (require.main === module) {
+  initialize()
+    .then(success => {
+      process.exit(success ? 0 : 1);
+    })
+    .catch(error => {
+      logger.error('[NotifyOllamaReady] Критическая ошибка:', error);
+      process.exit(1);
+    });
+}
+
+module.exports = {
+  initialize,
+  waitForOllama,
+  checkOllamaHealth,
+  getAvailableModels,
+  warmupModel
+};
+