diff --git a/unstructured_ingest/v2/processes/connectors/couchbase.py b/unstructured_ingest/v2/processes/connectors/couchbase.py index 0e0529328..ab36256c9 100644 --- a/unstructured_ingest/v2/processes/connectors/couchbase.py +++ b/unstructured_ingest/v2/processes/connectors/couchbase.py @@ -40,11 +40,9 @@ class CouchbaseConnectionConfig(ConnectionConfig): connection_string: str = "couchbase://localhost" scope: str = "_default" collection: str = "_default" - access_config: CouchbaseAccessConfig = enhanced_field( - default_factory=CouchbaseAccessConfig, sensitive=True - ) batch_size: int = 50 connector_type: str = CONNECTOR_TYPE + access_config: CouchbaseAccessConfig = enhanced_field(sensitive=True) @dataclass @@ -93,8 +91,8 @@ class CouchbaseUploaderConfig(UploaderConfig): @dataclass class CouchbaseUploader(Uploader): - upload_config: CouchbaseUploaderConfig connection_config: CouchbaseConnectionConfig + upload_config: CouchbaseUploaderConfig cluster: Optional["Cluster"] = field(init=False, default=None) connector_type: str = CONNECTOR_TYPE @@ -110,10 +108,9 @@ def connect_to_couchbase(self) -> "Cluster": from couchbase.cluster import Cluster from couchbase.options import ClusterOptions - access_conf = self.connection_config.access_config - connection_string = access_conf.connection_string - username = access_conf.username - password = access_conf.password + connection_string = self.connection_config.connection_string + username = self.connection_config.username + password = self.connection_config.access_config.password auth = PasswordAuthenticator(username, password) options = ClusterOptions(auth) @@ -131,7 +128,7 @@ def run(self, contents: list[UploadContent], **kwargs: Any) -> None: logger.info( f"writing {len(elements)} objects to destination " f"bucket, {self.connection_config.bucket} " - f"at {self.connection_config.access_config.connection_string}", + f"at {self.connection_config.connection_string}", ) bucket = self.cluster.bucket(self.connection_config.bucket) scope = bucket.scope(self.connection_config.scope)