broadfield-dev commited on
Commit
ddb86ab
·
verified ·
1 Parent(s): e068aac

Update model_logic.py

Browse files
Files changed (1) hide show
  1. model_logic.py +3 -3
model_logic.py CHANGED
@@ -37,15 +37,15 @@ API_URLS = {
37
  "GOOGLE": 'https://generativelanguage.googleapis.com/v1beta/models/',
38
  }
39
 
40
- MODELS_BY_PROVIDER = json.load(open("./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
  """
 
37
  "GOOGLE": 'https://generativelanguage.googleapis.com/v1beta/models/',
38
  }
39
 
40
+ # MODELS_BY_PROVIDER = json.load(open("./models.json")) ## commented for demo
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
  """