feat: Update Couchbase connection options in test_couchbasedb.py

microsoft · Sep 19, 2024 · 30bbab9 · 30bbab9
1 parent 8cf8fed
commit 30bbab9
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/unit/vector_stores/test_couchbasedb.py b/tests/unit/vector_stores/test_couchbasedb.py
@@ -3,6 +3,8 @@
 import time
 import unittest
 
+from couchbase.auth import PasswordAuthenticator
+from couchbase.options import ClusterOptions
 from dotenv import load_dotenv
 
 from graphrag.vector_stores.base import VectorStoreDocument, VectorStoreSearchResult
@@ -29,10 +31,12 @@ def setUpClass(cls):
  index_name=INDEX_NAME,
  vector_size=VECTOR_SIZE
  )
+ auth = PasswordAuthenticator(COUCHBASE_USERNAME, COUCHBASE_PASSWORD)
+ cluster_options = ClusterOptions(auth)
+
  cls.vector_store.connect(
  connection_string=COUCHBASE_CONNECTION_STRING,
- username=COUCHBASE_USERNAME,
- password=COUCHBASE_PASSWORD
+ cluster_options=cluster_options
  )
 
  @classmethod