Upload folder using huggingface_hub

ask.py

@@ -0,0 +1,19 @@
+from fastapi import APIRouter
+from pydantic import BaseModel
+from services.rag_pipeline import rag_pipeline
+
+router = APIRouter()
+
+# 요청 스키마
+class AskRequest(BaseModel):
+    query: str
+    top_k: int = 5
+
+# 응답 스키마
+class AskResponse(BaseModel):
+    output: str
+
+@router.post("/ask", response_model=AskResponse)
+async def ask(request: AskRequest):
+    output = rag_pipeline(request.query, top_k=request.top_k)
+    return {"output": output}

generation.py

@@ -0,0 +1,21 @@
+from fastapi import APIRouter
+from pydantic import BaseModel
+from generator.prompt_builder import build_prompt
+from generator.llm_inference import generate_answer
+
+router = APIRouter()
+
+# 요청 스키마
+class GenerateRequest(BaseModel):
+    query: str
+    context_docs: list
+
+# 응답 스키마
+class GenerateResponse(BaseModel):
+    output: str
+
+@router.post("/generate", response_model=GenerateResponse)
+async def generate(request: GenerateRequest):
+    prompt = build_prompt(request.query, request.context_docs)
+    output = generate_answer(prompt)
+    return {"output": output}

main.py

@@ -0,0 +1,34 @@
+from fastapi import FastAPI
+from fastapi.middleware.cors import CORSMiddleware
+from api.retrieval import router as retrieval_router
+from api.generation import router as generation_router
+from api.ask import router as ask_router
+
+app = FastAPI(
+    title="RAG 기반 문제 출제 시스템",
+    description="공인중개사 시험 문제 자동 생성 API",
+    version="1.0.0",
+)
+
+# (선택) CORS 설정 - 나중에 Next.js 프론트 연결할 때 필요함
+app.add_middleware(
+    CORSMiddleware,
+    allow_origins=["*"],  # 실제 배포 때는 프론트 도메인만 허용 추천
+    allow_credentials=True,
+    allow_methods=["*"],
+    allow_headers=["*"],
+)
+
+# 헬스 체크용 엔드포인트
+@app.get("/health")
+async def health_check():
+    return {"status": "ok"}
+
+# 기본 인덱스
+@app.get("/")
+async def root():
+    return {"message": "Welcome to the RAG Question Generator API!"}
+
+app.include_router(retrieval_router, prefix="/api")
+app.include_router(generation_router, prefix="/api")
+app.include_router(ask_router, prefix="/api")

retrieval.py

@@ -0,0 +1,19 @@
+from fastapi import APIRouter
+from pydantic import BaseModel
+from retriever.vectordb import search_documents
+
+router = APIRouter()
+
+# 요청받을 데이터 포맷 정의
+class SearchRequest(BaseModel):
+    query: str
+    top_k: int = 5  # 기본적으로 상위 5개 검색
+
+# 응답 데이터 포맷 정의
+class SearchResponse(BaseModel):
+    documents: list
+
+@router.post("/search", response_model=SearchResponse)
+async def search(request: SearchRequest):
+    docs = search_documents(request.query, top_k=request.top_k)
+    return {"documents": docs}