broadfield-dev commited on
Commit
d9e7144
·
verified ·
1 Parent(s): eb12278

Update model_logic.py

Browse files
Files changed (1) hide show
  1. model_logic.py +3 -2
model_logic.py CHANGED
@@ -37,14 +37,15 @@ API_URLS = {
37
  "GOOGLE": 'https://generativelanguage.googleapis.com/v1beta/models/',
38
  }
39
 
40
- MODELS_BY_PROVIDER = {
 
41
  "groq": {
42
  "default": "llama3-8b-8192",
43
  "models": {
44
  "Llama 3 8B (Groq)": "llama3-8b-8192",
45
  }
46
  }
47
- }
48
 
49
  def _get_api_key(provider: str, ui_api_key_override: str = None) -> str | None:
50
  """
 
37
  "GOOGLE": 'https://generativelanguage.googleapis.com/v1beta/models/',
38
  }
39
 
40
+ MODELS_BY_PROVIDER = json.load("./models.json")
41
+ '''MODELS_BY_PROVIDER = {
42
  "groq": {
43
  "default": "llama3-8b-8192",
44
  "models": {
45
  "Llama 3 8B (Groq)": "llama3-8b-8192",
46
  }
47
  }
48
+ }'''
49
 
50
  def _get_api_key(provider: str, ui_api_key_override: str = None) -> str | None:
51
  """