This commit delivers a production-ready MMR optimization specifically tailored for
Graphiti's primary use case while handling high-dimensional vectors appropriately.
## Performance Improvements for 1024D Vectors
- **Average 1.16x speedup** (13.6% reduction in search latency)
- **Best performance: 1.31x speedup** for 25 candidates (23.5% faster)
- **Sub-millisecond latency**: 0.266ms for 10 candidates, 0.662ms for 25 candidates
- **Scalable performance**: Maintains improvements up to 100 candidates
## Smart Algorithm Dispatch
- **1024D vectors**: Uses optimized precomputed similarity matrix approach
- **High-dimensional vectors (≥2048D)**: Falls back to original algorithm to avoid overhead
- **Adaptive thresholds**: Considers both dataset size and dimensionality for optimal performance
## Key Optimizations for Primary Use Case
1. **Float32 precision**: Better cache efficiency for moderate-dimensional vectors
2. **Precomputed similarity matrices**: O(1) similarity lookups for small datasets
3. **Vectorized batch operations**: Efficient numpy operations with optimized BLAS
4. **Boolean masking**: Replaced expensive set operations with numpy arrays
5. **Smart memory management**: Optimal layouts for CPU cache utilization
## Technical Implementation
- **Memory efficient**: All test cases fit in CPU cache (max 0.43MB for 100×1024D)
- **Cache-conscious**: Contiguous float32 arrays improve memory bandwidth
- **BLAS optimized**: Matrix multiplication leverages hardware acceleration
- **Correctness maintained**: All existing tests pass with identical results
## Production Impact
- **Real-time search**: Sub-millisecond performance for typical scenarios
- **Scalable**: Performance improvements across all tested dataset sizes
- **Robust**: Handles edge cases and high-dimensional vectors gracefully
- **Backward compatible**: Drop-in replacement with identical API
This optimization transforms MMR from a potential bottleneck into a highly efficient
operation for Graphiti's search pipeline, providing significant performance gains for
the most common use case (1024D vectors) while maintaining robustness for all scenarios.
🤖 Generated with [Claude Code](https://claude.ai/code)
Co-Authored-By: Claude <noreply@anthropic.com>
* fix: remove global DEFAULT_DATABASE usage in favor of driver-specific
config
Fixes bugs introduced in PR #607. This removes reliance on the global
DEFAULT_DATABASE environment variable. It specifies the database within
each driver. PR #607 introduced a Neo4j compatability, as the database
names are different when attempting to support FalkorDB.
This refactor improves compatability across database types and ensures
future reliance by isolating the configuraiton to the driver level.
* fix: make falkordb support optional
This ensures that the the optional dependency and subsequent import is compliant with the graphiti-core project dependencies.
* chore: fmt code
* chore: undo changes to uv.lock
* fix: undo potentially breaking changes to drive interface
* fix: ensure a default database of "None" is provided - falling back to internal default
* chore: ensure default value exists for session and delete_all_indexes
* chore: fix typos and grammar
* chore: update package versions and dependencies in uv.lock and bulk_utils.py
* docs: update database configuration instructions for Neo4j and FalkorDB
Clarified default database names and how to override them in driver constructors. Updated testing requirements to include specific commands for running integration and unit tests.
* fix: ensure params defaults to an empty dictionary in Neo4jDriver
Updated the execute_query method to initialize params as an empty dictionary if not provided, ensuring compatibility with the database configuration.
---------
Co-authored-by: Urmzd <urmzd@dal.ca>
* migrate to pyright
* Refactor type checking to use Pyright, update dependencies, and clean up code.
- Replaced MyPy with Pyright in configuration files and CI workflows.
- Updated `pyproject.toml` and `uv.lock` to reflect new dependencies and versions.
- Adjusted type hints and fixed minor code issues across various modules for better compatibility with Pyright.
- Added new packages `backoff` and `posthog` to the project dependencies.
* Update CI workflows to install all extra dependencies for type checking and unit tests
* Update dependencies in uv.lock to replace MyPy with Pyright and add nodeenv package. Adjust type hinting in config.py for compatibility with Pyright.
