Our new X account is live! Follow @wizwand_team for updates
WorkDL logo mark

Multi-Field Tool Retrieval

About

Integrating external tools enables Large Language Models (LLMs) to interact with real-world environments and solve complex tasks. Given the growing scale of available tools, effective tool retrieval is essential to mitigate constraints of LLMs' context windows and ensure computational efficiency. Existing approaches typically treat tool retrieval as a traditional ad-hoc retrieval task, matching user queries against the entire raw tool documentation. In this paper, we identify three fundamental challenges that limit the effectiveness of this paradigm: (i) the incompleteness and structural inconsistency of tool documentation; (ii) the significant semantic and granular mismatch between user queries and technical tool documents; and, most importantly, (iii) the multi-aspect nature of tool utility, that involves distinct dimensions, such as functionality, input constraints, and output formats, varying in format and importance. To address these challenges, we introduce Multi-Field Tool Retrieval, a framework designed to align user intent with tool representations through fine-grained, multi-field modeling. Experimental results show that our framework achieves SOTA performance on five datasets and a mixed benchmark, exhibiting superior generalizability and robustness.

Yichen Tang, Weihang Su, Yiqun Liu, Qingyao Ai• 2026

Related benchmarks

TaskDatasetResultRank
Tool RetrievalToolBench
NDCG@1058.54
44
Tool RetrievalAPIGen
NDCG@100.8575
44
Tool RetrievalAPIBank
NDCG@1073.2
44
Tool RetrievalGorilla
NDCG@100.3849
44
Tool RetrievalToolink
NDCG@100.5796
44
Tool RetrievalMixed
NDCG@100.59
44
Showing 6 of 6 rows

Other info

Follow for update