chore: update gitignore; feat: 新增customer_agent业务模块与api路由

tests/test_milvus_collections.py

@@ -0,0 +1,52 @@
+import unittest
+from unittest.mock import AsyncMock, call
+from unittest.mock import patch
+
+from rag.milvus_collections import KNOWLEDGE_COLLECTIONS, ensure_collections
+
+
+class MilvusCollectionTests(unittest.IsolatedAsyncioTestCase):
+    async def test_creates_all_project_knowledge_collections(self):
+        client = AsyncMock()
+        client.has_collection.return_value = False
+
+        await ensure_collections(client)
+
+        self.assertEqual(
+            client.has_collection.await_args_list,
+            [call(name) for name in KNOWLEDGE_COLLECTIONS],
+        )
+        self.assertEqual(client.create_collection.await_count, len(KNOWLEDGE_COLLECTIONS))
+
+    async def test_reuses_existing_collections(self):
+        client = AsyncMock()
+        client.has_collection.return_value = True
+
+        await ensure_collections(client)
+
+        client.create_collection.assert_not_awaited()
+
+    async def test_collection_schema_contains_metadata_and_768_vector(self):
+        client = AsyncMock()
+        client.has_collection.return_value = False
+
+        await ensure_collections(client)
+
+        schema = client.create_collection.await_args.kwargs["schema"]
+        fields = {field["name"] for field in schema.to_dict()["fields"]}
+        self.assertTrue({"chunk_id", "doc_id", "title", "section_title", "text", "strategy", "vector"} <= fields)
+        vector = next(field for field in schema.to_dict()["fields"] if field["name"] == "vector")
+        self.assertEqual(vector["params"]["dim"], 768)
+
+    async def test_default_path_uses_configured_milvus_client(self):
+        client = AsyncMock()
+        client.has_collection.return_value = True
+
+        with patch("rag.milvus_collections.configured_milvus_client", return_value=client):
+            await ensure_collections()
+
+        self.assertEqual(client.has_collection.await_count, len(KNOWLEDGE_COLLECTIONS))
+
+
+if __name__ == "__main__":
+    unittest.main()