HugMeBytes commited on
Commit
7723293
·
verified ·
1 Parent(s): 3f58b5c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -21,12 +21,14 @@ def call_groq_api(prompt):
21
  return "Error: GROQ_API_KEY environment variable not set."
22
 
23
  headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
24
- data = {"model": "llama3-70b-8192", "messages": [{"role": "user", "content": prompt}]}
25
 
26
  try:
27
  response = requests.post("https://api.groq.com/openai/v1/chat/completions", json=data, headers=headers)
28
- response.raise_for_status()
 
29
  return response.json()["choices"][0]["message"]["content"]
 
30
  except requests.exceptions.RequestException as e:
31
  return f"API Error: {str(e)}"
32
  except (KeyError, IndexError) as e:
 
21
  return "Error: GROQ_API_KEY environment variable not set."
22
 
23
  headers = {"Authorization": f"Bearer {api_key}", "Content-Type": "application/json"}
24
+ data = {"model": "mixtral-8x7b-32768", "messages": [{"role": "user", "content": prompt}]}
25
 
26
  try:
27
  response = requests.post("https://api.groq.com/openai/v1/chat/completions", json=data, headers=headers)
28
+ if response.status_code != 200:
29
+ return f"API Error {response.status_code}: {response.text}"
30
  return response.json()["choices"][0]["message"]["content"]
31
+
32
  except requests.exceptions.RequestException as e:
33
  return f"API Error: {str(e)}"
34
  except (KeyError, IndexError) as e: