| 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)] | |
| } | |