from smolagents import tool

def clean_answer_with_prompt(agent_output: str) -> str:
    """
    Extracts and cleans the final answer from the agent output.
    For GAIA, ensure no 'FINAL ANSWER:' prefix is returned — just the answer.
    """
    if "FINAL ANSWER:" in agent_output:
        return agent_output.split("FINAL ANSWER:")[-1].strip()
    return agent_output.strip()

def build_prompt(question: str, context: str) -> str:
    """
    Combine the system instruction, context, and question to build the LLM prompt.
    """
    system_instruction = (
        "You are an intelligent assistant helping answer complex real-world questions. "
        "Use the provided context to reason and provide a concise factual answer. "
        "Only answer what is asked. Do not include 'FINAL ANSWER:' or extra explanation.\n\n"
    )
    return f"{system_instruction}Context:\n{context}\n\nQuestion: {question}\nAnswer:"

@tool
def greeting_tool(name: str) -> str:
    """
    Generates a custom greeting for the guest.

    Args:
        name: Name of the guest

    Returns:
        A friendly greeting message.
    """
    return f"Welcome to the gala, {name}! We're honored to have you with us."