簡介
在上一課中,您學習如何對文件進行分塊。但純向量搜尋有一個主要缺點:它只能按含義(語義)進行搜索,無法按屬性(創建日期、作者、文檔類型...)進行過濾。
範例: 使用者詢問「2025 年休假政策」。向量搜尋可能會返回 2023 年類似內容的政策 - 但年份錯誤!元資料過濾解決:
year == 2025 AND category == "HR"。
本文涵蓋了 3 種升級檢索的技術:
- 元資料 — 為每個區塊附加附加資訊
- 過濾 — 在搜尋之前/之後過濾區塊
- 混合搜尋——向量+關鍵字組合(BM25)
1. 元資料 — 將資訊附加到區塊
1.1 什麼是元資料?
向量儲存中的每個區塊由 3 個部分組成:
┌─────────────────────────────────────────┐
│ Chunk │
│ ├── content: "Nghỉ phép 15 ngày..." │
│ ├── embedding: [0.12, -0.34, ...] │ ← vector search dùng
│ └── metadata: { │ ← filtering dùng
│ source: "hr-policy.pdf", │
│ page: 5, │
│ year: 2025, │
│ department: "HR", │
│ author: "Nguyen Van A" │
│ } │
└─────────────────────────────────────────┘
1.2 自動擷取元數據
"""Gắn metadata khi chunk tài liệu"""
from langchain.text_splitter import RecursiveCharacterTextSplitter
from langchain_community.document_loaders import PyPDFLoader
from datetime import datetime
# Load PDF — tự động có metadata page number
loader = PyPDFLoader("hr-policy-2025.pdf")
pages = loader.load()
# Thêm metadata custom
for page in pages:
page.metadata.update({
"source_type": "pdf",
"department": "HR",
"year": 2025,
"language": "vi",
"last_updated": "2025-01-15",
})
# Chunk — metadata được kế thừa cho mỗi chunk
splitter = RecursiveCharacterTextSplitter(chunk_size=500, chunk_overlap=50)
chunks = splitter.split_documents(pages)
print(chunks[0].metadata)
# {'source': 'hr-policy-2025.pdf', 'page': 0,
# 'source_type': 'pdf', 'department': 'HR', 'year': 2025, ...}
1.3 元資料應該附加到什麼?
| 元資料欄位 | 範例 | 使用案例 |
|---|---|---|
source | “hr-policy.pdf” | 檢索來源 |
page | 5 | 用戶驗證 |
year / date | 2025 | 2025按時間過濾 |
category | 「人力資源」、「財務」 | 依部門篩選 |
language | “vi”,“en” | 多語言 RAG |
author | “阮文A” | 依作者篩選 |
chunk_index | 3 | 排序順序 |
doc_type | “政策”、“常見問題” | 文件分類 |
💡練習 1: 載入包含 5 個不同檔案(PDF、TXT、DOCX)的資料夾。自動附加元數據,包括:來源、文件類型、文件大小、建立日期。
2.元資料過濾
2.1 查詢時過濾
"""Filter metadata trong Chroma"""
from langchain_community.vectorstores import Chroma
from langchain_openai import OpenAIEmbeddings
# Index chunks (đã có metadata)
vectorstore = Chroma.from_documents(
chunks,
OpenAIEmbeddings(model="text-embedding-3-small"),
collection_name="company_docs"
)
# Search KHÔNG filter — trả về kết quả từ mọi phòng ban
results = vectorstore.similarity_search("nghỉ phép bao nhiêu ngày?", k=5)
# Search CÓ filter — chỉ tìm trong tài liệu HR năm 2025
results = vectorstore.similarity_search(
"nghỉ phép bao nhiêu ngày?",
k=5,
filter={"year": 2025, "department": "HR"} # Exact match
)
# Filter phức tạp: $and, $or, $gt, $lt, $in
results = vectorstore.similarity_search(
"chính sách lương",
k=5,
filter={
"$and": [
{"year": {"$gte": 2024}}, # Năm >= 2024
{"department": {"$in": ["HR", "Finance"]}}, # HR hoặc Finance
]
}
)
2.2 自查詢擷取器 — 從查詢自解析篩選器
"""Self-Query: AI tự tách query thành search + filter"""
from langchain.retrievers import SelfQueryRetriever
from langchain.chains.query_constructor.base import AttributeInfo
from langchain_openai import ChatOpenAI
# Mô tả metadata fields cho LLM hiểu
metadata_field_info = [
AttributeInfo(name="year", description="Năm ban hành", type="integer"),
AttributeInfo(name="department", description="Phòng ban: HR, Finance, IT", type="string"),
AttributeInfo(name="doc_type", description="Loại: policy, faq, guide", type="string"),
]
retriever = SelfQueryRetriever.from_llm(
llm=ChatOpenAI(model="gpt-4o-mini", temperature=0),
vectorstore=vectorstore,
document_contents="Tài liệu nội bộ công ty về chính sách và quy trình",
metadata_field_info=metadata_field_info,
)
# User hỏi: "Chính sách HR năm 2025 về nghỉ phép"
# → LLM tự parse:
# search_query = "chính sách nghỉ phép"
# filter = {"year": 2025, "department": "HR"}
results = retriever.invoke("Chính sách HR năm 2025 về nghỉ phép")
Flow:
User query: "Chính sách HR năm 2025 về nghỉ phép"
│
┌─────────┴─────────┐
│ Self-Query LLM │
│ (parse intent) │
└─────────┬─────────┘
│
┌───────────────┼───────────────┐
│ │ │
search_query filter_year filter_dept
"nghỉ phép" 2025 "HR"
│ │ │
└───────────────┼───────────────┘
│
┌─────────┴─────────┐
│ Vector Store │
│ (search+filter) │
└─────────┬─────────┘
│
Filtered results
💡練習 2: 為至少包含 3 個元資料欄位的文件集建立自查詢檢索器。用 5 個自然問題進行測試。檢查LLM解析過濾器是否正確。
3. 混合搜尋 — 向量 + 關鍵字
3.1 純向量搜尋的問題
Query: "Nghị định 168/2024/NĐ-CP"
Vector search: tìm theo ý nghĩa → có thể trả về
Nghị định 150/2023 (nội dung tương tự nhưng SAI số!)
Keyword search (BM25): tìm đúng "168/2024/NĐ-CP" → ĐÚNG
→ Kết hợp cả 2 = Hybrid Search
| 搜尋類型 | 強 | 弱 |
|---|---|---|
| 向量 | 理解含義、同義詞、上下文 | 需要精確匹配時錯誤(代碼、編號、名稱) |
| 關鍵字 (BM25) | 精確匹配、代碼、專有名詞 | 不理解同義詞、上下文 |
| 混合 | 結合兩者優勢 | 需要調整體重 |
3.2 實作混合搜尋
"""Hybrid search với BM25 + Vector"""
from langchain_community.retrievers import BM25Retriever
from langchain.retrievers import EnsembleRetriever
from langchain_community.vectorstores import Chroma
from langchain_openai import OpenAIEmbeddings
# Chuẩn bị documents (đã chunk)
# chunks = [Document(...), Document(...), ...]
# 1. Vector retriever
vectorstore = Chroma.from_documents(chunks, OpenAIEmbeddings())
vector_retriever = vectorstore.as_retriever(search_kwargs={"k": 5})
# 2. BM25 retriever (keyword-based)
bm25_retriever = BM25Retriever.from_documents(chunks, k=5)
# 3. Ensemble (hybrid) — weight 50/50
hybrid_retriever = EnsembleRetriever(
retrievers=[vector_retriever, bm25_retriever],
weights=[0.5, 0.5], # Tùy chỉnh: 0.7/0.3 nếu ưu tiên vector
)
# Query
results = hybrid_retriever.invoke("Nghị định 168/2024/NĐ-CP")
3.3 倒數秩融合(RRF)
當合併 2 個檢索器的結果時,需要 merge + rating 方法:
Vector results: BM25 results:
1. Doc A (score 0.95) 1. Doc C (score 8.2)
2. Doc B (score 0.88) 2. Doc A (score 7.1)
3. Doc C (score 0.82) 3. Doc D (score 6.5)
RRF formula: score(d) = Σ 1/(k + rank(d)) (k=60 default)
Doc A: 1/(60+1) + 1/(60+2) = 0.0164 + 0.0161 = 0.0325 ← Top 1!
Doc C: 1/(60+3) + 1/(60+1) = 0.0159 + 0.0164 = 0.0323 ← Top 2
Doc B: 1/(60+2) + 0 = 0.0161 ← Top 3
Doc D: 0 + 1/(60+3) = 0.0159 ← Top 4
Doc A 出現在兩隻獵犬身上 → 最高排名!
3.4 Pinecone 混合搜尋(生產就緒)
"""Pinecone native hybrid search — sparse + dense vectors"""
from pinecone import Pinecone
from pinecone_text.sparse import BM25Encoder
# Sparse encoder (BM25)
bm25 = BM25Encoder()
bm25.fit([chunk.page_content for chunk in chunks])
# Dense encoder (embedding)
from langchain_openai import OpenAIEmbeddings
embeddings = OpenAIEmbeddings(model="text-embedding-3-small")
# Index với cả 2 loại vector
pc = Pinecone(api_key="your-key")
index = pc.Index("hybrid-rag")
for chunk in chunks:
dense = embeddings.embed_query(chunk.page_content)
sparse = bm25.encode_queries(chunk.page_content)
index.upsert(vectors=[{
"id": chunk.metadata.get("id", str(hash(chunk.page_content))),
"values": dense, # Dense vector
"sparse_values": sparse, # Sparse vector (BM25)
"metadata": chunk.metadata
}])
# Query hybrid
query = "Nghị định 168/2024"
results = index.query(
vector=embeddings.embed_query(query),
sparse_vector=bm25.encode_queries(query),
top_k=5,
alpha=0.5, # 0=pure sparse, 1=pure dense, 0.5=hybrid
)
💡練習 3: 對一組文件實施混合搜尋。比較結果:(a) 僅載體,(b) 僅 BM25,(c) 混合。使用 10 個測驗題,記錄每種類型的準確性。
4. 調整混合權重
4.1 何時優先考慮向量與關鍵字?
| 使用案例 | 向量重量 | BM25重量 | 原因 |
|---|---|---|---|
| 常見問題/一般問答 | 0.7 | 0.7 0.3 | 0.3使用者以多種不同方式詢問 |
| 法律/法規 | 0.3 | 0.3 0.7 | 0.7需要精確匹配規則代碼 |
| 技術文件 | 0.5 | 0.5 0.5 | 0.5需要關鍵字和語意 |
| 多語言 | 0.8 | 0.8 0.2 | 0.2 Vector更適合跨語言 |
| 程式碼文檔 | 0.4 | 0.4 0.6 | 0.6函數名稱=關鍵字 |
4.2 自動調整權重
"""Benchmark hybrid weights trên golden test set"""
test_queries = [
{"q": "nghỉ phép bao nhiêu ngày", "expected_doc": "hr-policy.pdf"},
{"q": "Nghị định 168/2024", "expected_doc": "legal/nd168.pdf"},
# ... 10+ câu test
]
weight_configs = [
(0.3, 0.7), (0.4, 0.6), (0.5, 0.5),
(0.6, 0.4), (0.7, 0.3), (0.8, 0.2),
]
best_config = None
best_accuracy = 0
for vec_w, bm25_w in weight_configs:
hybrid = EnsembleRetriever(
retrievers=[vector_retriever, bm25_retriever],
weights=[vec_w, bm25_w],
)
correct = 0
for test in test_queries:
results = hybrid.invoke(test["q"])
sources = [r.metadata["source"] for r in results[:3]]
if test["expected_doc"] in sources:
correct += 1
accuracy = correct / len(test_queries)
print(f"Vector={vec_w}, BM25={bm25_w}: {accuracy:.0%}")
if accuracy > best_accuracy:
best_accuracy = accuracy
best_config = (vec_w, bm25_w)
print(f"\nBest: Vector={best_config[0]}, BM25={best_config[1]} ({best_accuracy:.0%})")
總結
| 概念 | 記住 |
|---|---|
| 元資料 | 附加到區塊的附加資訊(來源、年份、類別...) |
| 過濾 | 在搜尋前/之後按元資料過濾區塊 |
| 自助查詢 | LLM自動將問題解析為搜尋+過濾 |
| BM25 | 關鍵字搜索,精準匹配強 |
| 混合搜尋 | Vector+BM25,結合兩者的優點 |
| RRF | 倒數排名融合 — 合併 2 個檢索器結果 |
| 重量調整 | 黃金測試集基準選擇比例 |
一般練習
- ✅ 完成3個小練習(1,2,3)
- 完整管道: 載入 10 多個文件 → 附加完整元資料 → 索引到 Chroma → 實作混合搜尋 → 在 20 個測試問題上比較準確度向量與混合。
- **自查詢+混合:**將自查詢擷取器與混合搜尋結合。用戶詢問「2025年人力資源政策關於薪資」→自篩選部門+年份+混合搜尋內容。
- 儀表板: 建立 Streamlit 應用程式:上傳文件 → 附加元資料 → 使用 UI 篩選器搜尋(下拉清單選擇年份、部門...)。
下一篇文章: 查詢轉換 - HyDE、多重查詢、Step-Back - 將 1 個問題轉換為多個變體,以實現更準確的搜尋。