retrieval

retrieval

Retrieval tool — search memory backends for relevant context.

Classes

RetrievalTool

RetrievalTool(
    backend: Optional[MemoryBackend] = None,
    *,
    top_k: int = 5,
)

Bases: BaseTool

Search the memory backend and return formatted context.

Source code in src/diapason/tools/retrieval.py

def __init__(
    self,
    backend: Optional[MemoryBackend] = None,
    *,
    top_k: int = 5,
) -> None:
    self._backend = backend
    self._top_k = top_k

Functions: