Update main.py

main.py

@@ -91,6 +91,7 @@ async def core():
     
 
 def getPrompt(prompt, modelID, attempts=1):
+    response = {}
     if modelID != magic_prompt_model:
         chat = [
             {"role": "user", "content": prompt_base},
@@ -98,7 +99,7 @@ def getPrompt(prompt, modelID, attempts=1):
             {"role": "user", "content": prompt},
             ]
         try:
-            response = client.chat.completions.create(messages=chat, temperature=1, max_tokens=2048, top_p=1, stream=False, stop=None, model=modelID) 
+            response = groqClient.chat.completions.create(messages=chat, temperature=1, max_tokens=2048, top_p=1, stream=False, stop=None, model=modelID) 
         except Exception as e:
             print(f"An error occurred: {e}")
             if attempts < 3: