132 lines
5.4 KiB
Python
132 lines
5.4 KiB
Python
"""
|
||
Configuration module for smart speaker.
|
||
Loads environment variables from .env file.
|
||
"""
|
||
|
||
# Этот модуль отвечает за конфигурацию всего проекта.
|
||
# Он загружает настройки из файла .env (переменные окружения) и определяет константы.
|
||
|
||
import os
|
||
import time
|
||
from pathlib import Path
|
||
from dotenv import load_dotenv
|
||
|
||
# Базовая директория проекта (корневая папка, где лежит .env)
|
||
BASE_DIR = Path(__file__).resolve().parents[2]
|
||
|
||
# Загружаем переменные из файла .env в корневом каталоге
|
||
load_dotenv(BASE_DIR / ".env")
|
||
|
||
# --- Настройки AI ---
|
||
# AI_PROVIDER опционален. Приоритет у единственного активного AI API key.
|
||
# Если активных ключей несколько, AI-модуль вернет ошибку конфигурации.
|
||
AI_PROVIDER = os.getenv("AI_PROVIDER", "openrouter").strip().lower()
|
||
|
||
# OpenRouter
|
||
OPENROUTER_API_KEY = os.getenv("OPENROUTER_API_KEY")
|
||
OPENROUTER_MODEL = os.getenv("OPENROUTER_MODEL", "openai/gpt-4o-mini")
|
||
OPENROUTER_API_URL = os.getenv(
|
||
"OPENROUTER_API_URL", "https://openrouter.ai/api/v1/chat/completions"
|
||
)
|
||
|
||
# OpenAI
|
||
OPENAI_API_KEY = os.getenv("OPENAI_API_KEY")
|
||
OPENAI_MODEL = os.getenv("OPENAI_MODEL", "gpt-4o-mini")
|
||
OPENAI_API_URL = os.getenv(
|
||
"OPENAI_API_URL", "https://api.openai.com/v1/chat/completions"
|
||
)
|
||
|
||
# Gemini (через официальный OpenAI-compatible endpoint Google)
|
||
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY")
|
||
GEMINI_MODEL = os.getenv("GEMINI_MODEL", "gemini-2.5-flash")
|
||
GEMINI_API_URL = os.getenv(
|
||
"GEMINI_API_URL",
|
||
"https://generativelanguage.googleapis.com/v1beta/openai/chat/completions",
|
||
)
|
||
|
||
# Z.ai
|
||
ZAI_API_KEY = os.getenv("ZAI_API_KEY")
|
||
ZAI_MODEL = os.getenv("ZAI_MODEL", "glm-5")
|
||
ZAI_API_URL = os.getenv(
|
||
"ZAI_API_URL", "https://api.z.ai/api/paas/v4/chat/completions"
|
||
)
|
||
|
||
# Anthropic Claude
|
||
ANTHROPIC_API_KEY = os.getenv("ANTHROPIC_API_KEY")
|
||
ANTHROPIC_MODEL = os.getenv("ANTHROPIC_MODEL", "claude-sonnet-4-20250514")
|
||
ANTHROPIC_API_URL = os.getenv(
|
||
"ANTHROPIC_API_URL", "https://api.anthropic.com/v1/messages"
|
||
)
|
||
ANTHROPIC_API_VERSION = os.getenv("ANTHROPIC_API_VERSION", "2023-06-01")
|
||
|
||
# --- Настройки распознавания речи (Deepgram) ---
|
||
# Ключ для облачного STT (Speech-to-Text)
|
||
DEEPGRAM_API_KEY = os.getenv("DEEPGRAM_API_KEY")
|
||
|
||
# --- Настройки активации голосом (Porcupine) ---
|
||
# Ключ доступа PicoVoice
|
||
PORCUPINE_ACCESS_KEY = os.getenv("PORCUPINE_ACCESS_KEY")
|
||
# Wake word label and common ASR aliases.
|
||
WAKE_WORD = "Waltron"
|
||
WAKE_WORD_ALIASES = (
|
||
"waltron",
|
||
"voltron",
|
||
"волтрон",
|
||
"уолтрон",
|
||
"валтрон",
|
||
)
|
||
# Путь к файлу модели ключевого слова (.ppn), который лежит в папке assets/models
|
||
PORCUPINE_KEYWORD_PATH = BASE_DIR / "assets" / "models" / "Waltron_en_linux_v4_0_0.ppn"
|
||
# Чувствительность wake word (0..1). Выше = ловит легче, но больше ложных срабатываний.
|
||
PORCUPINE_SENSITIVITY = float(os.getenv("PORCUPINE_SENSITIVITY", "0.8"))
|
||
|
||
# --- Параметры аудио ---
|
||
# Частота дискретизации для микрофона (стандарт для распознавания речи)
|
||
SAMPLE_RATE = 16000
|
||
CHANNELS = 1
|
||
|
||
# Выбор устройства ввода (микрофона).
|
||
# Если не задано, используем default input device PortAudio (если есть).
|
||
# Пример:
|
||
# - AUDIO_INPUT_DEVICE_NAME=pulse
|
||
# - AUDIO_INPUT_DEVICE_INDEX=2
|
||
AUDIO_INPUT_DEVICE_NAME = os.getenv("AUDIO_INPUT_DEVICE_NAME", "").strip() or None
|
||
_audio_index_raw = os.getenv("AUDIO_INPUT_DEVICE_INDEX", "").strip()
|
||
try:
|
||
AUDIO_INPUT_DEVICE_INDEX = int(_audio_index_raw) if _audio_index_raw else None
|
||
except Exception:
|
||
AUDIO_INPUT_DEVICE_INDEX = None
|
||
|
||
# Выбор устройства вывода (динамик).
|
||
# Если не задано, используем default output device PortAudio (если есть).
|
||
# Пример:
|
||
# - AUDIO_OUTPUT_DEVICE_NAME=pulse
|
||
# - AUDIO_OUTPUT_DEVICE_INDEX=5
|
||
AUDIO_OUTPUT_DEVICE_NAME = os.getenv("AUDIO_OUTPUT_DEVICE_NAME", "").strip() or None
|
||
_audio_out_index_raw = os.getenv("AUDIO_OUTPUT_DEVICE_INDEX", "").strip()
|
||
try:
|
||
AUDIO_OUTPUT_DEVICE_INDEX = (
|
||
int(_audio_out_index_raw) if _audio_out_index_raw else None
|
||
)
|
||
except Exception:
|
||
AUDIO_OUTPUT_DEVICE_INDEX = None
|
||
|
||
# --- Настройка времени ---
|
||
# Устанавливаем часовой пояс на Москву, чтобы будильник работал корректно
|
||
|
||
os.environ["TZ"] = "Europe/Moscow"
|
||
time.tzset()
|
||
|
||
# --- Настройки синтеза речи (TTS) ---
|
||
# Голос для русского языка (eugene - мужской голос)
|
||
TTS_SPEAKER = "eugene" # Доступные (ru): aidar, baya, kseniya, xenia, eugene
|
||
# Голос для английского языка
|
||
TTS_EN_SPEAKER = os.getenv("TTS_EN_SPEAKER", "en_0")
|
||
# Частота дискретизации для воспроизведения (качество звука)
|
||
TTS_SAMPLE_RATE = 48000
|
||
|
||
# --- Настройки погоды ---
|
||
WEATHER_LAT = os.getenv("WEATHER_LAT")
|
||
WEATHER_LON = os.getenv("WEATHER_LON")
|
||
WEATHER_CITY = os.getenv("WEATHER_CITY", "Ухта")
|