config.py

"""
모임통장 AI 에이전트 설정 파일
"""
import os
from dotenv import load_dotenv

# .env 파일이 있으면 로드 (선택사항)
load_dotenv()

# API Keys - 환경변수 우선, 필요 시 api_keys.py
IGNORE_API_KEYS = os.getenv("IGNORE_API_KEYS", "0") == "1"
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
TAVILY_API_KEY = os.getenv("TAVILY_API_KEY")

if not IGNORE_API_KEYS and not GOOGLE_API_KEY:
    try:
        from api_keys import GOOGLE_API_KEY as FILE_GOOGLE_API_KEY, TAVILY_API_KEY as FILE_TAVILY_API_KEY
        GOOGLE_API_KEY = GOOGLE_API_KEY or FILE_GOOGLE_API_KEY
        TAVILY_API_KEY = TAVILY_API_KEY or FILE_TAVILY_API_KEY
        print("✅ api_keys.py에서 API 키를 성공적으로 로드했습니다.")
    except ImportError:
        print("⚠️ api_keys.py를 찾을 수 없습니다. 환경변수에서 로드를 시도합니다.")

SERPAPI_API_KEY = os.getenv("SERPAPI_API_KEY")

# Model Settings
GEMINI_MODEL_NAME = "gemini-2.0-flash-exp"

# RAG Settings
CHUNK_SIZE = 1000
CHUNK_OVERLAP = 200
TOP_K_RESULTS = 3

# Data Paths
KB_DATASET_PATH = "data/kb_products.csv"
VECTOR_STORE_PATH = "data/vector_store"

# Streamlit Settings
PAGE_TITLE = "모임통장 AI 어드바이저"
PAGE_ICON = "💰"