cloud config connect timeout to 120

Author: phact

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,