refactor: 客服agent的切片结构调整重构
This commit is contained in:
@@ -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 ",
|
||||
|
||||
Reference in New Issue
Block a user