WIP: debug

Author: tellet-q

ansible/playbooks/roles/run-transfer-speed/files/shard_transfer.py

@@ -106,7 +106,7 @@ def upload_vectors(self, vectors: np.ndarray):
             parallel=16,
         )
 
-    def wait_for_green(self, timeout: int = 1800):
+    def wait_for_green(self, timeout: int = 600):
         print("Waiting for green status...")
         wait_time = 5.0
         total = 0
@@ -116,10 +116,10 @@ def wait_for_green(self, timeout: int = 1800):
             total += wait_time
             info = self.primary.get_collection(COLLECTION_NAME)
             status = info.status
-            indexed = info.indexed_vectors_count
-            total_vectors = info.vectors_count
-            segments = info.segments_count
-            print(f"  [{total:.0f}s] status={status}, indexed={indexed}/{total_vectors}, segments={segments}")
+            indexed = info.indexed_vectors_count or 0
+            total_points = info.points_count or 0
+            segments = info.segments_count or 0
+            print(f"  [{total:.0f}s] status={status}, indexed={indexed}/{total_points}, segments={segments}")
 
             if status != models.CollectionStatus.GREEN:
                 continue