datbkpro commited on
Commit
6fd0088
·
verified ·
1 Parent(s): aeca306

Update config/settings.py

Browse files
Files changed (1) hide show
  1. 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
- GROQ_API_KEY = os.getenv("GROQ_API_KEY")
 
 
 
 
 
 
 
 
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 = 768 # For Vietnamese model
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