Only assign self._connected at end of .connect() method

Author: tsmith023

weaviate/connect/v4.py

@@ -894,8 +894,6 @@ def connect(self) -> None:
         if self._connected:
             return None
 
-        self._connected = True
-
         self._open_connections_rest(self._auth, "sync")
 
         # need this to get the version of weaviate for version checks and proper GRPC configuration
@@ -940,6 +938,8 @@ def connect(self) -> None:
                 self._connected = False
                 raise e
 
+        self._connected = True
+
     def wait_for_weaviate(self, startup_period: int) -> None:
         for _i in range(startup_period):
             try:
@@ -1073,8 +1073,6 @@ async def connect(self) -> None:
         if self._connected:
             return None
 
-        self._connected = True
-
         await executor.aresult(self._open_connections_rest(self._auth, "async"))
 
         # need this to get the version of weaviate for version checks and proper GRPC configuration
@@ -1096,7 +1094,6 @@ async def connect(self) -> None:
             raise WeaviateStartUpError(f"Could not connect to Weaviate:{e}.") from e
 
         self.open_connection_grpc("async")
-        self._connected = True
         if self.embedded_db is not None:
             try:
                 await self.wait_for_weaviate(10)