gardio_test / config.py
dev-yuje's picture
fix
53616e0
raw
history blame contribute delete
657 Bytes
import os
# ============================================================
# ํ”„๋กœ์ ํŠธ ๊ณตํ†ต ์„ค์ •
# ============================================================
# LLM ์„ค์ • (Google Gemini ์‚ฌ์šฉ)
LLM_MODEL = "gemini-2.5-flash-lite"
# Google AI Studio API ํ‚ค
# ๋ณด์•ˆ์ƒ ํ™˜๊ฒฝ๋ณ€์ˆ˜์—์„œ ๋จผ์ € ์ฐพ๊ณ , ์—†์œผ๋ฉด ํ•˜๋“œ์ฝ”๋”ฉ๋œ ๊ฐ’์„ ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค.
# ์ž„๋ฒ ๋”ฉ ๋ชจ๋ธ (FAISS ์ธ๋ฑ์Šค์™€ ์ผ์น˜ํ•ด์•ผ ํ•จ)
EMBEDDING_MODEL = "intfloat/multilingual-e5-small"
# FAISS ์ธ๋ฑ์Šค ๊ฒฝ๋กœ
FAISS_PATH = "./faiss"
# PDF ์†Œ์Šค ๊ฒฝ๋กœ
PDF_SOURCE = "./sources/KCB_Credit_Evaluation_System.pdf"
# FAISS ๊ฒ€์ƒ‰ ์‹œ ๋ฐ˜ํ™˜ํ•  ๋ฌธ์„œ ์ˆ˜
RETRIEVER_K = 3