From d6785daf339852a968df289da08d8d0c8df62c5c Mon Sep 17 00:00:00 2001
From: daxcay <daxtoncaylor@gmail.com>
Date: Wed, 16 Oct 2024 23:11:51 +0530
Subject: [PATCH] Patch for skipping for normal load

---
 classes/DataSet_GroqChat.py      | 24 ++++++++++++------------
 classes/DataSet_GroqChatImage.py | 23 ++++++++++++-----------
 2 files changed, 24 insertions(+), 23 deletions(-)

diff --git a/classes/DataSet_GroqChat.py b/classes/DataSet_GroqChat.py
index 32ed644..d85a867 100644
--- a/classes/DataSet_GroqChat.py
+++ b/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):
diff --git a/classes/DataSet_GroqChatImage.py b/classes/DataSet_GroqChatImage.py
index f1dcacc..74eb788 100644
--- a/classes/DataSet_GroqChatImage.py
+++ b/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)