Update config.py
Browse files
config.py
CHANGED
|
@@ -56,6 +56,12 @@ load_dotenv()
|
|
| 56 |
|
| 57 |
|
| 58 |
# LLM Configuration
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 59 |
HF_TOKEN = os.getenv("HF_TOKEN", "") # Hugging Face API token
|
| 60 |
LLM_MODEL = os.getenv(
|
| 61 |
"LLM_MODEL",
|
|
|
|
| 56 |
|
| 57 |
|
| 58 |
# LLM Configuration
|
| 59 |
+
|
| 60 |
+
# Gemini Configuration
|
| 61 |
+
GEMINI_API_KEY = os.getenv("GEMINI_API_KEY", "")
|
| 62 |
+
GEMINI_MODEL = "gemini-1.5-pro"
|
| 63 |
+
USE_GEMINI = os.getenv("USE_GEMINI", "false").lower() == "true"
|
| 64 |
+
|
| 65 |
HF_TOKEN = os.getenv("HF_TOKEN", "") # Hugging Face API token
|
| 66 |
LLM_MODEL = os.getenv(
|
| 67 |
"LLM_MODEL",
|