From 01e27def9e471d2db3a6841957acffc6fad56a97 Mon Sep 17 00:00:00 2001 From: quicksandzn Date: Thu, 25 Apr 2024 16:15:32 +0800 Subject: [PATCH] optimize: reformat and move test to api/tests/integration_tests/vdb/tcvectordb --- .../vdb/tencent => integration_tests/vdb}/__init__.py | 0 api/tests/integration_tests/vdb/tcvectordb/__init__.py | 0 .../vdb/tcvectordb}/test_tencent.py | 3 ++- 3 files changed, 2 insertions(+), 1 deletion(-) rename api/tests/{unit_tests/core/rag/datasource/vdb/tencent => integration_tests/vdb}/__init__.py (100%) create mode 100644 api/tests/integration_tests/vdb/tcvectordb/__init__.py rename api/tests/{unit_tests/core/rag/datasource/vdb/tencent => integration_tests/vdb/tcvectordb}/test_tencent.py (99%) diff --git a/api/tests/unit_tests/core/rag/datasource/vdb/tencent/__init__.py b/api/tests/integration_tests/vdb/__init__.py similarity index 100% rename from api/tests/unit_tests/core/rag/datasource/vdb/tencent/__init__.py rename to api/tests/integration_tests/vdb/__init__.py diff --git a/api/tests/integration_tests/vdb/tcvectordb/__init__.py b/api/tests/integration_tests/vdb/tcvectordb/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/api/tests/unit_tests/core/rag/datasource/vdb/tencent/test_tencent.py b/api/tests/integration_tests/vdb/tcvectordb/test_tencent.py similarity index 99% rename from api/tests/unit_tests/core/rag/datasource/vdb/tencent/test_tencent.py rename to api/tests/integration_tests/vdb/tcvectordb/test_tencent.py index 7906fad3b2..ced0a13d28 100644 --- a/api/tests/unit_tests/core/rag/datasource/vdb/tencent/test_tencent.py +++ b/api/tests/integration_tests/vdb/tcvectordb/test_tencent.py @@ -1,7 +1,8 @@ import pytest -from extensions.ext_redis import redis_client + from core.rag.datasource.vdb.tencent.tencent_vector import TencentConfig, TencentVector from core.rag.models.document import Document +from extensions.ext_redis import redis_client def _create_tencent_vector() -> TencentVector: