prompt_builder.py

"""
Flare – Prompt Builder  (v4-fix · detection_prompt + examples)
==============================================================
"""

from typing import List, Dict
from utils import log


# ─────────────────────────────────────────────────────────────────────────────
# INTENT PROMPT
# ─────────────────────────────────────────────────────────────────────────────
def build_intent_prompt(general_prompt: str,
                        conversation: List[Dict[str, str]],
                        user_input: str,
                        intents: List) -> str:
    # === INTENT INDEX ===
    lines = ["### INTENT INDEX ###"]
    for it in intents:
        # IntentConfig object attribute access
        det = it.detection_prompt.strip() if it.detection_prompt else ""
        det_part = f'  • detection_prompt → "{det}"' if det else ""
        exs = " | ".join(it.examples) if it.examples else ""
        ex_part = f"  • examples → {exs}" if exs else ""

        newline_between = "\n" if det_part and ex_part else ""
        lines.append(f"{it.name}:{det_part}{newline_between}{ex_part}")

    intent_index = "\n".join(lines)

    # === HISTORY ===
    history_block = "\n".join(
        f"{m['role'].upper()}: {m['content']}" for m in conversation[-10:]
    )

    prompt = (
        f"{general_prompt}\n\n"
        f"{intent_index}\n\n"
        f"Conversation so far:\n{history_block}\n\n"
        f"USER: {user_input.strip()}"
    )
    log("✅ Intent prompt built (index with detection_prompt)")
    return prompt


# ─────────────────────────────────────────────────────────────────────────────
# PARAMETER PROMPT (değişmedi)
# ─────────────────────────────────────────────────────────────────────────────
_FMT = (
    "Return exactly ONE line in the format:\n"
    "#PARAMETERS:{"
    "\"extracted\":[{\"name\":\"<param>\",\"value\":\"<val>\"},...],"
    "\"missing\":[\"<param>\",...]}"
)

def build_parameter_prompt(intent_cfg,
                           missing_params: List[str],
                           user_input: str,
                           conversation: List[Dict[str, str]]) -> str:
    parts: List[str] = [
        "You will extract ONLY the parameters listed below.",
        "If a parameter cannot be found OR fails validation, keep it in the "
        "\"missing\" list. Never guess values."
    ]
    for p in intent_cfg.parameters:
        if p.name in missing_params:
            parts.append(f"* {p.name}: {p.extraction_prompt}")
    parts.append(_FMT)

    history_block = "\n".join(
        f"{m['role'].upper()}: {m['content']}" for m in conversation[-10:]
    )

    prompt = (
        "\n".join(parts) +
        "\n\nConversation so far:\n" + history_block +
        "\n\nUSER: " + user_input.strip()
    )
    log("✅ Parameter prompt built")
    return prompt