Skip to content
Preprint

Extracting Knowledge from Tools in LLM Agents

Aug 2026 · 0 citations
Computer Science

TL;DR

This paper proposes ToolSiphon, a query-only extraction attack that introduces two complementary signals: a target-discriminative signal, implemented through Tool Contrastive Analysis, to steer queries toward the target tool; and a response-grounded factual signal, implemented through Evidence Chained Feedback, to mitigate argument compression and progressively expand extraction coverage.

Abstract

LLM agents commonly use knowledge-based tools and access their underlying files, databases, and search indexes through tool invocation. This integration improves agents'ability to provide domain-specific services but also introduces the risk of tool-mediated knowledge extraction: source content exposed to an agent for legitimate responses may be progressively recovered from its outputs, enabling reconstruction of the knowledge source behind a target tool. This paper systematically investigates this risk and identifies two challenges introduced by tool invocation: tool-selection uncertainty, where an agent may invoke a competing tool instead of the target tool, and tool-argument compression, where fine-grained query information may be lost when the agent generates tool arguments. To tackle these challenges, we propose ToolSiphon, a query-only extraction attack that introduces two complementary signals: a target-discriminative signal, implemented through Tool Contrastive Analysis, to steer queries toward the target tool; and a response-grounded factual signal, implemented through Evidence Chained Feedback, to mitigate argument compression and progressively expand extraction coverage. Across three types of knowledge-based tools and six domain-specific datasets, ToolSiphon recovers 74.3% of source records on average when coarse-grained information about non-target tools is available, with 83.2% textual recovery and 90.2% semantic similarity. Even without such information, it recovers 66.3% of source records. ToolSiphon also remains effective against representative defenses and on three real-world agent platforms.

View source

Similar papers

Preprint Aug 2026

ToolRobustBench: Stage-Wise Perturbation Evaluation and Failure Diagnosis for Tool-Calling Agents

ToolRobustBench provides a deterministic and cascade-aware benchmark for diagnosing robustness beyond clean tool-calling accuracy, where a tool-calling agent is an LLM system that selects a tool, supplies structured arguments, and interprets its returned feedback.

YiShan Zheng, Yuan Wu, Yi Chang · 4 citations
#artificial intelligence Preprint Sep 2026

Toollery: Scaling LLM Agents to Thousands of Skills and Tools

Toollery is presented, a training-free candidate-compression framework for scalable LLM skill/tool selection that keeps online selection bounded to a compact top-$k$ candidate set and improves recall over ordinary specification retrieval.

Xiang-Xi Tian, Ran-Yun Guan · 0 citations
Preprint Aug 2026

Hybrid Semantic Tool Discovery for Enterprise MCP Gateway: Architecture and Implementation

This work presents SCOUT (Selective Context Optimization for Universal Tooling for Universal Tooling), which reframes tool exposure as a context-selection problem, injecting only tools relevant to the current step, and reduces MCP tool-token consumption by 99%, cutting per-query inference cost at enterprise scale.

Olympia Saha, Amy Wang, Srinivasan Manoharan · 0 citations
#artificial intelligence Preprint Sep 2026

Agents'Overreliance on Unreliable Tools

LLM agents use tools to access information and perform computations beyond their parametric knowledge. Existing tool-use benchmarks evaluate whether agents select and call the right tools, assuming that tool returns are reliable. However, tools can return plausible but incorrect outputs. We evaluate 14 models with thre...

Hoyeol Yang, Woojung Song, Taewon Kim et al. · 2 citations
Preprint Aug 2026

Risk-Aware Reranking for Agentic Tool Retrieval

It is indicated that retrieval-stage filtering can reduce the candidate action space exposed to agents before execution, complementing downstream tool-use safeguards, and providing a conservative operating point for safety-critical deployments.

Qinfei Li, Xiaoxuan Dong, Jin Zhang et al. · 2 citations
Conference Sep 2026

Mitigating Tool Overuse for LLMs via Active Knowledge Boundary Probing

RADAR is proposed, an automated framework for knowledge boundary discovery and tool overuse mitigation that globally propagates labels and rewrites conflicts, yielding deployment-aligned decisions with fewer unnecessary tool calls.

Zhao-Yu Yang, Wenjun Ke, Yuan-Yao Li et al. · 0 citations

We use cookies to run the site and, with your consent, for analytics and to show ads. See our Cookie Policy.