Tool Calling is Linearly Readable and Steerable in Language Models
On the map: LLM agents and tool use → Tool learning and function calling
Abstract (arXiv)
Language-model agents can take real actions by calling tools, so choosing the wrong tool can cause errors that are difficult to undo. Most evaluations only observe the tool choice after the model generates a call. We read this choice from the model before generation, and we steer it. For each tool, we average the model's hidden states from a few example requests to obtain a tool vector. Comparing a new request with these tool vectors predicts which tool it needs. The difference between two tool vectors gives a steering direction that can move the model toward a chosen tool without retraining. Across eight instruction-tuned models from 4B to 27B parameters, steering changes the generated call to a chosen target tool in 56-78% of held-out tool pairs, depending on the model. We also observe steering on real APIs from τ-bench and ToolBench, though less reliably. At the final layer, steering could work simply by raising the score of the target tool name. To test this, we compare the steering direction with a direction that only raises that score, layer by layer. In the middle layers, on the three models we examine in depth, the steering direction switches more calls than the score-raising direction. So the tool vectors capture part of the tool choice before the final layer. The same tool vectors also help identify likely tool-selection errors before generation. Calls are more likely to be wrong when a request is similarly close to two tool vectors. Across four models, this signal achieves an AUROC of 0.61-0.78 and outperforms a first-token confidence baseline on three of them. Together, these results suggest that the model's hidden state provides a way to read, steer, and check tool choice before a call is made.
Key facts. Cited by 10 papers in the map (data through 14 Oct 2026). Placed in Tool learning and function calling, one of 16 directions in LLM agents and tool use. Builds on 25 of its 37 references that are in the map. First posted on arXiv 8 May 2026.
Builds on
25 of 37 references.
Closest ancestors · 6
Same direction, newest first
- 2026ASA: Backbone-Training-Free Representation Engineering for Tool-Calling Agents4 Feb
- 2023ToolLLM: Facilitating Large Language Models to Master 16000+ Real-world APIs31 Jul
- 2023Gorilla: Large Language Model Connected with Massive APIs24 May
- 2023ToolkenGPT: Augmenting Frozen Language Models with Massive Tools via Tool Embeddings19 May
- 2023API-Bank: A Comprehensive Benchmark for Tool-Augmented LLMs14 Apr
- 2023Toolformer: Language Models Can Teach Themselves to Use Tools9 Feb
Other references · 19
- 2024The Llama 3 Herd of Modelscited by 14,715
- 2022Locating and Editing Factual Associations in GPTcited by 2,691
- 2023Sparse Autoencoders Find Highly Interpretable Features in Language Modelscited by 1,420
- 2024τ-bench: A Benchmark for Tool-Agent-User Interaction in Real-World Domainscited by 1,402
- 2023Inference-Time Intervention: Eliciting Truthful Answers from a Language Modelcited by 1,302
- 2022Interpretability in the Wild: a Circuit for Indirect Object Identification in GPT-2 smallcited by 1,165
- 2024Refusal in Language Models Is Mediated by a Single Directioncited by 1,032
- 2023Steering Language Models With Activation Engineeringcited by 922
- 2023Towards Automated Circuit Discovery for Mechanistic Interpretabilitycited by 724
- 2023The Geometry of Truth: Emergent Linear Structure in Large Language Model Representations of True/False Datasetscited by 712
- 2023Dissecting Recall of Factual Associations in Auto-Regressive Language Modelscited by 528
- 2024Gemma Scope: Open Sparse Autoencoders Everywhere All At Once on Gemma 2cited by 394
7 more
- 2023Emergent Linear Representations in World Models of Self-Supervised Sequence Modelscited by 319
- 2024Jumping Ahead: Improving Reconstruction Fidelity with JumpReLU Sparse Autoencoderscited by 300
- 2023An Overview of Catastrophic AI Riskscited by 222
- 2024Not All Language Model Features Are One-Dimensionally Linearcited by 212
- 2020Causal Mediation Analysis for Interpreting Neural NLP: The Case of Gender Biascited by 191
- 2023Linear Representations of Sentiment in Large Language Modelscited by 163
- 2026Internal Representations as Indicators of Hallucinations in Agent Tool Selectioncited by 19
Built on by
Citing papers per year: 2026: 10.
10 papers.
- 2026Looking Is Not Picking: An Attention-Segment Account of Tool-Selection Failures in LLM Agentscited by 3
- 2026A Few Neurons Reveal When LLMs Misuse Tools: Sparse Detection and Selective Steering for Reliable Tool Usecited by 3
- 2026Tool-Call Dependency Structure is Linearly Decodable in LLM Agent Residual Streamscited by 1
- 2026Tunable Tool-Call Rates in LLM Agents via Representation Steeringcited by 1
- 2026Continuous Interaction Diffusion: A Diffusion-Native Runtime for Asynchronous Tool-Augmented Reasoningcited by 0
- 2026When Does Correction Become Repair? Mechanistic Auditing of Internal Interventions in Tool-Using LLMscited by 0
- 2026CounterSteer: Suppressing Indirect Prompt Injection with Activation Steeringcited by 0
- 2026Rep2Skill: Representation-Guided Skill Self-Evolution for LLM Agentscited by 0
- 2026What Can Component-Replacement Evidence Establish? A Critical Scoping Review of Local Decisions in LLM Agentscited by 0
- 2026Understanding and Mitigating Hallucination Escape in Tool-Using LLM Agentscited by 0
- 2026Understanding and Mitigating Hallucination Escape in Tool-Using LLM Agents3 Oct 2026
- 2026What Can Component-Replacement Evidence Establish? A Critical Scoping Review of Local Decisions in LLM Agents30 Sep 2026
- 2026Rep2Skill: Representation-Guided Skill Self-Evolution for LLM Agents30 Sep 2026
- 2026CounterSteer: Suppressing Indirect Prompt Injection with Activation Steering29 Sep 2026
- 2026When Does Correction Become Repair? Mechanistic Auditing of Internal Interventions in Tool-Using LLMs28 Sep 2026
- 2026Tunable Tool-Call Rates in LLM Agents via Representation Steering25 Aug 2026
- 2026Continuous Interaction Diffusion: A Diffusion-Native Runtime for Asynchronous Tool-Augmented Reasoning11 Aug 2026
- 2026A Few Neurons Reveal When LLMs Misuse Tools: Sparse Detection and Selective Steering for Reliable Tool Use31 Jul 2026
- 2026Looking Is Not Picking: An Attention-Segment Account of Tool-Selection Failures in LLM Agents15 Jun 2026
- 2026Tool-Call Dependency Structure is Linearly Decodable in LLM Agent Residual Streams25 May 2026
Similar, not cited
Close in wording, no citation either way
- 2026Loud Failures, Quiet Failures: Fault Detection and Recovery in Tool-Using Language Model Agents
- 2026ToolRobustBench: Stage-Wise Perturbation Evaluation and Failure Diagnosis for Tool-Calling Agents
- 2026Feedback That Backfires: Why Small Language Model Agents Repeat the Call They Just Watched Fail
- 2026Calibration is the Bottleneck: An Action-Class Diagnostic of Multi-Turn Tool-Calling
- 2026Benchmarking the Benchmarks: A Validity Audit of Tool-Calling Evaluation
- 2026ToolFailBench: Diagnosing Tool-Use Failures in LLM Agents
- 2026The Calls are Coming from Inside the Model: Investigating Probe-based Detection of Tool-Calling Errors in LLMs
- 2025Tool Preferences in Agentic LLMs are Unreliable