Update codex/code_generator.py
Browse files- 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 |
-
|
|
|
|
|
|
|
| 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"]
|