检索知识库
Milvus 中已经存储了文本块向量,现在可以进行向量查询了。
以下函数创建了 1 个查询 pipeline。注意,这是本教程中最为关键的一个步骤!
ops.ann_search.osschat_milvus(host=MILVUS_HOST, port=MILVUS_PORT, **{'metric_type': 'IP', 'limit': 3, 'output_fields': ['text']}))
OSSChat_milvus(https://towhee.io/ann-search/osschat-milvus) 查询 Milvus 向量数据库中与查询文本相匹配的文档片段。
以下为整个查询 pipeline 代码:
pipe_search = (pipe.input('collection_name', 'query').map('query', 'query_vec', ops.sentence_embedding.sbert(model_name=EMBED_MODEL)).map('query_vec', 'query_vec', ops.np_normalize()).map((&