Skip to main content

Module retrieval

Module retrieval 

Source
Expand description

Semantic search engine for OS capabilities.

ToolRetriever maintains an in-memory index of all available tools (built-in OS tools, installed programs, OS services, and MCP bridges) and retrieves the most relevant ones for a given query using the embedding module’s cosine similarity.

§Usage

use std::sync::Arc;
use oxios_kernel::embedding::TfIdfEmbeddingProvider;
use oxios_kernel::tools::retrieval::{ToolRetriever, ToolEntry};

let embedder = Arc::new(TfIdfEmbeddingProvider);
let mut retriever = ToolRetriever::new(embedder);

let tool = ToolEntry {
    name: "exec".into(),
    category: "os-tool".into(),
    description: "Execute a shell command in a workspace".into(),
    skill_path: None,
    command: None,
};
retriever.index_tool(tool).await;

let query_embedding = retriever.embedder().embed("run a bash command").await?;
let results = retriever.retrieve(&query_embedding, 5);

Structs§

ScoredTool
A tool ranked by relevance to a query.
ToolEntry
A searchable entry in the tool index.
ToolRetriever
Semantic search engine for OS capabilities.

Functions§

build_kernel_manifest
Build a markdown kernel manifest from the set of active domains.
format_capability_index
Format retrieved tools as an XML capability index suitable for injection into an agent’s system prompt.