added llm config variable
Browse files- app/llm.py +1 -1
app/llm.py
CHANGED
|
@@ -23,7 +23,7 @@ class LLMProcessor:
|
|
| 23 |
config_path = os.path.join(base_dir, "config", "config.yml")
|
| 24 |
with open(config_path, "r") as file:
|
| 25 |
self.config_data = yaml.safe_load(file)
|
| 26 |
-
|
| 27 |
# --- Gemini Initialization ---
|
| 28 |
gemini_config = llm_config.get("gemini", {})
|
| 29 |
self.gemini_api_key = os.getenv("GEMINI_API_KEY") # From Hugging Face secrets
|
|
|
|
| 23 |
config_path = os.path.join(base_dir, "config", "config.yml")
|
| 24 |
with open(config_path, "r") as file:
|
| 25 |
self.config_data = yaml.safe_load(file)
|
| 26 |
+
llm = self.config_data.get("llm", {})
|
| 27 |
# --- Gemini Initialization ---
|
| 28 |
gemini_config = llm_config.get("gemini", {})
|
| 29 |
self.gemini_api_key = os.getenv("GEMINI_API_KEY") # From Hugging Face secrets
|