fix: remove psycopg dependency from rate_runner (#606)

Signed-off-by: min.tian <[email protected]>

Author: alwayslove2013

vectordb_bench/backend/clients/api.py

@@ -135,6 +135,7 @@ class VectorDB(ABC):
 
     "The filtering types supported by the VectorDB Client, default only non-filter"
     supported_filter_types: list[FilterOp] = [FilterOp.NonFilter]
+    name: str = ""
 
     @classmethod
     def filter_supported(cls, filters: Filter) -> bool:

vectordb_bench/backend/runner/rate_runner.py

@@ -7,7 +7,6 @@
 
 from vectordb_bench import config
 from vectordb_bench.backend.clients import api
-from vectordb_bench.backend.clients.pgvector.pgvector import PgVector
 from vectordb_bench.backend.dataset import DataSetIterator
 from vectordb_bench.backend.utils import time_it
 
@@ -48,7 +47,7 @@ def _insert_embeddings(db: api.VectorDB, emb: list[list[float]], metadata: list[
                     msg = f"Insert failed and retried more than {config.MAX_INSERT_RETRY} times"
                     raise RuntimeError(msg) from None
 
-        if isinstance(db, PgVector):
+        if db.name == "PgVector":
             # pgvector is not thread-safe for concurrent insert,
             #   so we need to copy the db object, make sure each thread has its own connection
             db_copy = deepcopy(db)