cloud config connect timeout to 120

datastax · Aug 27, 2024 · c442876 · c442876
1 parent 2cb9ee3
commit c442876
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/impl/astra_vector.py b/impl/astra_vector.py
@@ -22,7 +22,6 @@
 from cassandra.query import (
     UNSET_VALUE,
     SimpleStatement,
-    ValueSequence,
     dict_factory,
     named_tuple_factory, PreparedStatement,
 )
@@ -146,7 +145,6 @@ async def async_setup(self):
         if self.dbid is None:
             await self.get_or_create_db()
 
-        # Attempt to connect synchronously (assuming connect is a sync method)
         session = self.connect()
         if session:
             self.session: Session = session
@@ -337,7 +335,7 @@ def connect(self, retry=False) -> Session:
                         with open(bundlepath, "wb") as f:
                             f.write(r.content)
                 # Connect to the cluster
-                cloud_config = {"secure_connect_bundle": bundlepath}
+                cloud_config = {"secure_connect_bundle": bundlepath, "connect_timeout": 120}
                 auth_provider = PlainTextAuthProvider(CASSANDRA_USER, token)
                 cluster = Cluster(
                     cloud=cloud_config,