src copy/models.py

"""Data models for the application."""

from dataclasses import dataclass

import pyaudio
from dotenv import load_dotenv

load_dotenv()


@dataclass
class AudioConfig:
    """Audio configuration settings."""

    format: int = pyaudio.paInt16
    channels: int = 1
    send_sample_rate: int = 16000
    receive_sample_rate: int = 24000
    chunk_size: int = 1024


@dataclass
class ModelConfig:
    """Gemini model configuration."""

    api_key: str
    name: str
    tools: dict
    generation_config: dict
    system_instruction: str