legco_ai_assistant/AGENTS.md

234 lines
9.2 KiB
Markdown

# RAG Video Q&A — Project Knowledge Base
**Generated:** 2026-04-22
**Source:** development_plan.md
**Status:** Greenfield (no code yet)
---
## OVERVIEW
RAG-powered Video Q&A web app. Phase 1: text → ChromaDB retrieval → bullet-point answer. Phase 2: video upload → real-time ASR → auto/manual RAG query. FastAPI backend + React 18 (Vite) frontend.
## STRUCTURE
```
app/
├── backend/ # FastAPI (Python)
│ ├── app/
│ │ ├── main.py
│ │ ├── routers/ # query.py, ingest.py, video.py, ws_asr.py
│ │ ├── services/ # rag.py, llm_client.py, asr_client.py, video_service.py
│ │ ├── models/ # Pydantic schemas
│ │ ├── core/ # config.py, database.py
│ │ └── utils/ # chunking.py, metadata_extraction.py
│ ├── uploads/ # video storage (max 300MB)
│ ├── requirements.txt
│ └── .env.example
├── frontend/ # React 18 + TS + Vite
│ ├── src/
│ │ ├── components/ # shadcn/ui + custom
│ │ ├── pages/
│ │ ├── lib/
│ │ │ └── api.ts # API client (TanStack Query)
│ │ └── App.tsx
│ ├── package.json
│ └── vite.config.ts
├── chroma_db/ # Persistent vector store
├── Dockerfile
├── docker-compose.yml
├── nginx.conf
└── deploy.sh
```
## WHERE TO LOOK
| Task | Location | Notes |
|------|----------|-------|
| API routes | `backend/app/routers/` | Versioned `/api/v1/...` |
| Business logic | `backend/app/services/` | RAG, LLM, ASR, video |
| Schemas | `backend/app/models/` | Pydantic request/response |
| Config | `backend/app/core/config.py` | `.env` driven |
| DB init | `backend/app/core/database.py` | ChromaDB persistent |
| Frontend API | `frontend/src/lib/api.ts` | TanStack Query |
| UI components | `frontend/src/components/` | shadcn/ui + Tailwind |
## CODE MAP
*Greenfield — no code yet. See development_plan.md for full specification.*
## CONVENTIONS
- **Backend**: `snake_case` files; routers thin, services thick; `.env` for all LLM/ASR config
- **Frontend**: PascalCase components; `lib/api.ts` single API client; TanStack Query for server state
- **API**: Path versioning `/api/v1/`; WebSocket at `/ws/asr/{video_id}`
- **RAG**: Strict prompt — answer ONLY from retrieved context; bullet-point format
- **Metadata**: Every doc chunk must have `filename`, `upload_date`, `content_summary`
## ANTI-PATTERNS (THIS PROJECT)
- Hardcode LLM URLs/keys — always `.env`
- Business logic in routers — belongs in `services/`
- Non-persistent ChromaDB — must use `chroma_db/` directory
- LLM hallucination outside retrieved context — strict RAG prompt enforced
- Plain text responses — always bullet points with source metadata
- Missing document metadata — breaks source attribution
- Add authentication — public demo only
- Mobile-first design — desktop only at this stage
## UNIQUE STYLES
- **Dual ASR trigger**: automatic (on transcript update) + manual "Ask from Video" button
- **Layout**: Top-Left video player | Top-Right transcript + input | Bottom RAG response
- **Provider switching**: same codebase runs dev (OpenRouter/Alibaba Cloud) and prod (local vLLM)
- **Video limit**: 300MB max, MP4 + common formats
## TESTING
**Backend test directory**: `backend/app/test/`
**Naming convention** (pytest, flat structure, phase-prefixed):
```
test_phase<N>_<module_or_feature>.py
```
**Examples**:
- `test_phase1_ingest.py` — Document upload & ChromaDB ingestion
- `test_phase1_query.py` — RAG query endpoint
- `test_phase1_rag_service.py` — RAG retrieval + strict prompt logic
- `test_phase1_llm_client.py` — LLM client (mocked provider)
- `test_phase1_chunking.py` — Document chunking utils
- `test_phase1_metadata.py` — Metadata extraction
- `test_phase2_video_upload.py` — Video upload (<300MB, format validation)
- `test_phase2_asr_client.py` ASR transcription client
- `test_phase2_ws_asr.py` WebSocket audio streaming
- `test_phase2_query_from_video.py` Auto/manual trigger from transcript
- `test_integration_phase1.py` End-to-end text RAG answer
- `test_integration_phase2.py` End-to-end video ASR RAG answer
**Rules**:
- Use `pytest` + `pytest-asyncio` for async tests
- Mock all external LLM/ASR calls (do not hit live APIs in tests)
- Use `tmp_path` fixture for ChromaDB test instances
- Each test file must have a module-level docstring describing coverage
## SUB-PHASE DEVELOPMENT
**Workflow**: Plan Write Test Implement Make Test Pass Commit
### Sub-Phase Naming
Use decimal notation: **Phase X.Y** where X = major phase, Y = sub-phase number.
| Example | Scope |
|---------|-------|
| Phase 1.1 | Project setup, config, database |
| Phase 1.2 | Ingestion pipeline |
| Phase 1.3 | Query pipeline (3-step LLM workflow) |
| Phase 1.4 | Testing & polish |
| Phase 2.1 | Video upload backend |
| Phase 2.2 | ASR integration |
### Test-First Rule (MANDATORY)
Every sub-phase follows **test-driven delivery**:
1. **Write test first** Before writing implementation code, write the test that defines "done"
2. **Implement** Write the minimum code to make the test pass
3. **Run test** Verify test passes (both unit and acceptance where applicable)
4. **Commit** Only commit after tests pass. Never commit broken tests.
5. **Next sub-phase** Only start next sub-phase after current is committed
**Enforcement**:
- Each Implementation Task in a sub-phase plan must list its test file(s)
- Tests must be in the `backend/app/test/` or `frontend/src/test/` directory
- Pre-commit: `pytest` must pass for backend, `npm test` for frontend
### Sub-Phase Plan Template
Each sub-phase plan (stored in `.plans/`) must include:
1. **Objective** What this sub-phase delivers
2. **Test Files** List of test files to write BEFORE implementation
3. **Acceptance Criteria** List of behaviors that must work
4. **Acceptance Tests** `test_acceptance_<subphase>.py` file(s) with real environment
5. **Implementation Tasks** Atomic steps, each referencing its test file
### Acceptance Testing Rules
**Unit tests** (`test_phase*.py`) mocked, fast, CI-safe
**Acceptance tests** (`test_acceptance_*.py`) real environment, actual LLM/ASR calls
**Acceptance test requirements**:
- Run against real services (ChromaDB instance, actual LLM API, ASR if applicable)
- Name format: `test_acceptance_<subphase>_<feature>.py`
- Location: `backend/app/test/acceptance/`
- Use `pytest` markers: `@pytest.mark.acceptance` and `@pytest.mark.slow`
- Each acceptance test file must have docstring describing real environment setup
- Acceptance tests run manually before sub-phase completion, not in CI
**Example acceptance test**:
```python
"""Acceptance test: Phase 1 RAG query with real Qwen LLM.
Prerequisites:
- ChromaDB running (local or docker)
- .env configured with valid LLM_BASE_URL and LLM_API_KEY
- Test documents ingested via /api/v1/ingest
"""
import pytest
@pytest.mark.acceptance
@pytest.mark.slow
def test_query_with_real_llm():
"""Query should return bullet-point answer from actual LLM."""
# Real HTTP call to LLM provider
# Real ChromaDB retrieval
pass
```
**Sub-phase completion checklist**:
- [ ] All unit tests written BEFORE implementation
- [ ] All unit tests pass (`pytest app/test/test_phase*.py -v`)
- [ ] All acceptance tests pass (`pytest app/test/acceptance/ -v -m acceptance`)
- [ ] Code reviewed (self or peer)
- [ ] Sub-phase plan marked complete in `.plans/`
- [ ] Git commit with clear message referencing sub-phase (e.g., "feat: Phase 1.2 ingestion pipeline with tests")
## COMMANDS
```bash
# Dev
backend: uvicorn app.main:app --reload --port 8000
frontend: npm run dev
# Unit tests (mocked, CI-safe)
backend: cd backend && pytest app/test/test_phase*.py -v
# Acceptance tests (real LLM/ASR/ChromaDB)
backend: cd backend && pytest app/test/acceptance/ -v -m acceptance
# Prod
docker-compose up -d
./deploy.sh
```
## PLAN STORAGE
**All development plans** (including sub-plans, debug plans, and task breakdowns) **must be stored in `.plans/`**.
```
.plans/
├── development_plan.md # Main development plan (root-level)
├── phase1_backend_plan.md # Phase 1 backend tasks
├── phase1_frontend_plan.md # Phase 1 frontend tasks
├── phase2_backend_plan.md # Phase 2 backend tasks
├── phase2_frontend_plan.md # Phase 2 frontend tasks
├── debug_<date>_<issue>.md # Debug/diagnosis logs
└── _template.md # Plan template (optional)
```
**Rules**:
- Name format: `<purpose>_<optional_date>.md` (snake_case)
- Use `debug_` prefix for troubleshooting logs
- Root `development_plan.md` stays at root as canonical source
- Sub-plans reference root plan, never duplicate it
## NOTES
- No routing library specified single-page app likely sufficient
- No client state library specified `useState`/`useReducer` + TanStack Query
- WebSocket client not specified may need to expand `lib/api.ts`
- shadcn/ui components are copied, not imported as npm package
- Alibaba Cloud reference: https://modelstudio.console.alibabacloud.com/ap-southeast-1?switchAgent=101503&tab=doc&productCode=p_efm&switchUserType=3#/doc/?type=model&url=2989727