from langchain_core.messages import AIMessage
from .state import AgentState


def generate_response(state: AgentState):
    result = state["tool_result"]

    if result is None:
        message = "No tool result is available."
    else:
        message = result["message"]

    return {
        "messages": [AIMessage(content=message)]
    }