Cognitive-rag / 06_conversation_memory / config.py
config.py
Raw
"""
Configuration for Conversation Memory System.

Minimal config for embedding and Pinecone storage.
"""

from pathlib import Path

# ============================================================================
# MEMORY CONFIGURATION
# ============================================================================

# Pinecone namespace for conversation memories
CONVERSATION_MEMORY_NAMESPACE = "conversation_memory"

# NVIDIA API Key - loaded from environment or .envnvidia file
import os
_nvidia_envfile = Path(__file__).parent.parent / "04_vectoredb" / ".envnvidia"
NVIDIA_API_KEY = os.environ.get("NVIDIA_API_KEY") or (
    _nvidia_envfile.read_text().strip() if _nvidia_envfile.exists() else ""
)

# Embedding model (same as document embeddings for consistency - NVIDIA NIM)
EMBEDDING_MODEL = "nvidia/llama-3.2-nv-embedqa-1b-v2"
EMBEDDING_DIMENSION = 2048

# Session file for persistence across restarts
SESSION_FILE = Path(__file__).parent / ".conversation_session"


# ============================================================================
# RETRY CONFIGURATION
# ============================================================================

MAX_RETRIES = 3
RETRY_DELAY = 2.0  # seconds