Update codex/code_generator.py

codex/code_generator.py

@@ -1,14 +1,19 @@
 import os
 import requests
 
-HF_API = "https://api-inference.huggingface.co/models/Qwen/Qwen2.5-Coder-7B-Instruct"
+HF_API = "https://api-inference.huggingface.co/v1/chat/completions"
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
+MODEL = "meta-llama/Llama-3.1-8B-Instruct"  # Stable + free-tier friendly
+
 def generate_code(task):
     if not HF_TOKEN:
         raise RuntimeError("HF_TOKEN not set")
 
-    headers = {"Authorization": f"Bearer {HF_TOKEN}"}
+    headers = {
+        "Authorization": f"Bearer {HF_TOKEN}",
+        "Content-Type": "application/json"
+    }
 
     prompt = f"""
 You are a senior Python engineer.
@@ -28,20 +33,21 @@ Rules:
 """
 
     payload = {
-        "inputs": prompt,
-        "parameters": {
-            "max_new_tokens": 350,
-            "temperature": 0.2,
-            "top_p": 0.9
-        }
+        "model": MODEL,
+        "messages": [
+            {"role": "system", "content": "You generate safe Python code only."},
+            {"role": "user", "content": prompt}
+        ],
+        "max_tokens": 300,
+        "temperature": 0.2
     }
 
     r = requests.post(HF_API, headers=headers, json=payload, timeout=60)
     r.raise_for_status()
 
-    text = r.json()[0]["generated_text"]
+    data = r.json()
+    text = data["choices"][0]["message"]["content"]
 
-    # Strip markdown if present
     if "```" in text:
         text = text.split("```")[-1]