refactor: 客服agent的切片结构调整重构

This commit is contained in:
2026-09-11 17:31:16 +08:00
parent b1b764eac9
commit e561c97a4e
20 changed files with 3309 additions and 230 deletions
+4 -3
View File
@@ -1,6 +1,7 @@
import unittest
from unittest.mock import AsyncMock
from rag.embedding import EMBEDDING_DIMENSION
from rag.ingestion import ingest_document_atomic
@@ -35,7 +36,7 @@ class AtomicIngestionTests(unittest.IsolatedAsyncioTestCase):
async def test_milvus_failure_cleans_up_partial_document_rows(self):
milvus = AsyncMock()
milvus.insert.side_effect = RuntimeError("insert down")
embedder = AsyncMock(return_value=[[0.0] * 768])
embedder = AsyncMock(return_value=[[0.0] * EMBEDDING_DIMENSION])
with self.assertRaises(RuntimeError):
await ingest_document_atomic(
@@ -49,7 +50,7 @@ class AtomicIngestionTests(unittest.IsolatedAsyncioTestCase):
async def test_success_inserts_complete_document_rows(self):
milvus = AsyncMock()
embedder = AsyncMock(return_value=[[0.0] * 768])
embedder = AsyncMock(return_value=[[0.0] * EMBEDDING_DIMENSION])
result = await ingest_document_atomic(
"plain text", "doc-4", "Policy", "fin_policy", "default",
@@ -62,7 +63,7 @@ class AtomicIngestionTests(unittest.IsolatedAsyncioTestCase):
async def test_cleans_text_before_embedding_and_milvus_insert(self):
milvus = AsyncMock()
embedder = AsyncMock(return_value=[[0.0] * 768, [0.0] * 768])
embedder = AsyncMock(return_value=[[0.0] * EMBEDDING_DIMENSION, [0.0] * EMBEDDING_DIMENSION])
await ingest_document_atomic(
"\ufeff基金名称:示例基金\r\n\r\n\r\n风险等级:R3\t ",