Update prompt.py

prompt.py

@@ -2,19 +2,17 @@ SYSTEM_PROMPT = """You are a precise coding assistant.
 Write clean, correct, minimal code.
 Prefer readable solutions over clever ones.
 When assumptions are required, make them explicit in comments.
+Return code first. Add a short explanation only if useful.
 """
 
-USER_PROMPT_TEMPLATE = """{system_prompt}
 
-{context}
-Prompt:
-{user_prompt}
-"""
+def build_messages(user_prompt: str, system_prompt: str = SYSTEM_PROMPT, context: str = "") -> list[dict]:
+    user_content = user_prompt.strip()
 
+    if context.strip():
+        user_content = f"{context.strip()}\n\nPrompt:\n{user_content}"
 
-def build_full_prompt(user_prompt: str, system_prompt: str = SYSTEM_PROMPT, context: str = "") -> str:
-    return USER_PROMPT_TEMPLATE.format(
-        system_prompt=system_prompt.strip(),
-        context=context.strip(),
-        user_prompt=user_prompt.strip(),
-    ).strip()
+    return [
+        {"role": "system", "content": system_prompt.strip()},
+        {"role": "user", "content": user_content},
+    ]