[FEAT] minIO test.txt 파일 받기

Author: bokyeong0405

api/main.py

@@ -1,5 +1,5 @@
 from fastapi import APIRouter
-from api.routes import user, ranking, question, theme, notification, answer, discussion, comment, badges
+from api.routes import user, ranking, question, theme, notification, answer, discussion, comment, badges, download_model
 
 api_router = APIRouter()
 
@@ -11,4 +11,5 @@
 api_router.include_router(answer.router, prefix='/answer', tags=["answer"])
 api_router.include_router(discussion.router, prefix='/discussion', tags=["discussion"])
 api_router.include_router(comment.router, prefix='/comment', tags=["comment"])
-api_router.include_router(badges.router, prefix='/badges', tags=["badges"])
+api_router.include_router(badges.router, prefix='/badges', tags=["badges"])
+api_router.include_router(download_model.router, prefix='/download_model', tags=["download_model"])

api/routes/download_model.py

@@ -0,0 +1,22 @@
+from fastapi import APIRouter
+# from models.model_loader import llama_model
+from core.minio_service import download_model_from_minio
+import os
+
+router = APIRouter()
+
+@router.get("/")
+async def download_model():
+    """
+    MinIO에서 모델 다운로드 API
+    """
+    model_s3_path = "test_folder/test.txt"  # MinIO에 저장된 경로
+    local_path = "./downloaded_model/test.txt" # local 저장 경로
+
+    result = download_model_from_minio(model_s3_path, local_path)
+
+    if result:
+        return {"message": "✅ 모델 다운로드 성공!", "local_path": result}
+    else:
+        return {"error": "❌ 모델 다운로드 실패"}
+

core/minio_service.py

@@ -0,0 +1,34 @@
+import boto3
+import os
+from dotenv import load_dotenv
+
+# .env 로드
+load_dotenv(override=True)
+
+MINIO_URL = os.getenv("MINIO_URL")
+AWS_ACCESS_KEY_ID = os.getenv("AWS_ACCESS_KEY_ID")
+AWS_SECRET_ACCESS_KEY = os.getenv("AWS_SECRET_ACCESS_KEY")
+S3_BUCKET_NAME = os.getenv("S3_BUCKET_NAME")
+
+# MinIO 클라이언트 생성
+s3 = boto3.client(
+    "s3",
+    endpoint_url=f"http://{MINIO_URL}",
+    aws_access_key_id=AWS_ACCESS_KEY_ID,
+    aws_secret_access_key=AWS_SECRET_ACCESS_KEY
+)
+
+def download_model_from_minio(model_path: str, local_save_path: str):
+    """
+    MinIO에서 파일 다운로드 (예: test.txt)
+    :param model_path: MinIO 내 저장된 파일 경로 (예: test_folder/test.txt)
+    :param local_save_path: 로컬에 저장할 경로 (예: ./downloaded_model/test.txt)
+    """
+    try:
+        os.makedirs(os.path.dirname(local_save_path), exist_ok=True)  # 로컬 저장 폴더 생성
+        s3.download_file(S3_BUCKET_NAME, model_path, local_save_path)
+        print(f"✅ {model_path} 다운로드 완료 → {local_save_path}")
+        return local_save_path
+    except Exception as e:
+        print(f"❌ MinIO에서 {model_path} 다운로드 실패: {e}")
+        return None

downloaded_model/test.txt

@@ -0,0 +1 @@
+MinIO ���� �׽�Ʈ

models/model_loader.py

@@ -0,0 +1,26 @@
+import os
+import torch
+from config import minio_client, bucket_name
+
+MODEL_PATH = "models/llama_model.safetensors"
+
+# 모델 다운로드 함수
+def download_model():
+    """MinIO에서 모델을 다운로드하여 로컬에 저장"""
+    if not os.path.exists(MODEL_PATH):
+        print("Downloading model from MinIO...")
+        minio_client.fget_object(bucket_name, "llama/finetuned_model.safetensors", MODEL_PATH)
+        print("Model downloaded.")
+
+# 모델 로드 함수
+def load_model():
+    """모델을 로드하고 FastAPI에서 사용 가능하게 설정"""
+    download_model()  # MinIO에서 모델 다운로드
+    print("Loading Llama model...")
+    model = torch.load(MODEL_PATH, map_location="cpu")  # CPU 또는 GPU로 로드
+    model.eval()  # 모델을 평가 모드로 설정
+    print("Model loaded successfully!")
+    return model
+
+# 모델 로드
+llama_model = load_model()

requirements.txt

@@ -7,4 +7,8 @@ sqlalchemy
 pymysql
 websockets
 redis
-httpx
+httpx
+torch
+minio
+boto3
+dotenv