LightRAG/starter/MULTITENANT_TESTING_ENABLED.md
Raphael MANSUY fe9b8ec02a
tests: stabilize integration tests + skip external services; fix multi-tenant API behavior and idempotency (#4)
* feat: Implement multi-tenant architecture with tenant and knowledge base models

- Added data models for tenants, knowledge bases, and related configurations.
- Introduced role and permission management for users in the multi-tenant system.
- Created a service layer for managing tenants and knowledge bases, including CRUD operations.
- Developed a tenant-aware instance manager for LightRAG with caching and isolation features.
- Added a migration script to transition existing workspace-based deployments to the new multi-tenant architecture.

* chore: ignore lightrag/api/webui/assets/ directory

* chore: stop tracking lightrag/api/webui/assets (ignore in .gitignore)

* feat: Initialize LightRAG Multi-Tenant Stack with PostgreSQL

- Added README.md for project overview, setup instructions, and architecture details.
- Created docker-compose.yml to define services: PostgreSQL, Redis, LightRAG API, and Web UI.
- Introduced env.example for environment variable configuration.
- Implemented init-postgres.sql for PostgreSQL schema initialization with multi-tenant support.
- Added reproduce_issue.py for testing default tenant access via API.

* feat: Enhance TenantSelector and update related components for improved multi-tenant support

* feat: Enhance testing capabilities and update documentation

- Updated Makefile to include new test commands for various modes (compatibility, isolation, multi-tenant, security, coverage, and dry-run).
- Modified API health check endpoint in Makefile to reflect new port configuration.
- Updated QUICK_START.md and README.md to reflect changes in service URLs and ports.
- Added environment variables for testing modes in env.example.
- Introduced run_all_tests.sh script to automate testing across different modes.
- Created conftest.py for pytest configuration, including database fixtures and mock services.
- Implemented database helper functions for streamlined database operations in tests.
- Added test collection hooks to skip tests based on the current MULTITENANT_MODE.

* feat: Implement multi-tenant support with demo mode enabled by default

- Added multi-tenant configuration to the environment and Docker setup.
- Created pre-configured demo tenants (acme-corp and techstart) for testing.
- Updated API endpoints to support tenant-specific data access.
- Enhanced Makefile commands for better service management and database operations.
- Introduced user-tenant membership system with role-based access control.
- Added comprehensive documentation for multi-tenant setup and usage.
- Fixed issues with document visibility in multi-tenant environments.
- Implemented necessary database migrations for user memberships and legacy support.

* feat(audit): Add final audit report for multi-tenant implementation

- Documented overall assessment, architecture overview, test results, security findings, and recommendations.
- Included detailed findings on critical security issues and architectural concerns.

fix(security): Implement security fixes based on audit findings

- Removed global RAG fallback and enforced strict tenant context.
- Configured super-admin access and required user authentication for tenant access.
- Cleared localStorage on logout and improved error handling in WebUI.

chore(logs): Create task logs for audit and security fixes implementation

- Documented actions, decisions, and next steps for both audit and security fixes.
- Summarized test results and remaining recommendations.

chore(scripts): Enhance development stack management scripts

- Added scripts for cleaning, starting, and stopping the development stack.
- Improved output messages and ensured graceful shutdown of services.

feat(starter): Initialize PostgreSQL with AGE extension support

- Created initialization scripts for PostgreSQL extensions including uuid-ossp, vector, and AGE.
- Ensured successful installation and verification of extensions.

* feat: Implement auto-select for first tenant and KB on initial load in WebUI

- Removed WEBUI_INITIAL_STATE_FIX.md as the issue is resolved.
- Added useTenantInitialization hook to automatically select the first available tenant and KB on app load.
- Integrated the new hook into the Root component of the WebUI.
- Updated RetrievalTesting component to ensure a KB is selected before allowing user interaction.
- Created end-to-end tests for multi-tenant isolation and real service interactions.
- Added scripts for starting, stopping, and cleaning the development stack.
- Enhanced API and tenant routes to support tenant-specific pipeline status initialization.
- Updated constants for backend URL to reflect the correct port.
- Improved error handling and logging in various components.

* feat: Add multi-tenant support with enhanced E2E testing scripts and client functionality

* update client

* Add integration and unit tests for multi-tenant API, models, security, and storage

- Implement integration tests for tenant and knowledge base management endpoints in `test_tenant_api_routes.py`.
- Create unit tests for tenant isolation, model validation, and role permissions in `test_tenant_models.py`.
- Add security tests to enforce role-based permissions and context validation in `test_tenant_security.py`.
- Develop tests for tenant-aware storage operations and context isolation in `test_tenant_storage_phase3.py`.

* feat(e2e): Implement OpenAI model support and database reset functionality

* Add comprehensive test suite for gpt-5-nano compatibility

- Introduced tests for parameter normalization, embeddings, and entity extraction.
- Implemented direct API testing for gpt-5-nano.
- Validated .env configuration loading and OpenAI API connectivity.
- Analyzed reasoning token overhead with various token limits.
- Documented test procedures and expected outcomes in README files.
- Ensured all tests pass for production readiness.

* kg(postgres_impl): ensure AGE extension is loaded in session and configure graph initialization

* dev: add hybrid dev helper scripts, Makefile, docker-compose.dev-db and local development docs

* feat(dev): add dev helper scripts and local development documentation for hybrid setup

* feat(multi-tenant): add detailed specifications and logs for multi-tenant improvements, including UX, backend handling, and ingestion pipeline

* feat(migration): add generated tenant/kb columns, indexes, triggers; drop unused tables; update schema and docs

* test(backward-compat): adapt tests to new StorageNameSpace/TenantService APIs (use concrete dummy storages)

* chore: multi-tenant and UX updates — docs, webui, storage, tenant service adjustments

* tests: stabilize integration tests + skip external services; fix multi-tenant API behavior and idempotency

- gpt5_nano_compatibility: add pytest-asyncio markers, skip when OPENAI key missing, prevent module-level asyncio.run collection, add conftest
- Ollama tests: add server availability check and skip markers; avoid pytest collection warnings by renaming helper classes
- Graph storage tests: rename interactive test functions to avoid pytest collection
- Document & Tenant routes: support external_ids for idempotency; ensure HTTPExceptions are re-raised
- LightRAG core: support external_ids in apipeline_enqueue_documents and idempotent logic
- Tests updated to match API changes (tenant routes & document routes)
- Add logs and scripts for inspection and audit
2025-12-04 16:04:21 +08:00

209 lines
5.6 KiB
Markdown

# Multi-Tenant Testing Enabled by Default
## Summary
The LightRAG starter environment has been fully configured to test with **multi-tenant mode enabled by default** with **2 pre-configured demo tenants**.
## What Changed
### 1. ✅ Environment Configuration (`.env`)
- **MULTITENANT_MODE=demo** is now set by default
- This enables automatic multi-tenant testing with 2 demo tenants
### 2. ✅ Template Updated (`env.example`)
- Updated with clearer documentation
- Explicitly marked "demo" as the **DEFAULT** testing mode
- Includes information about the 2 pre-configured tenants:
- Tenant 1: acme-corp (kb-prod, kb-dev)
- Tenant 2: techstart (kb-main, kb-backup)
### 3. ✅ Docker Compose Configuration (`docker-compose.yml`)
- Added `MULTITENANT_MODE` environment variable to the lightrag-api service
- Defaults to `demo` mode when not explicitly set
### 4. ✅ Makefile Enhanced (`Makefile`)
#### `make up` command now displays:
```
📊 Testing Mode: demo
✓ Multi-Tenant Demo Mode (2 tenants)
• Tenant 1: acme-corp (kb-prod, kb-dev)
• Tenant 2: techstart (kb-main, kb-backup)
```
#### `make init-db` command now displays:
```
📦 Pre-configured Demo Tenants:
★ Tenant: acme-corp
- kb-prod (Production KB)
- kb-dev (Development KB)
★ Tenant: techstart
- kb-main (Main KB)
- kb-backup (Backup KB)
💡 Tips:
• Use X-Tenant-ID and X-KB-ID headers in API requests
• Example: curl -H 'X-Tenant-ID: acme-corp' -H 'X-KB-ID: kb-prod' ...
```
### 5. ✅ Database Initialization (`init-postgres.sql`)
- Already includes SQL to create both demo tenants and their knowledge bases
- Creates composite key constraints: (tenant_id, kb_id, id)
- Prevents cross-tenant data access at database level
### 6. ✅ Documentation Created
- `QUICK_START_MULTITENANT.md` - Updated quick start guide with multi-tenant examples
## How to Test
### Quick Start
```bash
cd starter
# Step 1: Setup
make setup
# Step 2: Start services with demo mode
make up
# Step 3: Initialize database
make init-db
# Step 4: Access services
# Web UI: http://localhost:3001
# API Docs: http://localhost:8000/docs
```
### API Testing with Both Tenants
**Test Tenant 1 (acme-corp):**
```bash
curl -X GET http://localhost:8000/health \
-H "X-Tenant-ID: acme-corp" \
-H "X-KB-ID: kb-prod"
```
**Test Tenant 2 (techstart):**
```bash
curl -X GET http://localhost:8000/health \
-H "X-Tenant-ID: techstart" \
-H "X-KB-ID: kb-main"
```
### Verify Database Setup
```bash
# Connect to database
make db-shell
# Check tenants
SELECT tenant_id, name FROM tenants;
# Check knowledge bases
SELECT tenant_id, kb_id, name FROM knowledge_bases;
# Exit
\q
```
## Demo Tenant Details
### Tenant 1: acme-corp
- **ID**: acme-corp
- **Name**: Acme Corporation
- **Description**: Enterprise customer - production deployment
- **Knowledge Bases**:
- `kb-prod` - Production KB for live data
- `kb-dev` - Development KB for testing
### Tenant 2: techstart
- **ID**: techstart
- **Name**: TechStart Inc
- **Description**: Startup customer - evaluation environment
- **Knowledge Bases**:
- `kb-main` - Main KB for primary knowledge
- `kb-backup` - Backup KB for archival data
## Key Features
**Multi-Tenant Isolation**: Complete data separation at database level
**Composite Keys**: (tenant_id, kb_id, id) prevents cross-tenant data access
**Automatic Setup**: Demo tenants created automatically on `make init-db`
**Clear Output**: `make up` displays the active testing mode
**API Headers**: X-Tenant-ID and X-KB-ID for multi-tenant requests
**Database Verification**: Built-in commands to verify tenant isolation
## Switching Testing Modes
### From Demo Mode → Single-Tenant Mode
```bash
# Edit .env
sed -i.bak 's/MULTITENANT_MODE=demo/MULTITENANT_MODE=on/' .env
# Restart services
make restart
make db-reset
make init-db
```
### From Demo Mode → Compatibility Mode (Single-Tenant Like main branch)
```bash
# Edit .env
sed -i.bak 's/MULTITENANT_MODE=demo/MULTITENANT_MODE=off/' .env
# Restart services
make restart
make db-reset
make init-db
# Note: In this mode, X-Tenant-ID and X-KB-ID headers are not required
curl http://localhost:8000/health # Works without headers
```
## Files Modified
1. **starter/.env** - MULTITENANT_MODE=demo (already present)
2. **starter/env.example** - Updated documentation
3. **starter/docker-compose.yml** - Added MULTITENANT_MODE env var
4. **starter/Makefile** - Enhanced `make up` and `make init-db` output
5. **starter/init-postgres.sql** - Creates demo tenants (already present)
## Files Created
1. **starter/QUICK_START_MULTITENANT.md** - Updated quick start guide
## Verification Checklist
-`make up` displays "Multi-Tenant Demo Mode (2 tenants)"
-`make init-db` displays pre-configured demo tenants
- ✅ env.example clearly shows MULTITENANT_MODE=demo as default
- ✅ docker-compose.yml passes MULTITENANT_MODE to API service
- ✅ Database initialization creates acme-corp and techstart tenants
- ✅ Each tenant has its configured knowledge bases
- ✅ Composite key constraints prevent cross-tenant access
- ✅ API can be accessed with X-Tenant-ID and X-KB-ID headers
## Next Steps
1. **Manual Testing**:
```bash
make up
make init-db
# Test both tenants with curl commands above
```
2. **Integration Testing**:
```bash
make test-multi
make test-security
```
3. **Documentation Review**:
- See `docs/adr/008-multi-tenant-testing-strategy.md` for detailed testing strategies
- Review `QUICK_START_MULTITENANT.md` for additional examples
---
**Status**: ✅ Complete and Ready for Testing
**Date**: November 22, 2025
**Testing Mode**: Multi-Tenant Demo (2 Tenants) - Enabled by Default