| from .state import AgentState, tools_config | |
| from .tools import TOOL_REGISTRY | |
| def execute_tool(state: AgentState): | |
| tool_name = state["selected_tool"] | |
| args = state["tool_args"] | |
| # Make sure the tool exists | |
| if tool_name not in TOOL_REGISTRY: | |
| return { | |
| "tool_result": { | |
| "status": "error", | |
| "message": "Unknown tool" | |
| } | |
| } | |
| tool_config = next( | |
| t for t in tools_config | |
| if t["name"] == tool_name | |
| ) | |
| allowed = { | |
| p["name"] for p in tool_config["parameters"] | |
| } | |
| if not set(args).issubset(allowed): | |
| return { | |
| "tool_result": { | |
| "status": "error", | |
| "message": "Unexpected tool arguments" | |
| } | |
| } | |
| # Execute the selected function | |
| function = TOOL_REGISTRY[tool_name] | |
| result = function(**args) | |
| return { | |
| "tool_result": result | |
| } | |