Update config/settings.py
Browse files- config/settings.py +10 -11
config/settings.py
CHANGED
|
@@ -4,7 +4,15 @@ from dotenv import load_dotenv
|
|
| 4 |
load_dotenv()
|
| 5 |
|
| 6 |
class Settings:
|
| 7 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
|
| 9 |
# Multilingual Model Settings
|
| 10 |
VIETNAMESE_EMBEDDING_MODEL = 'keepitreal/vietnamese-sbert'
|
|
@@ -13,13 +21,6 @@ class Settings:
|
|
| 13 |
MULTILINGUAL_EMBEDDING_MODEL = 'sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2'
|
| 14 |
MULTILINGUAL_LLM_MODEL = "llama-3.1-8b-instant"
|
| 15 |
|
| 16 |
-
# Fallback models in case primary models fail
|
| 17 |
-
FALLBACK_MULTILINGUAL_EMBEDDING_MODEL = 'sentence-transformers/all-MiniLM-L6-v2'
|
| 18 |
-
|
| 19 |
-
# Default models (fallback)
|
| 20 |
-
DEFAULT_EMBEDDING_MODEL = 'dangvantuan/vietnamese-embedding'
|
| 21 |
-
DEFAULT_LLM_MODEL = "Vietnamese_LLaMA2_13B_8K_SFT_General_Domain_Knowledge"
|
| 22 |
-
|
| 23 |
WHISPER_MODEL = "whisper-large-v3-turbo"
|
| 24 |
|
| 25 |
# TTS Settings
|
|
@@ -30,9 +31,7 @@ class Settings:
|
|
| 30 |
}
|
| 31 |
|
| 32 |
# RAG Settings
|
| 33 |
-
EMBEDDING_DIMENSION =
|
| 34 |
-
MULTILINGUAL_EMBEDDING_DIMENSION = 4096 # For Nemotron model
|
| 35 |
-
|
| 36 |
TOP_K_RESULTS = 3
|
| 37 |
|
| 38 |
# SpeechBrain VAD Settings
|
|
|
|
| 4 |
load_dotenv()
|
| 5 |
|
| 6 |
class Settings:
|
| 7 |
+
# Ưu tiên lấy từ environment variables (cho Hugging Face)
|
| 8 |
+
GROQ_API_KEY = os.getenv("GROQ_API_KEY") or os.getenv("X_GROQ_API_KEY")
|
| 9 |
+
|
| 10 |
+
# Kiểm tra API Key
|
| 11 |
+
if not GROQ_API_KEY:
|
| 12 |
+
print("❌ CẢNH BÁO: Không tìm thấy GROQ_API_KEY")
|
| 13 |
+
print("💡 Trên Hugging Face Spaces, hãy thêm biến môi trường:")
|
| 14 |
+
print(" - Tên: GROQ_API_KEY")
|
| 15 |
+
print(" - Value: gsk_xxxxxxxxxxxxxxxxxxxxxxxxxxxxxxxx")
|
| 16 |
|
| 17 |
# Multilingual Model Settings
|
| 18 |
VIETNAMESE_EMBEDDING_MODEL = 'keepitreal/vietnamese-sbert'
|
|
|
|
| 21 |
MULTILINGUAL_EMBEDDING_MODEL = 'sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2'
|
| 22 |
MULTILINGUAL_LLM_MODEL = "llama-3.1-8b-instant"
|
| 23 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 24 |
WHISPER_MODEL = "whisper-large-v3-turbo"
|
| 25 |
|
| 26 |
# TTS Settings
|
|
|
|
| 31 |
}
|
| 32 |
|
| 33 |
# RAG Settings
|
| 34 |
+
EMBEDDING_DIMENSION = 384
|
|
|
|
|
|
|
| 35 |
TOP_K_RESULTS = 3
|
| 36 |
|
| 37 |
# SpeechBrain VAD Settings
|