diff --git a/src/tests/test_nodes.py b/src/tests/test_nodes.py index cdae9f6..49137bb 100644 --- a/src/tests/test_nodes.py +++ b/src/tests/test_nodes.py @@ -7,7 +7,6 @@ from src.pipeline.core import llm, es, nodes @pytest.mark.asyncio async def test_embedding(): - return await llm.init_client() await es.init_client() @@ -24,7 +23,8 @@ async def test_embedding(): chunkNode = nodes.ChunkDocumentsNode() embeddingNode = nodes.EmbeddingDocumentsNode() writeToESNode = nodes.WriteDocumentsToESNode() - readNode >> chunkNode >> embeddingNode >> writeToESNode + readNode >> chunkNode >> embeddingNode + # >> writeToESNode flow = AsyncFlow(readNode) await flow.run_async(shared) @@ -44,6 +44,7 @@ async def test_search(): shared = { "text": "那座山盛产金属矿物", "index": "test_kb", + "results": [], # [{es_id, doc_id, title, type, created_at, score, content}] } embeddingNode = nodes.EmbeddingNode()