Patch for skipping for normal load

2024-10-16 23:11:51 +05:30
parent 7cdcac9c9d
commit d6785daf33
2 changed files with 24 additions and 23 deletions
@@ -8,17 +8,18 @@ import numpy as np


 def getGroqModels(key):
-
-    url = "https://api.groq.com/openai/v1/models"
-    headers = {
-        "Authorization": f"Bearer {key}",
-        "Content-Type": "application/json"
-    }
-    response = requests.get(url, headers=headers)
-    response = response.json()
-
-    return [item['id'] for item in response['data'] if 'id' in item]
-
+    try:
+        url = "https://api.groq.com/openai/v1/models"
+        headers = {
+            "Authorization": f"Bearer {key}",
+            "Content-Type": "application/json"
+        }
+        response = requests.get(url, headers=headers)
+        response.raise_for_status()
+        response = response.json()
+        return [item['id'] for item in response['data'] if 'id' in item]
+    except (requests.exceptions.RequestException, KeyError, ValueError):
+        return ['Key Invalid']

 api_key = os.environ.get("GROQ_API_KEY")
 api_models = getGroqModels(api_key)