fastbktree 0.1.1

A Rust BK-tree implementation for fast fuzzy string matching in Python

fastbktree :deciduous_tree:

A Rust implementation of BK-trees for fast fuzzy string matching in Python.

Installation

uv add fastbktree
# pip install fastbktree

Quickstart

from fastbktree import BKTree

# Create a BK-tree from a list of strings
tree = BKTree(["hello", "help", "hell", "shell", "helper"])
results = tree.search("helo", max_distance=2)

Reference

BKTree(iterable)

Creates a new BK-tree from an iterable of strings.

tree = BKTree(["word1", "word2", "word3"])

tree.search(query: str, max_distance: int) -> List[Tuple[str, int]]

Searches for strings similar to query within max_distance Levenshtein distance. Returns a list of (string, distance) tuples.

results = tree.search("query", max_distance=2)
# [('hello', 1), ('help', 1), ('shell', 2), ('hell', 1)]

levenshtein_distance(a: str, b: str) -> int

Pure Rust implementation of Levenshtein distance between two strings.

from fastbktree import levenshtein_distance

distance = levenshtein_distance("kitten", "sitting")  # 3

License

MIT