Build latest book artifacts / build (push) Canceled after 0s
dependency resolution / resolve (3.11) (push) Canceled after 0s
dependency resolution / resolve (3.13) (push) Canceled after 0s
deploy-pages / build (push) Canceled after 0s
deploy-pages / deploy (push) Canceled after 0s
i18n consistency check / check (push) Canceled after 0s
provider adoption tests / test (chapter2/context-compression) (push) Canceled after 0s
provider adoption tests / test (chapter2/prompt-injection) (push) Canceled after 0s
provider adoption tests / test (chapter2/system-hint) (push) Canceled after 0s
provider adoption tests / test (chapter3/log-sanitization) (push) Canceled after 0s
web-search-agent tests / test (push) Canceled after 0s
web-search-agent tests / agentbook (push) Canceled after 0s
117 lines
2.9 KiB
Markdown
117 lines
2.9 KiB
Markdown
# Legal Document Indexing Script
|
|
|
|
This script indexes local Chinese legal documents from the `laws` directory into the retrieval pipeline.
|
|
|
|
## Features
|
|
|
|
- **Smart Chunking**: Respects paragraph boundaries with configurable soft (1024 chars) and hard limits (2048 chars)
|
|
- **Automatic Cleanup**: Cleans existing indexes before processing
|
|
- **Category Support**: Process specific legal categories or all documents
|
|
- **Progress Tracking**: Real-time progress updates and statistics
|
|
- **Verification**: Built-in test queries to verify indexing
|
|
|
|
## Prerequisites
|
|
|
|
1. Ensure the retrieval pipeline is running:
|
|
```bash
|
|
# Terminal 1: Start dense service
|
|
python dense_service.py
|
|
|
|
# Terminal 2: Start sparse service
|
|
python sparse_service.py
|
|
|
|
# Terminal 3: Start main pipeline
|
|
python main.py
|
|
```
|
|
|
|
2. The `laws` directory should be present with legal documents organized by category:
|
|
```
|
|
laws/
|
|
├── 1-宪法/
|
|
├── 2-宪法相关法/
|
|
├── 3-民法典/
|
|
├── 3-民法商法/
|
|
├── 4-行政法/
|
|
├── 5-经济法/
|
|
├── 6-社会法/
|
|
├── 7-刑法/
|
|
└── 8-诉讼与非诉讼程序法/
|
|
```
|
|
|
|
## Usage
|
|
|
|
### Basic Usage
|
|
```bash
|
|
# Index all legal documents
|
|
python index_local_laws.py
|
|
|
|
# Index with verification tests
|
|
python index_local_laws.py --verify
|
|
```
|
|
|
|
### Advanced Options
|
|
```bash
|
|
# Index only first 10 documents
|
|
python index_local_laws.py --max-docs 10
|
|
|
|
# Index specific categories only
|
|
python index_local_laws.py --categories "宪法" "民法典" "刑法"
|
|
|
|
# Use custom pipeline URL
|
|
python index_local_laws.py --pipeline-url http://localhost:8080
|
|
|
|
# Skip cleanup (append to existing index)
|
|
python index_local_laws.py --no-cleanup
|
|
```
|
|
|
|
## Chunking Strategy
|
|
|
|
The script uses intelligent chunking that:
|
|
1. Accumulates paragraphs until soft limit (1024 chars) is exceeded
|
|
2. Continues adding if next paragraph fits within hard limit (2048 chars)
|
|
3. Cuts at paragraph boundary when possible
|
|
4. Force splits oversized paragraphs at hard limit
|
|
|
|
This approach ensures:
|
|
- Legal provisions remain intact when possible
|
|
- Context is preserved within chunks
|
|
- Search relevance is optimized
|
|
|
|
## Output Statistics
|
|
|
|
After indexing, the script displays:
|
|
- Processing time
|
|
- Number of documents and categories processed
|
|
- Total chunks created and indexed
|
|
- Average chunks per document
|
|
- Processing speed
|
|
- Any errors encountered
|
|
|
|
## Verification
|
|
|
|
Use the `--verify` flag to run test searches:
|
|
```bash
|
|
python index_local_laws.py --verify
|
|
```
|
|
|
|
Test queries include:
|
|
- 民法典 (Civil Code)
|
|
- 合同法 (Contract Law)
|
|
- 劳动法 (Labor Law)
|
|
- 刑法 (Criminal Law)
|
|
- 宪法 (Constitution)
|
|
|
|
## Document Store
|
|
|
|
The script maintains a local `document_store.json` file tracking:
|
|
- Document metadata
|
|
- Number of chunks per document
|
|
- Indexing timestamps
|
|
- Category information
|
|
|
|
## Error Handling
|
|
|
|
- Documents that fail to read are skipped
|
|
- Failed chunk indexing is logged but doesn't stop processing
|
|
- Statistics track all errors for review
|