luisejdm commited on
Commit
e309431
·
1 Parent(s): c464284

Add config file

Browse files
Files changed (1) hide show
  1. config.py +19 -0
config.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from __future__ import annotations
2
+
3
+ import os
4
+ from dataclasses import dataclass
5
+
6
+
7
+ @dataclass(frozen=True)
8
+ class AppConfig:
9
+ model_name: str = "meta/llama-3.3-70b-instruct"
10
+ nvidia_env_key: str = "NVIDIA_API_KEY"
11
+ base_url: str = "https://integrate.api.nvidia.com/v1"
12
+
13
+
14
+ def load_environment() -> str | None:
15
+ api_key = os.getenv(AppConfig.nvidia_env_key)
16
+ print(f"NVIDIA API Key loaded: {'✓' if api_key else '✗'}")
17
+ if not api_key:
18
+ print(f"⚠️ {AppConfig.nvidia_env_key} not set in environment")
19
+ return api_key