Update codex/code_generator.py

codex/code_generator.py

@@ -1,7 +1,7 @@
 import os
 import requests
 
-HF_API = "https://api-inference.huggingface.co/models/bigcode/starcoder"
+HF_API = "https://api-inference.huggingface.co/models/Qwen/Qwen2.5-Coder-7B-Instruct"
 HF_TOKEN = os.environ.get("HF_TOKEN")
 
 def generate_code(task):
@@ -11,6 +11,8 @@ def generate_code(task):
     headers = {"Authorization": f"Bearer {HF_TOKEN}"}
 
     prompt = f"""
+You are a senior Python engineer.
+
 Write safe Python code to accomplish this task:
 
 {task}
@@ -21,14 +23,26 @@ Rules:
 - No network calls
 - Must define a function
 - Must assign final output to variable named `result`
+- Must not use input()
 - Return only Python code
 """
 
-    payload = {"inputs": prompt, "parameters": {"max_new_tokens": 300}}
+    payload = {
+        "inputs": prompt,
+        "parameters": {
+            "max_new_tokens": 350,
+            "temperature": 0.2,
+            "top_p": 0.9
+        }
+    }
+
     r = requests.post(HF_API, headers=headers, json=payload, timeout=60)
     r.raise_for_status()
 
     text = r.json()[0]["generated_text"]
-    code = text.split("```")[-1].strip()
 
-    return code
+    # Strip markdown if present
+    if "```" in text:
+        text = text.split("```")[-1]
+
+    return text.strip()