韬略问答Agent多跳策略的多轮检索模块
Project description
多跳策略的多轮检索模块
使用方法
from taolue_multi_retrival.multi_retrieve import efficient_rag
efficient_rag(search_rerank_func, query, top_k, libs, kb_ids, kb_file_ids)
search_rerank_func函数：
def search_rerank_func(query, top_k, libs, kb_ids, kb_file_ids)
参数说明
参数 类型 默认值 说明 search_rerank_func 函数 传入执行检索+精排操作的函数 query str 原始query libs List[str] ["es_chunk","es_summary","es_embedding","es_summary_embedding"] 一般情况下检索所有库，第二轮检索开始只检索es切片库、es摘要库，不检索向量库。 kb_ids List[int] / 哪个字段有值用哪个，Tuple结构：(知识库id, 文件id列表) kb_file_ids List[Tuple[int, List[int]]] / top_k int 10 每轮检索的数据量
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file taolue_multi_retrival-1.0.4.tar.gz.
File metadata
- Download URL: taolue_multi_retrival-1.0.4.tar.gz
- Upload date:
- Size: 11.3 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.10.15
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
340cb31b1859fc24f784620c7a1d5d42578723e53684f270820769095dea62ad
|
|
| MD5 |
01b6fe6224c4f7b94748debd8a68645c
|
|
| BLAKE2b-256 |
9274618686d258b56cdf97eb5f4b6a400a03843f5d01a95f079c9d6b1fa10776
|
File details
Details for the file taolue_multi_retrival-1.0.4-py3-none-any.whl.
File metadata
- Download URL: taolue_multi_retrival-1.0.4-py3-none-any.whl
- Upload date:
- Size: 12.2 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/6.1.0 CPython/3.10.15
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
64cb94df843e8fe8826fac2ab7f5f41eaf8de36dac6d7678ec950a81128cefd8
|
|
| MD5 |
b085c25a8533a6cc20a98c80b8c8116d
|
|
| BLAKE2b-256 |
151a0e2b0da06539aad9c7699c073c57192be5feba00a01d93e8cd4f69f226e1
|
