Nelly-43 commited on
Commit
5a07c19
·
1 Parent(s): 770e251

Update api-key access for llama model

Browse files
Files changed (2) hide show
  1. setup.py +1 -1
  2. src/config_loader.py +1 -0
setup.py CHANGED
@@ -12,7 +12,7 @@ _ENV_KEY_MAP = {
12
  "openai": "OPENAI_API_KEY",
13
  "anthropic": "ANTHROPIC_API_KEY",
14
  "gemini": "GEMINI_API_KEY",
15
- "meta-llama": "HF_API_KEY",
16
  }
17
 
18
  _PROVIDER_CHOICES = {"1": "openai", "2": "anthropic", "3": "gemini", "4": "meta-llama"}
 
12
  "openai": "OPENAI_API_KEY",
13
  "anthropic": "ANTHROPIC_API_KEY",
14
  "gemini": "GEMINI_API_KEY",
15
+ "meta-llama": "META-LLAMA_API_KEY",
16
  }
17
 
18
  _PROVIDER_CHOICES = {"1": "openai", "2": "anthropic", "3": "gemini", "4": "meta-llama"}
src/config_loader.py CHANGED
@@ -10,6 +10,7 @@ _ENV_KEY_MAP = {
10
  "openai": "OPENAI_API_KEY",
11
  "anthropic": "ANTHROPIC_API_KEY",
12
  "gemini": "GEMINI_API_KEY",
 
13
  }
14
 
15
 
 
10
  "openai": "OPENAI_API_KEY",
11
  "anthropic": "ANTHROPIC_API_KEY",
12
  "gemini": "GEMINI_API_KEY",
13
+ "meta-llama": "META-LLAMA_API_KEY",
14
  }
15
 
16