add the code

prompt_builder.py

@@ -0,0 +1,272 @@
+from typing import List, Optional
+
+from schemas import CodeTaskType, RetrievedEvidence
+
+
+def format_evidence(evidence_list: Optional[List[RetrievedEvidence]]) -> str:
+    if not evidence_list:
+        return "No external technical evidence provided."
+
+    lines = []
+    for index, item in enumerate(evidence_list, start=1):
+        lines.append(f"Source {index}")
+        lines.append(f"Type: {item.source_type}")
+        lines.append(f"Title: {item.title}")
+        lines.append(f"Snippet: {item.snippet}")
+        if item.url:
+            lines.append(f"URL: {item.url}")
+        if item.score is not None:
+            lines.append(f"Score: {item.score}")
+        lines.append("")
+
+    return "\n".join(lines).strip()
+
+
+def build_shared_context(
+    message: str,
+    language: Optional[str] = None,
+    framework: Optional[str] = None,
+    code: Optional[str] = None,
+    error_message: Optional[str] = None,
+    previous_context: Optional[str] = None,
+    evidence_list: Optional[List[RetrievedEvidence]] = None,
+) -> str:
+    parts = [
+        f"User Request:\n{message.strip()}",
+        f"Language:\n{language.strip() if language else 'Not specified'}",
+        f"Framework:\n{framework.strip() if framework else 'Not specified'}",
+        f"Previous Context:\n{previous_context.strip() if previous_context else 'None'}",
+        f"Error Message:\n{error_message.strip() if error_message else 'None'}",
+        f"Code:\n{code.strip() if code else 'No code provided'}",
+        f"Technical Evidence:\n{format_evidence(evidence_list)}",
+    ]
+    return "\n\n".join(parts)
+
+
+def build_generate_prompt(
+    message: str,
+    language: Optional[str] = None,
+    framework: Optional[str] = None,
+    previous_context: Optional[str] = None,
+) -> str:
+    shared = build_shared_context(
+        message=message,
+        language=language,
+        framework=framework,
+        previous_context=previous_context,
+    )
+
+    return (
+        "You are a senior software engineer.\n"
+        "Your task is to generate code based on the user's request.\n"
+        "Follow these rules:\n"
+        "1. Generate clean, correct, production-style code.\n"
+        "2. Match the requested language and framework if provided.\n"
+        "3. If the language or framework is missing, infer a reasonable default from the request.\n"
+        "4. Keep the explanation concise.\n"
+        "5. Return code that is directly usable.\n\n"
+        f"{shared}\n\n"
+        "Output Format:\n"
+        "Explanation:\n"
+        "<short explanation>\n\n"
+        "Code:\n"
+        "<generated code>"
+    )
+
+
+def build_fix_prompt(
+    message: str,
+    code: Optional[str] = None,
+    error_message: Optional[str] = None,
+    language: Optional[str] = None,
+    framework: Optional[str] = None,
+    previous_context: Optional[str] = None,
+    evidence_list: Optional[List[RetrievedEvidence]] = None,
+) -> str:
+    shared = build_shared_context(
+        message=message,
+        language=language,
+        framework=framework,
+        code=code,
+        error_message=error_message,
+        previous_context=previous_context,
+        evidence_list=evidence_list,
+    )
+
+    return (
+        "You are a senior software engineer and debugging expert.\n"
+        "Your task is to fix the user's code.\n"
+        "Follow these rules:\n"
+        "1. Identify the most likely root cause.\n"
+        "2. Preserve the user's intended logic where possible.\n"
+        "3. Use the provided technical evidence if relevant.\n"
+        "4. If the error details are incomplete, make the safest reasonable fix and mention assumptions.\n"
+        "5. Return the corrected code and a concise explanation.\n\n"
+        f"{shared}\n\n"
+        "Output Format:\n"
+        "Root Cause:\n"
+        "<short root cause>\n\n"
+        "Explanation:\n"
+        "<short explanation>\n\n"
+        "Code:\n"
+        "<fixed code>"
+    )
+
+
+def build_explain_prompt(
+    message: str,
+    code: Optional[str] = None,
+    language: Optional[str] = None,
+    framework: Optional[str] = None,
+    previous_context: Optional[str] = None,
+) -> str:
+    shared = build_shared_context(
+        message=message,
+        language=language,
+        framework=framework,
+        code=code,
+        previous_context=previous_context,
+    )
+
+    return (
+        "You are a senior software engineer.\n"
+        "Your task is to explain the user's code or technical question clearly.\n"
+        "Follow these rules:\n"
+        "1. Explain in a simple and structured way.\n"
+        "2. Focus on the most important logic.\n"
+        "3. If code is provided, explain what it does and why.\n"
+        "4. Keep the explanation practical and concise.\n\n"
+        f"{shared}\n\n"
+        "Output Format:\n"
+        "Explanation:\n"
+        "<clear explanation>"
+    )
+
+
+def build_refactor_prompt(
+    message: str,
+    code: Optional[str] = None,
+    language: Optional[str] = None,
+    framework: Optional[str] = None,
+    previous_context: Optional[str] = None,
+) -> str:
+    shared = build_shared_context(
+        message=message,
+        language=language,
+        framework=framework,
+        code=code,
+        previous_context=previous_context,
+    )
+
+    return (
+        "You are a senior software engineer.\n"
+        "Your task is to refactor the user's code.\n"
+        "Follow these rules:\n"
+        "1. Preserve the original behavior unless the user asked for functional changes.\n"
+        "2. Improve readability, maintainability, and structure.\n"
+        "3. Keep the result practical and clean.\n"
+        "4. Return the improved code and a concise explanation of changes.\n\n"
+        f"{shared}\n\n"
+        "Output Format:\n"
+        "Explanation:\n"
+        "<short explanation>\n\n"
+        "Code:\n"
+        "<refactored code>"
+    )
+
+
+def build_review_prompt(
+    message: str,
+    code: Optional[str] = None,
+    language: Optional[str] = None,
+    framework: Optional[str] = None,
+    previous_context: Optional[str] = None,
+) -> str:
+    shared = build_shared_context(
+        message=message,
+        language=language,
+        framework=framework,
+        code=code,
+        previous_context=previous_context,
+    )
+
+    return (
+        "You are a senior software engineer performing a code review.\n"
+        "Your task is to review the user's code and identify issues or improvements.\n"
+        "Follow these rules:\n"
+        "1. Focus on correctness, readability, maintainability, and possible bugs.\n"
+        "2. Be specific and practical.\n"
+        "3. If improvements are needed, suggest them clearly.\n"
+        "4. Do not rewrite the whole code unless necessary.\n\n"
+        f"{shared}\n\n"
+        "Output Format:\n"
+        "Review:\n"
+        "<review points>\n\n"
+        "Suggestions:\n"
+        "<practical suggestions>"
+    )
+
+
+def build_prompt(
+    task_type: CodeTaskType,
+    message: str,
+    code: Optional[str] = None,
+    error_message: Optional[str] = None,
+    language: Optional[str] = None,
+    framework: Optional[str] = None,
+    previous_context: Optional[str] = None,
+    evidence_list: Optional[List[RetrievedEvidence]] = None,
+) -> str:
+    if task_type == CodeTaskType.GENERATE:
+        return build_generate_prompt(
+            message=message,
+            language=language,
+            framework=framework,
+            previous_context=previous_context,
+        )
+
+    if task_type == CodeTaskType.FIX:
+        return build_fix_prompt(
+            message=message,
+            code=code,
+            error_message=error_message,
+            language=language,
+            framework=framework,
+            previous_context=previous_context,
+            evidence_list=evidence_list,
+        )
+
+    if task_type == CodeTaskType.EXPLAIN:
+        return build_explain_prompt(
+            message=message,
+            code=code,
+            language=language,
+            framework=framework,
+            previous_context=previous_context,
+        )
+
+    if task_type == CodeTaskType.REFACTOR:
+        return build_refactor_prompt(
+            message=message,
+            code=code,
+            language=language,
+            framework=framework,
+            previous_context=previous_context,
+        )
+
+    if task_type == CodeTaskType.REVIEW:
+        return build_review_prompt(
+            message=message,
+            code=code,
+            language=language,
+            framework=framework,
+            previous_context=previous_context,
+        )
+
+    return build_explain_prompt(
+        message=message,
+        code=code,
+        language=language,
+        framework=framework,
+        previous_context=previous_context,
+    )