From 915ee385db5db89f5866986a6f6e834c1829d49a Mon Sep 17 00:00:00 2001 From: wangxiaolei Date: Mon, 16 Mar 2026 14:32:09 +0800 Subject: [PATCH] fix: fix weaviate_vector test failed (#33511) --- .../core/rag/datasource/vdb/weaviate/test_weaviate_vector.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/api/tests/unit_tests/core/rag/datasource/vdb/weaviate/test_weaviate_vector.py b/api/tests/unit_tests/core/rag/datasource/vdb/weaviate/test_weaviate_vector.py index a32f27fc91..3bd656ba84 100644 --- a/api/tests/unit_tests/core/rag/datasource/vdb/weaviate/test_weaviate_vector.py +++ b/api/tests/unit_tests/core/rag/datasource/vdb/weaviate/test_weaviate_vector.py @@ -11,6 +11,7 @@ import unittest from types import SimpleNamespace from unittest.mock import MagicMock, patch +from core.rag.datasource.vdb.weaviate import weaviate_vector as weaviate_vector_module from core.rag.datasource.vdb.weaviate.weaviate_vector import WeaviateConfig, WeaviateVector from core.rag.models.document import Document @@ -19,6 +20,7 @@ class TestWeaviateVector(unittest.TestCase): """Tests for WeaviateVector class with focus on doc_type metadata handling.""" def setUp(self): + weaviate_vector_module._weaviate_client = None self.config = WeaviateConfig( endpoint="http://localhost:8080", api_key="test-key", @@ -27,6 +29,9 @@ class TestWeaviateVector(unittest.TestCase): self.collection_name = "Test_Collection_Node" self.attributes = ["doc_id", "dataset_id", "document_id", "doc_hash", "doc_type"] + def tearDown(self): + weaviate_vector_module._weaviate_client = None + @patch("core.rag.datasource.vdb.weaviate.weaviate_vector.weaviate") def _create_weaviate_vector(self, mock_weaviate_module): """Helper to create a WeaviateVector instance with mocked client."""