feat: 检索测试
This commit is contained in:
parent
cf9d972e00
commit
eb73101ddb
|
|
@ -7,7 +7,6 @@ from src.pipeline.core import llm, es, nodes
|
||||||
|
|
||||||
@pytest.mark.asyncio
|
@pytest.mark.asyncio
|
||||||
async def test_embedding():
|
async def test_embedding():
|
||||||
return
|
|
||||||
|
|
||||||
await llm.init_client()
|
await llm.init_client()
|
||||||
await es.init_client()
|
await es.init_client()
|
||||||
|
|
@ -24,7 +23,8 @@ async def test_embedding():
|
||||||
chunkNode = nodes.ChunkDocumentsNode()
|
chunkNode = nodes.ChunkDocumentsNode()
|
||||||
embeddingNode = nodes.EmbeddingDocumentsNode()
|
embeddingNode = nodes.EmbeddingDocumentsNode()
|
||||||
writeToESNode = nodes.WriteDocumentsToESNode()
|
writeToESNode = nodes.WriteDocumentsToESNode()
|
||||||
readNode >> chunkNode >> embeddingNode >> writeToESNode
|
readNode >> chunkNode >> embeddingNode
|
||||||
|
# >> writeToESNode
|
||||||
flow = AsyncFlow(readNode)
|
flow = AsyncFlow(readNode)
|
||||||
|
|
||||||
await flow.run_async(shared)
|
await flow.run_async(shared)
|
||||||
|
|
@ -44,6 +44,7 @@ async def test_search():
|
||||||
shared = {
|
shared = {
|
||||||
"text": "那座山盛产金属矿物",
|
"text": "那座山盛产金属矿物",
|
||||||
"index": "test_kb",
|
"index": "test_kb",
|
||||||
|
"results": [], # [{es_id, doc_id, title, type, created_at, score, content}]
|
||||||
}
|
}
|
||||||
|
|
||||||
embeddingNode = nodes.EmbeddingNode()
|
embeddingNode = nodes.EmbeddingNode()
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue