文章主要內容和創(chuàng)新點主要內容本文針對大型語言模型（LLM）驅動的AI代理在工具檢索中存在的問題，提出了一種名為Tool Graph Retriever（TGR，工具圖檢索器）的方法。背景：隨著AI代理配備的工具數量激增，受限于模型上下文長度，需通過檢索篩選工具?，F有方法主要依賴用戶查詢與工具描述的語義相似性，忽略工具間的依賴關系，常遺漏任務執(zhí)行必需的前置工具（如“更新郵箱”需先“驗證”和“登錄”，但現有方法可能僅檢索到“更新郵箱”）。TGR核心步驟：依賴識別：構建工具依賴識別數據集TDI300K，訓練判別器以識別工具間的依賴關系（如t_a依賴t_b、無依賴、t_b依賴t_a）；基于圖的工具編碼：將工具表示為節(jié)點、依賴關系為邊的有向圖，通過圖卷積將依賴信息整合到工具表示中；在線檢索：使用整合依賴信息的工具表示計算與用戶查詢的相似度，完成檢索。實驗結果：在API-Bank和ToolBench等數據集上，TGR顯著提升了現有主流方法的檢索性能（Recall、NDCG、Pass Rate指標），達到當前最優(yōu)（SOTA）。創(chuàng)新點提出TGR方