search_by_id 用于主键 id 检索。根据主键 id,搜索与其距离最近的 limit 个向量。
dense_weight 调节稠密向量在混合检索中的权重。参数 | 类型 | 是否必选 | 默认值 | 参数说明 |
|---|---|---|---|---|
id | int64 | 是 | 主键 id。 | |
filter | map | 否 | None | 过滤条件,详见标量过滤。
|
limit | int | 否 | 10 | 检索结果数量,最大5000个。 |
offset | int | 否 | 0 | 偏移量。仅分页场景下使用,不建议设置过大值,否则有深分页影响。默认值为0。设置值至少为0,语义和mysql的offset相同。 |
dense_weight | float | 否 | 0.5 | 对于标量过滤检索,dense_weight 用于控制稠密向量在检索中的权重。范围为[0.2,1]。仅在检索的索引为混合索引时有效。 |
output_fields | list | 否 | 过滤字段,指定要返回的标量或向量字段列表。
如果索引的距离方式为cosine,向量字段返回的向量是归一化后的向量。 | |
partition | string/int | 否 | "default" | 子索引名称,类型与 partition_by 的 field_type 一致,字段值对应 partition_by 的 field_value。
|
scale_k | float64 | 否 | 1 | 检索时的 ef 是索引参数 ef 的 scale_k 倍默认1,取值范围[0.1, 100],值越大检索结果越精确,但相应地会降低性能。 |
算子 | 算子说明 | 示例 |
|---|---|---|
must | 针对指定字段名生效,语义为必须在 [...] 之中,即 "must in"。 |
|
must_not | 针对指定字段名生效,语义为必须不在 [...] 之中,即 "must not in"。 |
|
range | 针对指定字段名生效,语义为必须在指定范围内。 |
|
range_out | 针对指定字段名生效,语义为必须在指定范围外。配置使用 |
|
and | 逻辑算子,针对逻辑查询需求,对多个条件取交集。 |
|
or | 逻辑算子,针对逻辑查询需求,对多个条件取并集。 |
|
运行示例前,请完成以下准备:
- 安装对应版本的 VikingDB Python SDK,将完整代码保存为
example.py后执行python3 example.py。- AK/SK 在火山引擎控制台“密钥管理”中创建;Host 和 Region 取自地域和访问域名。请按代码的参数或环境变量读取方式完成配置。
- Collection、Index 名称从 VikingDB 控制台对应资源详情获取;Task ID 使用创建或查询任务接口返回的 ID。请将
<YOUR_...>占位符替换为真实值。- 若示例包含 TOS 路径,请从 TOS 控制台确认桶名和对象路径,并替换为真实的
<YOUR_BUCKET_NAME>/<YOUR_TOS_PATH>。
请在火山引擎控制台的密钥管理页面获取 AK 和 SK,并将示例中的 <YOUR_AK>、<YOUR_SK> 等占位符替换为真实配置。
# 获取指定索引,程序初始化时调用即可,无需重复调用 from volcengine.viking_db.IndexClient import IndexClient index_client = IndexClient("<YOUR_COLLECTION_NAME>", "<YOUR_INDEX_NAME>", "api-vikingdb.volces.com", "cn-beijing", "<YOUR_AK>", "<YOUR_SK>", "http")
res = index_client.search_by_id("22", limit=2, output_fields=["doc_id", "like", "text_vector"], partition="default") # 异步调用 async def search_by_id(): index_client = IndexClient("<YOUR_COLLECTION_NAME>", "<YOUR_INDEX_NAME>", "api-vikingdb.volces.com", "cn-beijing", "<YOUR_AK>", "<YOUR_SK>", "http") res = await index_client.async_search_by_id("222") asyncio.run(search_by_id())
Python 调用执行上面的任务,返回 List 。Data 实例包含的属性如下表所示。
属性 | 说明 |
|---|---|
id | 主键 id。 |
fields | 请求返回中的 fields 字段,是具体的数据,字典类型。 |
score | 表示找到的向量和输入的向量的匹配程度。 |
id 支持 int64 或 string,具体类型必须与 Collection 主键字段一致。因此示例中的 "22" 适用于字符串主键;整数主键应传 22。