使用 SQL 进行全文搜索
与关注语义相似度的 向量搜索 不同,全文搜索允许你根据精确关键字搜索文档。在检索增强生成(RAG)场景中,你可以将全文搜索与向量搜索结合使用,以提升搜索质量。
TiDB 的全文搜索功能提供以下能力:
直接查询文本数据:你可以直接在任意 string 列上进行搜索,无需 embedding 过程。
支持多语言:无需指定语言即可获得高质量搜索。TiDB 的文本分析器支持同一张表中多种语言混合的文档,并会自动为每个文档选择最佳分析器。
按相关性排序:搜索结果可通过广泛采用的 BM25 排序 algorithm 按相关性排序。
完全兼容 SQL:所有 SQL 功能,如预过滤、后过滤、分组和 join,都可与全文搜索结合使用。
提示:
如需在 Python 中使用,请参见 使用 Python 进行全文搜索。
如需在 AI 应用中同时使用全文搜索和向量搜索,请参见 混合搜索。
快速开始
全文搜索仍处于早期阶段,我们正在持续向更多用户开放。目前,全文搜索仅在以下区域的 TiDB Cloud Starter 和 TiDB Cloud Essential 上可用:
- AWS:
Frankfurt (eu-central-1)和Singapore (ap-southeast-1)
在使用全文搜索前,请确保你的 TiDB Cloud Starter 集群已创建在支持的区域。如果还没有,请按照 创建 TiDB Cloud Starter 集群 进行创建。
要进行全文搜索,请按照以下步骤操作:
创建全文索引
要进行全文搜索,需要创建全文索引,它为高效搜索和排序提供必要的数据结构。全文索引既可以在新表上创建,也可以添加到已有表上。
创建带有全文索引的表:
CREATE TABLE stock_items(
id INT,
title TEXT,
FULLTEXT INDEX (title) WITH PARSER MULTILINGUAL
);
或为已有表添加全文索引:
CREATE TABLE stock_items(
id INT,
title TEXT
);
-- 你可以在这里插入一些数据。
-- 即使表中已有数据,也可以创建全文索引。
ALTER TABLE stock_items ADD FULLTEXT INDEX (title) WITH PARSER MULTILINGUAL ADD_COLUMNAR_REPLICA_ON_DEMAND;
WITH PARSER <PARSER_NAME> 子句中可用的 parser 包括:
STANDARD:速度快,适用于英文内容,通过空格和标点切分单词。MULTILINGUAL:支持多种语言,包括英文、中文、日文和韩文。
插入文本数据
向带有全文索引的表插入数据,与向其他表插入数据完全相同。
例如,你可以执行以下 SQL 语句插入多语言数据。TiDB 的多语言 parser 会自动处理文本。
INSERT INTO stock_items VALUES (1, "イヤホン bluetooth ワイヤレスイヤホン ");
INSERT INTO stock_items VALUES (2, "完全ワイヤレスイヤホン/ウルトラノイズキャンセリング 2.0 ");
INSERT INTO stock_items VALUES (3, "ワイヤレス ヘッドホン Bluetooth 5.3 65時間再生 ヘッドホン 40mm HD ");
INSERT INTO stock_items VALUES (4, "楽器用 オンイヤーヘッドホン 密閉型【国内正規品】");
INSERT INTO stock_items VALUES (5, "ワイヤレスイヤホン ハイブリッドANC搭載 40dBまでアクティブノイズキャンセル");
INSERT INTO stock_items VALUES (6, "Lightweight Bluetooth Earbuds with 48 Hours Playtime");
INSERT INTO stock_items VALUES (7, "True Wireless Noise Cancelling Earbuds - Compatible with Apple & Android, Built-in Microphone");
INSERT INTO stock_items VALUES (8, "In-Ear Earbud Headphones with Mic, Black");
INSERT INTO stock_items VALUES (9, "Wired Headphones, HD Bass Driven Audio, Lightweight Aluminum Wired in Ear Earbud Headphones");
INSERT INTO stock_items VALUES (10, "LED Light Bar, Music Sync RGB Light Bar, USB Ambient Lamp");
INSERT INTO stock_items VALUES (11, "无线消噪耳机-黑色 手势触控蓝牙降噪 主动降噪头戴式耳机(智能降噪 长久续航)");
INSERT INTO stock_items VALUES (12, "专业版USB7.1声道游戏耳机电竞耳麦头戴式电脑网课办公麦克风带线控");
INSERT INTO stock_items VALUES (13, "投影仪家用智能投影机便携卧室手机投影");
INSERT INTO stock_items VALUES (14, "无线蓝牙耳机超长续航42小时快速充电 流光金属耳机");
INSERT INTO stock_items VALUES (15, "皎月银 国家补贴 心率血氧监测 蓝牙通话 智能手表 男女表");
执行全文搜索
要执行全文搜索,可以使用 FTS_MATCH_WORD() function。
示例:搜索最相关的 10 个文档
SELECT * FROM stock_items
WHERE fts_match_word("bluetoothイヤホン", title)
ORDER BY fts_match_word("bluetoothイヤホン", title)
DESC LIMIT 10;
-- 结果按相关性排序,最相关的文档排在最前面。
+------+-----------------------------------------------------------------------------------------------------------+
| id | title |
+------+-----------------------------------------------------------------------------------------------------------+
| 1 | イヤホン bluetooth ワイヤレスイヤホン |
| 6 | Lightweight Bluetooth Earbuds with 48 Hours Playtime |
| 2 | 完全ワイヤレスイヤホン/ウルトラノイズキャンセリング 2.0 |
| 3 | ワイヤレス ヘッドホン Bluetooth 5.3 65時間再生 ヘッドホン 40mm HD |
| 5 | ワイヤレスイヤホン ハイブリッドANC搭載 40dBまでアクティブノイズキャンセル |
+------+-----------------------------------------------------------------------------------------------------------+
-- 尝试用另一种语言搜索:
SELECT * FROM stock_items
WHERE fts_match_word("蓝牙耳机", title)
ORDER BY fts_match_word("蓝牙耳机", title)
DESC LIMIT 10;
-- 结果按相关性排序,最相关的文档排在最前面。
+------+---------------------------------------------------------------------------------------------------------------+
| id | title |
+------+---------------------------------------------------------------------------------------------------------------+
| 14 | 无线蓝牙耳机超长续航42小时快速充电 流光金属耳机 |
| 11 | 无线消噪耳机-黑色 手势触控蓝牙降噪 主动降噪头戴式耳机(智能降噪 长久续航) |
| 15 | 皎月银 国家补贴 心率血氧监测 蓝牙通话 智能手表 男女表 |
+------+---------------------------------------------------------------------------------------------------------------+
示例:统计匹配用户查询的文档数量
SELECT COUNT(*) FROM stock_items
WHERE fts_match_word("bluetoothイヤホン", title);
+----------+
| COUNT(*) |
+----------+
| 5 |
+----------+
高级示例:与其他表 join 搜索结果
你可以将全文搜索与其他 SQL 功能(如 join 和子查询)结合使用。
假设你有一张 users 表和一张 tickets 表,并希望基于作者姓名的全文搜索结果查找其创建的工单:
CREATE TABLE users(
id INT,
name TEXT,
FULLTEXT INDEX (name) WITH PARSER STANDARD
);
INSERT INTO users VALUES (1, "Alice Smith");
INSERT INTO users VALUES (2, "Bob Johnson");
CREATE TABLE tickets(
id INT,
title TEXT,
author_id INT
);
INSERT INTO tickets VALUES (1, "Ticket 1", 1);
INSERT INTO tickets VALUES (2, "Ticket 2", 1);
INSERT INTO tickets VALUES (3, "Ticket 3", 2);
你可以使用子查询根据作者姓名查找匹配的用户 ID,然后在外层查询中使用这些 ID 搜索并 join 相关工单信息:
SELECT t.title AS TICKET_TITLE, u.id AS AUTHOR_ID, u.name AS AUTHOR_NAME FROM tickets t
LEFT JOIN users u ON t.author_id = u.id
WHERE t.author_id IN
(
SELECT id FROM users
WHERE fts_match_word("Alice", name)
);
+--------------+-----------+-------------+
| TICKET_TITLE | AUTHOR_ID | AUTHOR_NAME |
+--------------+-----------+-------------+
| Ticket 1 | 1 | Alice Smith |
| Ticket 2 | 1 | Alice Smith |
+--------------+-----------+-------------+
另请参阅
反馈与帮助
全文搜索仍处于早期阶段,开放区域有限。如果你希望在尚未开放的区域体验全文搜索,或有任何反馈与帮助需求,欢迎联系我们:
- 在 Discord 或 Slack 社区提问。
- 提交 TiDB Cloud 支持工单