Initial commit

.gitignore

@@ -8,10 +8,6 @@ __pycache__/
 venv/
 .env
 
-# 로컬 DB / 캐시
-chroma_db/
-*.sqlite3
-
 # IDE / OS
 .vscode/
 .idea/

config.py

@@ -15,3 +15,6 @@ EMBEDDER_MODEL_NAME = os.getenv("EMBEDDER_MODEL_NAME", "sentence-transformers/pa
 EMOTION_MODEL_DIR = Path(os.getenv("EMOTION_MODEL_DIR", BASE_DIR / "models" / "emotion-classification-model"))
 FALLBACK_MODEL_DIR = Path(os.getenv("FALLBACK_MODEL_DIR", BASE_DIR / "models" / "fallback-npc-model"))
 EMBEDDER_MODEL_DIR = Path(os.getenv("EMBEDDER_MODEL_DIR", BASE_DIR / "models" / "sentence-embedder"))
+
+# ChromaDB 디렉토리
+CHROMA_DIR = Path(os.getenv("CHROMA_DIR", BASE_DIR / "rag" / "chroma_DB"))

rag/rag_manager.py

@@ -2,8 +2,9 @@ import os, json
 from typing import List, Dict, Any, Optional
 from chromadb import PersistentClient
 from chromadb.utils.embedding_functions import EmbeddingFunction
+from config import CHROMA_DIR
 
-_client = PersistentClient(path="./rag")
+_client = PersistentClient(path=str(CHROMA_DIR))
 _collection = _client.get_or_create_collection(name="game_docs")
 _embedder: Optional[EmbeddingFunction] = None