Expose ProllyTree SQL API to Python and Add Chronological LangGraph TA-RAG Example

pyproject.toml

@@ -39,6 +39,7 @@ Repository = "https://github.com/zhangfengcdt/prollytree.git"
 "Bug Tracker" = "https://github.com/zhangfengcdt/prollytree/issues"
 
 [tool.maturin]
-features = ["python"]
+# Default features - can be overridden with --features flag
+features = ["python", "sql"]
 module-name = "prollytree"
 python-source = "python"

python/build_python.sh

@@ -13,9 +13,38 @@
 # limitations under the License.
 
 # Build script for ProllyTree Python bindings
+#
+# Usage:
+#   ./build_python.sh                    # Build with default Python bindings
+#   ./build_python.sh --with-sql         # Build with SQL support
+#   ./build_python.sh --all-features     # Build with all features (Python + SQL)
+#   ./build_python.sh --features "python sql"  # Specify features explicitly
+#   ./build_python.sh --install          # Build and install the package
+#   ./build_python.sh --with-sql --install  # Build with SQL and install
 
 set -e
 
+# Show help if requested
+if [[ "$1" == "--help" || "$1" == "-h" ]]; then
+    echo "Build script for ProllyTree Python bindings"
+    echo ""
+    echo "Usage:"
+    echo "  ./build_python.sh [OPTIONS]"
+    echo ""
+    echo "Options:"
+    echo "  --with-sql           Build with SQL support"
+    echo "  --all-features       Build with all features (Python + SQL)"
+    echo "  --features FEATURES  Specify features explicitly (e.g., 'python sql')"
+    echo "  --install            Install the built package after building"
+    echo "  --help, -h           Show this help message"
+    echo ""
+    echo "Examples:"
+    echo "  ./build_python.sh                       # Basic Python bindings"
+    echo "  ./build_python.sh --with-sql            # With SQL support"
+    echo "  ./build_python.sh --with-sql --install  # Build and install with SQL"
+    exit 0
+fi
+
 echo "🔧 Building ProllyTree Python bindings..."
 
 # Check if maturin is installed
@@ -27,9 +56,33 @@ fi
 # Change to project root directory
 cd "$(dirname "$0")/.."
 
+# Parse command line arguments for features
+FEATURES="python"
+for arg in "$@"; do
+    case $arg in
+        --features)
+            shift
+            FEATURES="$1"
+            shift
+            ;;
+        --features=*)
+            FEATURES="${arg#*=}"
+            shift
+            ;;
+        --with-sql)
+            FEATURES="python sql"
+            shift
+            ;;
+        --all-features)
+            FEATURES="python sql"
+            shift
+            ;;
+    esac
+done
+
 # Build the wheel
-echo "🍹 Building wheel with maturin..."
-maturin build --release --features python
+echo "🍹 Building wheel with maturin (features: $FEATURES)..."
+maturin build --release --features "$FEATURES"
 
 # Find the built wheel
 WHEEL_PATH=$(find target/wheels -name "prollytree-*.whl" | head -1)
@@ -54,17 +107,51 @@ from prollytree import ProllyTree, TreeConfig
 tree = ProllyTree()
 tree.insert(b'test', b'value')
 result = tree.find(b'test')
-print(f'✅ Test passed: {result == b\"value\"}')
+print(f'✅ Basic test passed: {result == b\"value\"}')
 "
+
+    # Test SQL functionality if available
+    if [[ "$FEATURES" == *"sql"* ]]; then
+        echo "🧪 Testing SQL functionality..."
+        python3 -c "
+import tempfile
+import subprocess
+import os
+from prollytree import ProllySQLStore
+
+# Create temp dir and init git
+with tempfile.TemporaryDirectory() as tmpdir:
+    subprocess.run(['git', 'init'], cwd=tmpdir, capture_output=True)
+    subprocess.run(['git', 'config', 'user.name', 'Test'], cwd=tmpdir, capture_output=True)
+    subprocess.run(['git', 'config', 'user.email', 'test@test.com'], cwd=tmpdir, capture_output=True)
+
+    # Create SQL store
+    store_dir = os.path.join(tmpdir, 'data')
+    os.makedirs(store_dir)
+    store = ProllySQLStore(store_dir)
+
+    # Test basic SQL operations
+    store.create_table('test', [('id', 'INTEGER'), ('name', 'TEXT')])
+    store.insert('test', [[1, 'Test']])
+    result = store.select('test')
+
+    print(f'✅ SQL test passed: {len(result) == 1 and result[0][\"name\"] == \"Test\"}')
+" || echo "⚠️  SQL test skipped (import failed - may need git in temp dir)"
+    fi
 fi
 
 echo "🎉 Build complete!"
 echo ""
+echo "Built with features: $FEATURES"
+echo ""
 echo "To install the wheel manually:"
 echo "  pip install $WHEEL_PATH"
 echo ""
 echo "To test the bindings:"
 echo "  python3 test_python_binding.py"
+if [[ "$FEATURES" == *"sql"* ]]; then
+    echo "  python3 python/examples/sql_example.py  # Test SQL functionality"
+fi
 echo ""
 echo "To publish to PyPI:"
 echo "  cd python && ./publish_python.sh test    # Publish to TestPyPI first"