From eb73101ddbf8bee21291989d81ef9ac7799840af Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E6=9D=8E=E5=A6=82=E5=A8=81?= Date: Fri, 19 Dec 2025 14:06:46 +0800 Subject: [PATCH] =?UTF-8?q?feat:=20=E6=A3=80=E7=B4=A2=E6=B5=8B=E8=AF=95?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- src/tests/test_nodes.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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()