Patch for skipping for normal load

classes/DataSet_GroqChat.py

@@ -3,22 +3,22 @@ import requests
 import os
 
 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()  # Raises HTTPError for bad responses
+        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)
 
-
 class DataSet_GroqChat:
 
     def __init__(self):

classes/DataSet_GroqChatImage.py

@@ -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)