mirror of
https://github.com/ruvnet/RuVector.git
synced 2026-05-27 00:25:10 +00:00
* feat(postgres): Add 7 advanced AI modules to ruvector-postgres Comprehensive implementation of advanced AI capabilities: ## New Modules (23,541 lines of code) ### 1. Self-Learning / ReasoningBank (`src/learning/`) - Trajectory tracking for query optimization - Pattern extraction using K-means clustering - ReasoningBank for pattern storage and matching - Adaptive search parameter optimization ### 2. Attention Mechanisms (`src/attention/`) - Scaled dot-product attention (core) - Multi-head attention with parallel heads - Flash Attention v2 (memory-efficient) - 10 attention types with PostgresEnum support ### 3. GNN Layers (`src/gnn/`) - Message passing framework - GCN (Graph Convolutional Network) - GraphSAGE with mean/max aggregation - Configurable aggregation methods ### 4. Hyperbolic Embeddings (`src/hyperbolic/`) - Poincaré ball model - Lorentz hyperboloid model - Hyperbolic distance metrics - Möbius operations ### 5. Sparse Vectors (`src/sparse/`) - COO format sparse vector type - Efficient sparse-sparse distance functions - BM25/SPLADE compatible - Top-k pruning operations ### 6. Graph Operations & Cypher (`src/graph/`) - Property graph storage (nodes/edges) - BFS, DFS, Dijkstra traversal - Cypher query parser (AST-based) - Query executor with pattern matching ### 7. Tiny Dancer Routing (`src/routing/`) - FastGRNN neural network - Agent registry with capabilities - Multi-objective routing optimization - Cost/latency/quality balancing ## Docker Infrastructure - Dockerfile with pgrx 0.12.6 and PostgreSQL 16 - docker-compose.yml with test runner - Initialization SQL with test tables - Shell scripts for dev/test/benchmark ## Feature Flags - `learning`, `attention`, `gnn`, `hyperbolic` - `sparse`, `graph`, `routing` - `ai-complete` and `graph-complete` bundles 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> * fix(docker): Copy entire workspace for pgrx build 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> * fix(docker): Build standalone crate without workspace 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> * docs: Update README to enhance clarity and structure * fix(postgres): Resolve compilation errors and Docker build issues - Fix simsimd Option/Result type mismatch in scaled_dot.rs - Fix f32/f64 type conversions in poincare.rs and lorentz.rs - Fix AVX512 missing wrapper functions by using AVX2 fallback - Fix Vec<Vec<f32>> to JsonB for pgrx pg_extern compatibility - Fix DashMap get() to get_mut() for mutable access - Fix router.rs dereference for best_score comparison - Update Dockerfile to copy pre-written SQL file for pgrx - Simplify init.sql to use correct function names - Add postgres-cli npm package for CLI tooling All changes tested successfully in Docker with: - Extension loads with AVX2 SIMD support (8 floats/op) - Distance functions verified working - PostgreSQL 16 container runs successfully 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> * feat: Add ruvLLM examples and enhanced postgres-cli Added from claude/ruvector-lfm2-llm-01YS5Tc7i64PyYCLecT9L1dN branch: - examples/ruvLLM: Complete LLM inference system with SIMD optimization - Pretraining, benchmarking, and optimization system - Real SIMD-optimized CPU inference engine - Comprehensive SOTA benchmark suite - Attention mechanisms, memory management, router Enhanced postgres-cli with full ruvector-postgres integration: - Sparse vector operations (BM25, top-k, prune, conversions) - Hyperbolic geometry (Poincare, Lorentz, Mobius operations) - Agent routing (Tiny Dancer system) - Vector quantization (binary, scalar, product) - Enhanced graph and learning commands 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> * fix(postgres-cli): Use native ruvector type instead of pgvector - Change createVectorTable to use ruvector type (native RuVector extension) - Add dimensions column for metadata since ruvector is variable-length - Update index creation to use simple btree (HNSW/IVFFlat TBD) - Tested against Docker container with ruvector extension 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> * feat(postgres): Add 53 SQL function definitions for all advanced modules Enable all advanced PostgreSQL extension functions by adding their SQL definitions to the extension file. This exposes all Rust #[pg_extern] functions to PostgreSQL. ## New SQL Functions (53 total) ### Hyperbolic Geometry (8 functions) - ruvector_poincare_distance, ruvector_lorentz_distance - ruvector_mobius_add, ruvector_exp_map, ruvector_log_map - ruvector_poincare_to_lorentz, ruvector_lorentz_to_poincare - ruvector_minkowski_dot ### Sparse Vectors (14 functions) - ruvector_sparse_create, ruvector_sparse_from_dense - ruvector_sparse_dot, ruvector_sparse_cosine, ruvector_sparse_l2_distance - ruvector_sparse_add, ruvector_sparse_scale, ruvector_sparse_to_dense - ruvector_sparse_nnz, ruvector_sparse_dim - ruvector_bm25_score, ruvector_tf_idf, ruvector_sparse_normalize - ruvector_sparse_topk ### GNN - Graph Neural Networks (5 functions) - ruvector_gnn_gcn_layer, ruvector_gnn_graphsage_layer - ruvector_gnn_gat_layer, ruvector_gnn_message_pass - ruvector_gnn_aggregate ### Routing/Agents - "Tiny Dancer" (11 functions) - ruvector_route_query, ruvector_route_with_context - ruvector_calculate_agent_affinity, ruvector_select_best_agent - ruvector_multi_agent_route, ruvector_create_agent_embedding - ruvector_get_routing_stats, ruvector_register_agent - ruvector_update_agent_performance, ruvector_adaptive_route - ruvector_fastgrnn_forward ### Learning/ReasoningBank (7 functions) - ruvector_record_trajectory, ruvector_get_verdict - ruvector_distill_memory, ruvector_adaptive_search - ruvector_learning_feedback, ruvector_get_learning_patterns - ruvector_optimize_search_params ### Graph/Cypher (8 functions) - ruvector_graph_create_node, ruvector_graph_create_edge - ruvector_graph_get_neighbors, ruvector_graph_shortest_path - ruvector_graph_pagerank, ruvector_cypher_query - ruvector_graph_traverse, ruvector_graph_similarity_search ## CLI Updates - Enabled hyperbolic geometry commands in postgres-cli - Added vector distance and normalize commands - Enhanced client with connection pooling and retry logic 🤖 Generated with [Claude Code](https://claude.com/claude-code) Co-Authored-By: Claude <noreply@anthropic.com> --------- Co-authored-by: Claude <noreply@anthropic.com>
138 lines
5.8 KiB
Markdown
138 lines
5.8 KiB
Markdown
# RuvLLM Documentation
|
|
|
|
## Overview
|
|
|
|
This directory contains documentation for the RuvLLM self-learning LLM architecture.
|
|
|
|
## Quick Links
|
|
|
|
- [Main README](../README.md) - Getting started, API reference, benchmarks
|
|
- [SPARC Documentation](./sparc/) - Design methodology documentation
|
|
|
|
## SPARC Methodology
|
|
|
|
The project was designed using the SPARC methodology:
|
|
|
|
| Phase | Document | Description |
|
|
|-------|----------|-------------|
|
|
| 1 | [Specification](./sparc/01-specification.md) | Requirements and acceptance criteria |
|
|
| 2 | [Pseudocode](./sparc/02-pseudocode.md) | Algorithm design and data flows |
|
|
| 3 | [Architecture](./sparc/03-architecture.md) | System design and component interactions |
|
|
| 4 | [Refinement](./sparc/04-refinement.md) | TDD implementation and iterative improvement |
|
|
| 5 | [Completion](./sparc/05-completion.md) | Integration, testing, and deployment |
|
|
|
|
## Architecture Overview
|
|
|
|
```
|
|
┌─────────────────────────────────────────────────────────────────┐
|
|
│ RuvLLM System │
|
|
├─────────────────────────────────────────────────────────────────┤
|
|
│ │
|
|
│ ┌─────────────┐ ┌─────────────┐ ┌─────────────┐ │
|
|
│ │ Embedding │ │ Memory │ │ Router │ │
|
|
│ │ Service │ │ (HNSW) │ │ (FastGRNN) │ │
|
|
│ └──────┬──────┘ └──────┬──────┘ └──────┬──────┘ │
|
|
│ │ │ │ │
|
|
│ └────────────────┼────────────────┘ │
|
|
│ │ │
|
|
│ ┌──────┴──────┐ │
|
|
│ │ Orchestrator │ │
|
|
│ └──────┬──────┘ │
|
|
│ │ │
|
|
│ ┌─────────────┐ ┌──────┴──────┐ ┌─────────────┐ │
|
|
│ │ Attention │ │ Inference │ │ Learning │ │
|
|
│ │ Engine │ │ Pool │ │ Service │ │
|
|
│ └─────────────┘ └─────────────┘ └─────────────┘ │
|
|
│ │
|
|
└─────────────────────────────────────────────────────────────────┘
|
|
```
|
|
|
|
## Module Documentation
|
|
|
|
### Core Modules
|
|
|
|
| Module | File | Description |
|
|
|--------|------|-------------|
|
|
| `orchestrator` | `src/orchestrator.rs` | Main coordinator, request processing pipeline |
|
|
| `memory` | `src/memory.rs` | HNSW-based semantic memory with graph expansion |
|
|
| `router` | `src/router.rs` | FastGRNN routing with EWC learning |
|
|
| `attention` | `src/attention.rs` | Multi-head graph attention with edge features |
|
|
| `embedding` | `src/embedding.rs` | Tokenization, embedding, and caching |
|
|
| `inference` | `src/inference.rs` | LFM2 model pool management |
|
|
| `learning` | `src/learning.rs` | Self-learning feedback loops |
|
|
| `compression` | `src/compression.rs` | Memory compression and clustering |
|
|
|
|
### Supporting Modules
|
|
|
|
| Module | File | Description |
|
|
|--------|------|-------------|
|
|
| `config` | `src/config.rs` | Configuration system with builder pattern |
|
|
| `error` | `src/error.rs` | Error types and result aliases |
|
|
| `types` | `src/types.rs` | Core domain types and structs |
|
|
|
|
## API Examples
|
|
|
|
### Basic Query
|
|
|
|
```rust
|
|
use ruvllm::{Config, RuvLLM};
|
|
|
|
let config = Config::builder().build()?;
|
|
let llm = RuvLLM::new(config).await?;
|
|
let response = llm.query("What is Rust?").await?;
|
|
```
|
|
|
|
### Session Management
|
|
|
|
```rust
|
|
let session = llm.new_session();
|
|
let r1 = llm.query_session(&session, "Tell me about vectors").await?;
|
|
let r2 = llm.query_session(&session, "How are they used in ML?").await?;
|
|
```
|
|
|
|
### Feedback Loop
|
|
|
|
```rust
|
|
use ruvllm::Feedback;
|
|
|
|
llm.feedback(Feedback {
|
|
request_id: response.request_id,
|
|
rating: Some(5),
|
|
correction: None,
|
|
task_success: Some(true),
|
|
}).await?;
|
|
```
|
|
|
|
## Performance Tuning
|
|
|
|
### Memory Configuration
|
|
|
|
```rust
|
|
Config::builder()
|
|
.hnsw_params(
|
|
32, // M: connections per node (higher = better recall, more memory)
|
|
200, // ef_construction: build quality (higher = slower build, better index)
|
|
64, // ef_search: search quality (higher = slower search, better recall)
|
|
)
|
|
```
|
|
|
|
### Router Configuration
|
|
|
|
```rust
|
|
Config::builder()
|
|
.router_hidden_dim(128) // Hidden state size (higher = more capacity)
|
|
```
|
|
|
|
### Learning Configuration
|
|
|
|
```rust
|
|
Config::builder()
|
|
.learning_enabled(true) // Enable self-learning
|
|
```
|
|
|
|
## Further Reading
|
|
|
|
- [LFM2 Paper](https://arxiv.org/abs/2511.23404v1) - Liquid Foundation Models
|
|
- [FastGRNN Paper](https://arxiv.org/abs/1901.02358) - Fast RNN architecture
|
|
- [HNSW Paper](https://arxiv.org/abs/1603.09320) - Approximate nearest neighbor search
|
|
- [EWC Paper](https://arxiv.org/abs/1612.00796) - Continual learning
|