LordXido commited on
Commit
15e2b3b
·
verified ·
1 Parent(s): 0443cfd

Update codex/code_generator.py

Browse files
Files changed (1) hide show
  1. codex/code_generator.py +5 -3
codex/code_generator.py CHANGED
@@ -35,15 +35,17 @@ Rules:
35
  payload = {
36
  "model": MODEL,
37
  "messages": [
38
- {"role": "system", "content": "You generate safe Python code only."},
39
  {"role": "user", "content": prompt}
40
  ],
41
  "max_tokens": 300,
42
- "temperature": 0.2
 
43
  }
44
 
45
  r = requests.post(GROQ_API, headers=headers, json=payload, timeout=60)
46
- r.raise_for_status()
 
 
47
 
48
  data = r.json()
49
  text = data["choices"][0]["message"]["content"]
 
35
  payload = {
36
  "model": MODEL,
37
  "messages": [
 
38
  {"role": "user", "content": prompt}
39
  ],
40
  "max_tokens": 300,
41
+ "temperature": 0.2,
42
+ "top_p": 1.0
43
  }
44
 
45
  r = requests.post(GROQ_API, headers=headers, json=payload, timeout=60)
46
+
47
+ if r.status_code != 200:
48
+ raise RuntimeError(f"Groq error {r.status_code}: {r.text}")
49
 
50
  data = r.json()
51
  text = data["choices"][0]["message"]["content"]