feat: 调整结构

This commit is contained in:
李如威 2026-01-07 10:05:32 +08:00
parent 5562641431
commit debd807416
2 changed files with 16 additions and 9 deletions

View File

@ -8,11 +8,21 @@ from src.pipeline.core import llm, es, nodes, utils
from src.pipeline.core.skills import load_skills from src.pipeline.core.skills import load_skills
@pytest.mark.asyncio @pytest.fixture(scope="session")
async def test_embedding(): async def init_llm():
await llm.init_client() await llm.init_client()
await es.init_client() yield
await llm.close_client()
@pytest.fixture(scope="session")
async def init_es():
await es.init_client()
yield
await es.close_client()
@pytest.mark.asyncio
async def test_embedding(init_llm, init_es):
logger.debug("file to es") logger.debug("file to es")
shared = { shared = {
@ -32,12 +42,9 @@ async def test_embedding():
logger.debug(json.dumps([{**x, "embedding":x["embedding"][:4]} for x in shared["documents"]], indent=4, ensure_ascii=False)) logger.debug(json.dumps([{**x, "embedding":x["embedding"][:4]} for x in shared["documents"]], indent=4, ensure_ascii=False))
await llm.close_client()
await es.close_client()
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_search(): async def test_search(init_llm, init_es):
await llm.init_client() await llm.init_client()
await es.init_client() await es.init_client()
logger.debug("search from es") logger.debug("search from es")
@ -73,7 +80,7 @@ async def test_search():
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_rerank(): async def test_rerank(init_llm, init_es):
await llm.init_client() await llm.init_client()
await es.init_client() await es.init_client()
logger.debug("search from es") logger.debug("search from es")
@ -113,7 +120,7 @@ async def test_rerank():
@pytest.mark.asyncio @pytest.mark.asyncio
async def test_agent(): async def test_agent(init_llm, init_es):
await llm.init_client() await llm.init_client()
await es.init_client() await es.init_client()