dhammawatthumpra commited on
Commit
f50b0fd
·
1 Parent(s): 7805140

chore: untrack and ignore SEARCH_ARCHITECTURE_V3.md spec

Browse files
Files changed (2) hide show
  1. .gitignore +1 -0
  2. SEARCH_ARCHITECTURE_V3.md +0 -499
.gitignore CHANGED
@@ -97,6 +97,7 @@ webui_tool_tipitaka*.py
97
  Dhamma-LM*.md
98
  TIPITAKA_WEB_ARCHITECTURE*.md
99
  Tipitaka-Web-Application-Tech-Stack*.md
 
100
 
101
  # Notebooks
102
  *.ipynb
 
97
  Dhamma-LM*.md
98
  TIPITAKA_WEB_ARCHITECTURE*.md
99
  Tipitaka-Web-Application-Tech-Stack*.md
100
+ SEARCH_ARCHITECTURE_V3.md
101
 
102
  # Notebooks
103
  *.ipynb
SEARCH_ARCHITECTURE_V3.md DELETED
@@ -1,499 +0,0 @@
1
- # 📖 Search Architecture Specification (V3)
2
- ## Next-Gen Hybrid Search for Tipitaka Web Application
3
-
4
- > **Version:** 3.3.0
5
- > **Updated:** 2026-05-26
6
- > **Status:** Revised — ยืนยันชื่อ model จาก HuggingFace และ Ollama จริง
7
-
8
- เอกสารนี้ระบุการออกแบบสถาปัตยกรรมการค้นหา V3 ที่ผสมผสาน Query Understanding ด้วย Local LLM (Qwen3.5-0.8B) เข้ากับ Hybrid Search (FTS5 + Qdrant) และ Reranking เพื่อแก้ปัญหาการค้นหาด้วยภาษาพูดหรือความจำเลือนลาง
9
-
10
- ---
11
-
12
- ## 0. Model Name Reference (ยืนยันจาก Official Source)
13
-
14
- | Platform | ชื่อ model ที่ถูกต้อง |
15
- |----------|----------------------|
16
- | HuggingFace | `Qwen/Qwen3.5-0.8B` |
17
- | Ollama | `qwen3.5:0.8b` |
18
-
19
- > **หมายเหตุ:** ไม่มี `-Instruct` ต่อท้ายบน HuggingFace
20
- > `Qwen/Qwen3.5-0.8B` คือ Instruct version (ใช้งานได้เลย)
21
- > `Qwen/Qwen3.5-0.8B-Base` คือ Base version (สำหรับ fine-tune เท่านั้น)
22
-
23
- ---
24
-
25
- ## 1. System Dataflow
26
-
27
- ```
28
- [ User Natural Query (ภาษาพูด) ]
29
-
30
-
31
- [ 1. Query Transformation ] ◄── SQLite Query Cache (exact match, normalize ก่อน)
32
- Qwen3.5-0.8B
33
- Local: via Ollama HTTP API (ollama pull qwen3.5:0.8b)
34
- HF Space: via Transformers in-process (Lazy loading เหมือน Jina-v5)
35
- Output: {"fts_queries": [...], "vector_queries": [...]}
36
-
37
- ▼ (Fallback: original query ถ้า Qwen ล้มเหลว)
38
-
39
- ├─────────────────────────────────────────┐
40
- ▼ asyncio.gather (parallel จริง) ▼
41
- [ 2A. Lexical Search (FTS5) ] [ 2B. Semantic Search (Qdrant) ]
42
- SQLite pages_fts jina-embeddings-v5 (1024d)
43
- asyncio.gather(*fts_tasks) Qdrant batch search
44
- │ │
45
- └──────────────────┬──────────────────────┘
46
-
47
- [ 3. RRF Score Fusion ]
48
- Deduplicate: (volume_id, page_number)
49
- RRF(d) = Σ 1/(k=60 + rank)
50
- Top 30 candidates
51
-
52
-
53
- [ 4. ONNX Reranking ]
54
- jina-reranker-v2-base-multilingual
55
- Re-score vs original user query (ไม่ใช่ transformed)
56
-
57
-
58
- [ 5. Post-Process & Highlight ]
59
- Highlight keywords จาก fts_queries (ไม่ใช่ original query)
60
-
61
- ┌──────────────────┴──────────────────┐
62
- ▼ ▼
63
- [ UI Search Page ] [ AI Assistant (RAG) ]
64
- (Paginated Reader Links) (DeepSeek / Gemma Generation)
65
- ```
66
-
67
- ---
68
-
69
- ## 2. Layer Specifications
70
-
71
- ### Layer 1: Query Transformation — Dual Runtime
72
-
73
- | รายการ | Local (Ollama) | HF Space (Transformers) |
74
- |--------|---------------|------------------------|
75
- | **Model** | `qwen3.5:0.8b` | `Qwen/Qwen3.5-0.8B` |
76
- | **Runtime** | Ollama HTTP API (localhost) | HuggingFace Transformers in-process |
77
- | **Load time** | Ollama daemon (always running) | Lazy loading (โหลดเมื่อเรียกใช้ครั้งแรก) |
78
- | **dtype** | Ollama จัดการเอง | `torch.float16` |
79
- | **device** | CPU (Ollama) | `cpu` |
80
- | **RAM** | ~0.8 GB (Ollama process) | ~0.8 GB (in-process) |
81
- | **Latency** | ~1–3s cache miss | ~2–4s cache miss |
82
- | **Max output tokens** | 80 tokens | 80 tokens |
83
- | **trust_remote_code** | ไม่จำเป็น | `True` (DeltaNet) |
84
-
85
- **Runtime detection — environment variable:**
86
- ```bash
87
- # .env local
88
- QWEN_RUNTIME=ollama
89
- QWEN_OLLAMA_URL=http://localhost:11434
90
- QWEN_OLLAMA_MODEL=qwen3.5:0.8b
91
-
92
- # .env HF Space (Secrets)
93
- QWEN_RUNTIME=transformers
94
- QWEN_MODEL_ID=Qwen/Qwen3.5-0.8B
95
- ```
96
-
97
- **Output JSON format:**
98
- ```json
99
- {
100
- "fts_queries": [
101
- "วิสาขา ภิกษุณี สนทนา",
102
- "นางวิสาขา สงฆ์ ภาษิต"
103
- ],
104
- "vector_queries": [
105
- "นางวิสาขาพูดคุยกับภิกษุณีสงฆ์เรื่องธรรมะ",
106
- "การสนทนาธรรมระ��ว่างอุบาสิกาและภิกษุณี"
107
- ]
108
- }
109
- ```
110
-
111
- **ความต่างของสองช่อง (สำคัญ):**
112
- - `fts_queries` → keyword-style สั้น ใช้ AND/OR logic ใน FTS5
113
- - `vector_queries` → natural sentence ยาวกว่า ให้ embedding เข้าใจ context
114
-
115
- **System Prompt:**
116
- ```
117
- คุณคือผู้เชี่ยวชาญพระไตรปิฎกฉบับมหาจุฬา 45 เล่ม
118
- หน้าที่: แปลงคำค้นของผู้ใช้เป็น JSON เพื่อค้นหาในฐานข้อมูล
119
-
120
- กฎ:
121
- 1. ตอบด้วย JSON เท่านั้น ห้ามมีข้อความอื่น
122
- 2. fts_queries: คีย์เวิร์ดสั้น 2-5 คำ รวมคำบาลี/ไวพจน์ จำกัด 2 รายการ
123
- 3. vector_queries: ประโยคสมบูรณ์ความหมายชัดเจน จำกัด 2 รายการ
124
- 4. ห้ามแต่งเนื้อหาที่ไม่มีในพระไตรปิฎก
125
-
126
- ตัวอย่าง:
127
- Input: "นางวิสาขาคุยกับภิกษุณี"
128
- Output: {"fts_queries":["วิสาขา ภิกษุณี","นางวิสาขา สงฆ์ ภาษิต"],"vector_queries":["นางวิสาขาสนทนากับภิกษุณีสงฆ์","การพูดคุยธรรมะระหว่างอุบาสิกาและภิกษุณี"]}
129
-
130
- Input: "พระพุทธเจ้าเปรียบจิตกับน้ำขุ่น"
131
- Output: {"fts_queries":["จิต อุทก สมาธิ","จิต น้ำ ขุ่น ใส"],"vector_queries":["พระพุทธเจ้าอุปมาจิตเหมือนน้ำที่ขุ่นและใส","สมาธิทำให้จิตผ่องใสดุจน้ำนิ่ง"]}
132
- ```
133
-
134
- ---
135
-
136
- ### Layer 2: Parallel Retrieval
137
-
138
- FTS5 และ Vector ทำงานพร้อมกันด้วย `asyncio.gather` จริงๆ ไม่ใช่ sequential loop
139
-
140
- **จำกัด queries:** สูงสุด 2 fts + 2 vector = 4 tasks parallel ต่อ 1 request
141
-
142
- ---
143
-
144
- ### Layer 3: RRF Score Fusion
145
-
146
- $$RRF\_Score(d) = \sum_{m \in M} \frac{1}{k + r_m(d)}$$
147
-
148
- - k = 60 (default มาตรฐาน)
149
- - Dedup key: `(volume_id, page_number)`
150
- - Top 30 candidates ส่งต่อ reranker
151
-
152
- ---
153
-
154
- ### Layer 4: ONNX Reranking
155
-
156
- - Model: `jina-reranker-v2-base-multilingual` (ONNX CPU)
157
- - **Re-score กับ original user query เสมอ** — ไม่ใช่ transformed query
158
- - Fallback: ถ้า reranker ไม่พร้อม ใช้ RRF score เดิม
159
-
160
- ---
161
-
162
- ### Layer 5: Highlight
163
-
164
- - ใช้ keywords จาก `fts_queries` (transformed) ไม่ใช่ original query
165
- - ส่งเป็น metadata ไปยัง frontend สำหรับ `<mark>` tag
166
-
167
- ---
168
-
169
- ## 3. RAM Budget
170
-
171
- ### Local (RTX 5070 laptop 8GB VRAM / 32GB RAM)
172
-
173
- | Component | RAM |
174
- |-----------|-----|
175
- | SQLite in-memory | ~238 MB |
176
- | Qdrant Embedded | ~1.2–1.5 GB |
177
- | Jina-v5 (ST in-process) | ~1.2 GB |
178
- | ONNX Reranker | ~500 MB |
179
- | Qwen3.5-0.8B (Ollama — แยก process) | ~0.8 GB |
180
- | FastAPI + Python overhead | ~300 MB |
181
- | **รวม** | **~4.3–4.5 GB** ✅ |
182
-
183
- ### HF Space (CPU Basic = 16GB)
184
-
185
- | Component | RAM |
186
- |-----------|-----|
187
- | SQLite in-memory | ~238 MB |
188
- | Qdrant Embedded | ~1.2–1.5 GB |
189
- | Jina-v5 (ST in-process) | ~1.2 GB |
190
- | ONNX Reranker | ~500 MB |
191
- | Qwen3.5-0.8B (float16, in-process) | ~800 MB |
192
- | FastAPI + Python + Ubuntu OS | ~800 MB |
193
- | **รวม** | **~4.8–5.0 GB** ✅ |
194
- | **Headroom** | **~11 GB** |
195
-
196
- ---
197
-
198
- ## 4. Latency Profile
199
-
200
- | สถานการณ์ | Local (Ollama) | HF Space (Transformers) |
201
- |-----------|---------------|------------------------|
202
- | Cache hit | ~50ms | ~50ms |
203
- | Qwen transform | ~1–3s | ~2–4s |
204
- | FTS5 + Qdrant parallel | ~150ms | ~200ms |
205
- | RRF fusion | ~10ms | ~10ms |
206
- | ONNX rerank | ~200ms | ~300ms |
207
- | **รวม cache miss** | **~2–4s** | **~3–5s** |
208
- | **รวม cache hit** | **~400ms** | **~600ms** |
209
-
210
- ---
211
-
212
- ## 5. Implementation Blueprint
213
-
214
- ### 5.1 QueryTransformService — รองรับทั้งสอง Runtime & Lazy Loading
215
-
216
- ```python
217
- # app/services/query_transform_service.py
218
- import os
219
- import json
220
- import re
221
- import unicodedata
222
- from typing import Optional
223
-
224
- FALLBACK = lambda q: {"fts_queries": [q], "vector_queries": [q]}
225
-
226
- SYSTEM_PROMPT = """คุณคือผู้เชี่ยวชาญพระไตรปิฎกฉบับมหาจุฬา 45 เล่ม
227
- ตอบด้วย JSON เท่านั้น รูปแบบ: {"fts_queries":[...],"vector_queries":[...]}
228
- แต่ละช่องมีได้สูงสุด 2 รายการ ห้ามมีข้อความอื่นนอกจาก JSON"""
229
-
230
- class QueryTransformService:
231
- def __init__(self, db):
232
- self.db = db
233
-
234
- # ตั้งค่าพารามิเตอร์และ Runtime จากสภาพแวดล้อม
235
- self.runtime = os.getenv("QWEN_RUNTIME", "transformers")
236
- self.ollama_url = os.getenv("QWEN_OLLAMA_URL", "http://localhost:11434")
237
- self.ollama_model = os.getenv("QWEN_OLLAMA_MODEL", "qwen3.5:0.8b")
238
- self.model_id = os.getenv("QWEN_MODEL_ID", "Qwen/Qwen3.5-0.8B")
239
-
240
- # ตัวแปรสำหรับ Lazy Loading เมื่อใช้ transformers ในเครือข่าย HF Space
241
- self._tokenizer = None
242
- self._model = None
243
-
244
- def _normalize_key(self, query: str) -> str:
245
- # Normalize unicode to NFC
246
- q = unicodedata.normalize("NFC", query.strip().lower())
247
- # Strip out punctuation and symbols except space
248
- q = re.sub(r"[^\w\s\u0e00-\u0e7f]", "", q)
249
- # Compress multiple spaces
250
- return re.sub(r"\s+", " ", q).strip()
251
-
252
- async def transform(self, query: str) -> dict:
253
- key = self._normalize_key(query)
254
-
255
- # 1. ตรวจ cache ก่อนเสมอ
256
- cached = self.db.get_query_cache(key)
257
- if cached:
258
- return cached
259
-
260
- # 2. เรียก Qwen ตาม runtime
261
- try:
262
- if self.runtime == "ollama":
263
- result = await self._call_ollama(query)
264
- else:
265
- result = await self._call_transformers(query)
266
- self.db.set_query_cache(key, result)
267
- return result
268
- except Exception:
269
- return FALLBACK(query)
270
-
271
- async def _call_ollama(self, query: str) -> dict:
272
- """Local: ยิง Ollama HTTP API"""
273
- import httpx
274
- url = f"{self.ollama_url}/api/chat"
275
- payload = {
276
- "model": self.ollama_model, # qwen3.5:0.8b
277
- "messages": [
278
- {"role": "system", "content": SYSTEM_PROMPT},
279
- {"role": "user", "content": query}
280
- ],
281
- "stream": False,
282
- "options": {"num_predict": 80, "temperature": 0}
283
- }
284
- async with httpx.AsyncClient(timeout=10.0) as client:
285
- resp = await client.post(url, json=payload)
286
- raw = resp.json()["message"]["content"].strip()
287
- return self._parse_json(raw, query)
288
-
289
- async def _call_transformers(self, query: str) -> dict:
290
- """HF Space: Lazy Loading ในแอปเพื่อไม่ให้ Startup บล็อคและเกิด Timeout"""
291
- import torch
292
- from transformers import AutoTokenizer, AutoModelForCausalLM
293
-
294
- # โหลดโมเดลในหน่วยความจำเฉพาะเมื่อถูกใช้งานจริงครั้งแรก (Lazy Loading)
295
- if self._model is None or self._tokenizer is None:
296
- self._tokenizer = AutoTokenizer.from_pretrained(
297
- self.model_id,
298
- trust_remote_code=True
299
- )
300
- # โหลดด้วย float32 หรือ bfloat16 เพื่อความปลอดภัยบน CPU ของ HF Space
301
- self._model = AutoModelForCausalLM.from_pretrained(
302
- self.model_id,
303
- torch_dtype=torch.float32,
304
- device_map="cpu",
305
- trust_remote_code=True
306
- )
307
- self._model.eval()
308
-
309
- messages = [
310
- {"role": "system", "content": SYSTEM_PROMPT},
311
- {"role": "user", "content": query}
312
- ]
313
- text = self._tokenizer.apply_chat_template(
314
- messages, tokenize=False, add_generation_prompt=True
315
- )
316
- inputs = self._tokenizer(text, return_tensors="pt")
317
-
318
- with torch.no_grad():
319
- outputs = self._model.generate(
320
- **inputs,
321
- max_new_tokens=80,
322
- do_sample=False,
323
- temperature=None,
324
- top_p=None,
325
- pad_token_id=self._tokenizer.eos_token_id
326
- )
327
- raw = self._tokenizer.decode(
328
- outputs[0][inputs["input_ids"].shape[1]:],
329
- skip_special_tokens=True
330
- ).strip()
331
- return self._parse_json(raw, query)
332
-
333
- def _parse_json(self, raw: str, original: str) -> dict:
334
- try:
335
- # ค้นหาวงเล็บปีกกาตัวแรกและตัวสุดท้ายเพื่อสกัดเอาเฉพาะส่วน JSON
336
- start_idx = raw.find('{')
337
- end_idx = raw.rfind('}')
338
- if start_idx != -1 and end_idx != -1:
339
- json_str = raw[start_idx:end_idx+1]
340
- data = json.loads(json_str)
341
- if "fts_queries" in data and "vector_queries" in data:
342
- fts = [str(x).strip() for x in data["fts_queries"] if x][:2]
343
- vec = [str(x).strip() for x in data["vector_queries"] if x][:2]
344
- return {"fts_queries": fts, "vector_queries": vec}
345
- except Exception:
346
- pass
347
- return FALLBACK(original)
348
- ```
349
-
350
- ---
351
-
352
- ### 5.2 SQLite Cache Schema
353
-
354
- ```sql
355
- CREATE TABLE IF NOT EXISTS query_cache (
356
- cache_key TEXT PRIMARY KEY,
357
- result_json TEXT NOT NULL,
358
- hit_count INTEGER DEFAULT 1,
359
- created_at DATETIME DEFAULT CURRENT_TIMESTAMP,
360
- last_used DATETIME DEFAULT CURRENT_TIMESTAMP
361
- );
362
-
363
- CREATE INDEX IF NOT EXISTS idx_query_cache_last_used
364
- ON query_cache(last_used);
365
- ```
366
-
367
- ```python
368
- # app/database/sqlite_db.py (เพิ่ม methods)
369
- def get_query_cache(self, key: str) -> Optional[dict]:
370
- row = self.conn.execute(
371
- "SELECT result_json FROM query_cache WHERE cache_key = ?", (key,)
372
- ).fetchone()
373
- if row:
374
- self.conn.execute(
375
- """UPDATE query_cache
376
- SET hit_count = hit_count + 1, last_used = CURRENT_TIMESTAMP
377
- WHERE cache_key = ?""", (key,)
378
- )
379
- return json.loads(row[0])
380
- return None
381
-
382
- def set_query_cache(self, key: str, result: dict) -> None:
383
- self.conn.execute(
384
- """INSERT INTO query_cache (cache_key, result_json) VALUES (?, ?)
385
- ON CONFLICT(cache_key) DO UPDATE SET
386
- result_json = excluded.result_json,
387
- last_used = CURRENT_TIMESTAMP""",
388
- (key, json.dumps(result, ensure_ascii=False))
389
- )
390
- self.conn.commit()
391
-
392
- def evict_query_cache(self, max_entries: int = 10000) -> None:
393
- count = self.conn.execute(
394
- "SELECT COUNT(*) FROM query_cache"
395
- ).fetchone()[0]
396
- if count > max_entries:
397
- self.conn.execute("""
398
- DELETE FROM query_cache WHERE cache_key IN (
399
- SELECT cache_key FROM query_cache
400
- ORDER BY last_used ASC LIMIT ?
401
- )
402
- """, (count - max_entries,))
403
- self.conn.commit()
404
- ```
405
-
406
- ---
407
-
408
- ### 5.3 SearchService — asyncio.gather จริง
409
-
410
- ```python
411
- # app/services/search_service.py
412
- import asyncio
413
-
414
- class SearchService:
415
- def __init__(self, db, rag_service=None, query_transform_service=None):
416
- self.db = db
417
- self.rag_service = rag_service
418
- self.qts = query_transform_service
419
-
420
- async def search_hybrid(self, query: str, limit: int = 10, offset: int = 0) -> dict:
421
- # 1. Transform (cache หรือ Qwen)
422
- transformed = await self.qts.transform(query)
423
- fts_queries = transformed["fts_queries"]
424
- vector_queries = transformed["vector_queries"]
425
-
426
- # 2. Parallel — asyncio.gather ทั้งหมดจริงๆ
427
- fts_tasks = [self._get_fts_results(fq) for fq in fts_queries]
428
- vector_tasks = [self._get_vector_results(vq) for vq in vector_queries]
429
-
430
- fts_results_list, vector_results_list = await asyncio.gather(
431
- asyncio.gather(*fts_tasks),
432
- asyncio.gather(*vector_tasks)
433
- )
434
-
435
- # 3. RRF Fusion
436
- rrf_scores, candidate_data = {}, {}
437
- k = 60
438
-
439
- for results in fts_results_list:
440
- for rank, item in enumerate(results):
441
- key = (item["volume_id"], item["page_number"])
442
- rrf_scores[key] = rrf_scores.get(key, 0.0) + 1.0 / (k + rank + 1)
443
- candidate_data.setdefault(key, item)
444
-
445
- for results in vector_results_list:
446
- for rank, item in enumerate(results):
447
- key = (item["volume_id"], item["page_number"])
448
- rrf_scores[key] = rrf_scores.get(key, 0.0) + 1.0 / (k + rank + 1)
449
- candidate_data.setdefault(key, item)
450
-
451
- top_keys = sorted(rrf_scores, key=rrf_scores.get, reverse=True)[:30]
452
- top_candidates = [candidate_data[k] for k in top_keys]
453
-
454
- # 4. Rerank กับ original query เสมอ
455
- if (self.rag_service and
456
- self.rag_service.reranker and
457
- self.rag_service.reranker != "error"):
458
- final = await self._rerank_candidates(query, top_candidates)
459
- else:
460
- final = top_candidates
461
-
462
- # 5. Format + highlight ด้วย fts_queries keywords
463
- return self._format_response(final, limit, offset, fts_queries)
464
- ```
465
-
466
- ---
467
-
468
- ## 6. Requirements
469
-
470
- ```
471
- # requirements.txt (HF Space)
472
- transformers>=4.51.0 # minimum สำหรับ Qwen3.5 DeltaNet
473
- torch>=2.2.0
474
- accelerate>=0.27.0
475
- httpx>=0.27.0 # สำหรับ Ollama HTTP client (local)
476
- ```
477
-
478
- ```bash
479
- # Local: ติดตั้ง Ollama และ pull model
480
- ollama pull qwen3.5:0.8b
481
- ```
482
-
483
- ---
484
-
485
- ## 7. ข้อจำกัดที่รับรู้แล้ว (Phase 1)
486
-
487
- | ข้อจำกัด | ผลกระทบ | แนวทาง Phase 2 |
488
- |---------|---------|----------------|
489
- | Cache เป็น exact match | query ต่างกันเล็กน้อยไม่ hit | Levenshtein fuzzy match |
490
- | Transformers บน CPU | latency ~2–4s | GPU inference บน VPS |
491
- | ใช้ base Instruct ยังไม่ fine-tune | transformation ยังไม่ optimal | fine-tune ด้วย dataset ที่กำลังสร้าง |
492
- | Cold start HF Space เพิ่ม ~15–20s | restart ช้าขึ้น | Persistent Storage หรือ VPS |
493
- | Ollama local ต้องรัน daemon ก่อน | dev ต้อง start Ollama เอง | docker-compose ใน local setup |
494
-
495
- ---
496
-
497
- *Local: QWEN_RUNTIME=ollama (qwen3.5:0.8b)*
498
- *HF Space: QWEN_RUNTIME=transformers (Qwen/Qwen3.5-0.8B)*
499
- *Phase 2 migration path อ้างอิง Tipitaka-Web-Application-Tech-Stack-Phase1-HF.md*