Arabic-Rag-Chatbot / config.py
Ahmed-Alghamdi's picture
Update config.py
8769d03 verified
raw
history blame contribute delete
859 Bytes
import os
class Config:
DOCUMENT_FILE = 'RAG DATA.txt'
# ุฎู„ูƒ ุนู„ู‰ ุงู„ู…ูˆุฏูŠู„ ุงู„ู‚ูˆูŠ ููŠ ุงู„ุจุญุซุŒ ุทุงู„ู…ุง ุงู„ุณูŠุฑูุฑ ูŠุชุญู…ู„
EMBEDDING_MODEL = 'sentence-transformers/paraphrase-multilingual-mpnet-base-v2'
BATCH_SIZE = 32
# ู‡ู†ุง ุงู„ู…ูŠุฒุฉ ุงู„ู‚ูˆูŠุฉ ููŠ Gemini: ู†ู‚ุฏุฑ ู†ูƒุจุฑ ุนุฏุฏ ุงู„ูู‚ุฑุงุช ุจุฑุงุญุชู†ุง
# ู„ุฃู†ู‡ ูŠุณุชู‚ุจู„ ุชูˆูƒู†ุฒ ูƒุซูŠุฑุฉ ุฌุฏุงู‹ ู…ุฌุงู†ุงู‹
TOP_K = 8 # ุฒุฏู†ุง ุงู„ุนุฏุฏ ู„ุถู…ุงู† ูˆุตูˆู„ ูƒู„ ุงู„ู…ุนู„ูˆู…ุงุช
# ู‚ู„ู„ู†ุง ุงู„ู†ุณุจุฉ ุนุดุงู† ู…ุง ูŠู‚ูˆู„ "ู„ุง ุชุชูˆูุฑ ู…ุนู„ูˆู…ุงุช"
MIN_SIMILARITY_SCORE = 0.15
MAX_LENGTH = 1000
# ู…ูุชุงุญ ุฌูˆุฌู„
GOOGLE_API_KEY = os.getenv("GOOGLE_API_KEY")
# ู…ูˆุฏูŠู„ ุงู„ูู„ุงุด ุงู„ุณุฑูŠุน ูˆุงู„ู…ุฌุงู†ูŠ
GENAI_MODEL = "gemini-1.5-flash"