zeetay commited on
Commit
cb67f1d
·
1 Parent(s): bd01a67

updated groq model checker for better error handling

Browse files
Files changed (1) hide show
  1. check_groq_models.py +23 -10
check_groq_models.py CHANGED
@@ -1,21 +1,34 @@
1
  # backend/check_groq_models.py
 
 
 
 
 
2
  import os
3
  import requests
4
  from dotenv import load_dotenv
5
  from pathlib import Path
 
 
6
  load_dotenv(dotenv_path=Path(__file__).resolve().parent / ".env")
7
- API_URL = "https://api.groq.com/openai/v1/models"
8
 
 
9
  api_key = os.getenv("GROQ_API_KEY")
 
10
  if not api_key:
11
- raise SystemExit("Set GROQ_API_KEY in backend/.env")
12
 
13
- headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
 
 
 
14
 
15
- resp = requests.get(API_URL, headers=headers, timeout=15)
16
- resp.raise_for_status()
17
- data = resp.json().get("data", [])
18
- print("Available Groq models:")
19
- for m in data:
20
- # model id often in 'id' or 'model'
21
- print("-", m.get("id") or m.get("model") or m.get("name") or m)
 
 
 
1
  # backend/check_groq_models.py
2
+ """
3
+ Utility script to check available Groq LLM models.
4
+ Requires GROQ_API_KEY in backend/.env.
5
+ """
6
+
7
  import os
8
  import requests
9
  from dotenv import load_dotenv
10
  from pathlib import Path
11
+
12
+ # Load environment variables
13
  load_dotenv(dotenv_path=Path(__file__).resolve().parent / ".env")
 
14
 
15
+ API_URL = "https://api.groq.com/openai/v1/models"
16
  api_key = os.getenv("GROQ_API_KEY")
17
+
18
  if not api_key:
19
+ raise SystemExit(" Error: GROQ_API_KEY not set in backend/.env")
20
 
21
+ headers = {
22
+ "Authorization": f"Bearer {api_key}",
23
+ "Content-Type": "application/json",
24
+ }
25
 
26
+ try:
27
+ resp = requests.get(API_URL, headers=headers, timeout=15)
28
+ resp.raise_for_status()
29
+ data = resp.json().get("data", [])
30
+ print("\nAvailable Groq models:\n")
31
+ for m in data:
32
+ print("-", m.get("id") or m.get("model") or m.get("name") or m)
33
+ except requests.exceptions.RequestException as e:
34
+ print(f"API request failed: {e}")