AI Tool Retriever

This project provides a lightweight, extensible library for dynamically selecting the most relevant tools for your AI agent based on user queries.

Installation

npm install ai-tool-retriever

Usage

const retriever = await ToolRetriever.create({
	tools: allMyTools,
	embeddingProvider: anyEmbeddingProvider,
	store: inMemoryOrPersistentStore,
})

const relevantTools = await retriever.retrieve(userPrompt)

About

This repo contains the following packages:

• packages/ai-tool-retriever: The core library for semantic tool retrieval.
• packages/benchmark A package to benchmark the performance of embedding and search operations.
• packages/e2e A package to validate behavior in real-world scenarios.