import os
from huggingface_hub import snapshot_download

def download_bge_model():
    # 모델 저장 경로 설정
    model_path = "./ai_models/hf/BGE-m3-ko"
    
    # 디렉토리가 없으면 생성
    os.makedirs(model_path, exist_ok=True)
    
    # 모델 다운로드
    snapshot_download(
        repo_id="dragonkue/BGE-m3-ko",
        local_dir=model_path,
        revision="main"
    )
    
    print(f"모델이 {model_path}에 다운로드되었습니다.")

if __name__ == "__main__":
    download_bge_model()