|
|
@ -1,4 +1,4 @@ |
|
|
|
# ANN Search Operator: Milvus |
|
|
|
# ANN Search Operator: MilvusClient |
|
|
|
|
|
|
|
*author: junjie.jiangjjj* |
|
|
|
|
|
|
@ -20,7 +20,7 @@ Search embedding in [Milvus](https://milvus.io/), **please make sure you have in |
|
|
|
from towhee.dc2 import pipe, ops |
|
|
|
|
|
|
|
p = pipe.input('vec') \ |
|
|
|
.map('vec', ('id', 'score'), ops.ann_insert.milvus_client(host='127.0.0.1', port='19530', collection_name='textdb')) \ |
|
|
|
.map('vec', ('id', 'score'), ops.ann_search.milvus_client(host='127.0.0.1', port='19530', collection_name='textdb')) \ |
|
|
|
.output('id', 'score') |
|
|
|
|
|
|
|
# call pipeline |
|
|
@ -33,7 +33,7 @@ from towhee.dc2 import pipe, ops |
|
|
|
# search additional info url: |
|
|
|
|
|
|
|
p = pipe.input('vec') \ |
|
|
|
.map('vec', ('id', 'score', 'url'), ops.ann_insert.milvus_client(host='127.0.0.1', port='19530', collection_name='textdb', output_fields=['url'])) \ |
|
|
|
.map('vec', ('id', 'score', 'url'), ops.ann_search.milvus_client(host='127.0.0.1', port='19530', collection_name='textdb', output_fields=['url'])) \ |
|
|
|
.output('id', 'score', 'url') |
|
|
|
``` |
|
|
|
|
|
|
|