Tool not found in ToolManager when using custom tool

[niteshver]

Discussion: Custom Tool Not Found in ToolManager (Mesa-LLM)

📌 Summary

I am building a conflict-driven migration model using Mesa-LLM.
Each Citizen agent computes migration probability mathematically and uses the LLM only to explain its decision.

However, when running the model, I encounter a KeyError indicating that a custom tool is not found inside the agent’s ToolManager.

---

🧠 Model Architecture

Each Citizen agent:

• Computes migration probability using a logistic function.
• Uses an LLM only to explain its decision.
• Has a custom tool called move_to_safe_place.
• Uses a custom CITIZEN_TOOL_MANAGER.

---

🛠 Custom Tool Definition

@tool(tool_manager=CITIZEN_TOOL_MANAGER)
def move_to_safe_place(agent: "LLMAgent", state: str) -> str:
    """
    Migrate citizen to safe zone.

    Args:
        agent: The Citizen agent performing migration.

    Returns:
        A message describing the migration result.
    """

---

🚨 Where the Error Occurs

Inside the agent:

plan = self.reasoning.plan(
    prompt=prompt,
    obs=observation,
    selected_tools=["move_to_safe_place"],
)

---

❌ Error Message

KeyError: 'move_to_safe_place'

Traceback shows the failure occurs inside:

self.agent.tool_manager.get_all_tools_schema(selected_tools)

Specifically at:

self.tools[tool]

The tool "move_to_safe_place" is not found inside the agent’s ToolManager.

---

🔍 Observed Behavior

• The tool is decorated with @tool(tool_manager=CITIZEN_TOOL_MANAGER).
• The agent calls reasoning.plan() with selected_tools=["move_to_safe_place"].
• During execution, self.agent.tool_manager.tools does not contain "move_to_safe_place".

---

🤔 Suspected Causes

One of the following may be happening:

1. The agent instance is not actually using CITIZEN_TOOL_MANAGER.
2. The tool file is not imported, so registration never executes.
3. The tool name in selected_tools does not match the registered key.
4. The custom ToolManager is not properly attached to the agent.

Any guidance on properly wiring a custom ToolManager in Mesa-LLM would be greatly appreciated.