WIP: fix

Author: tellet-q

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

@@ -66,18 +66,28 @@ def storage_types(self, uri: str) -> dict:
         try:
             r = requests.get(f"{uri}/telemetry?details_level=6", timeout=10)
             if not r.ok:
+                print(f"    Telemetry request failed: {r.status_code}")
                 return {}
-            for coll in r.json().get("result", {}).get("collections", {}).get("collections", []):
+            data = r.json()
+            collections = data.get("result", {}).get("collections", {}).get("collections", [])
+            if not collections:
+                print(f"    No collections in telemetry, keys: {list(data.get('result', {}).keys())}")
+                return {}
+            for coll in collections:
                 if coll.get("id") == COLLECTION:
                     types = {}
                     for shard in coll.get("shards", []):
-                        for seg in shard.get("local", {}).get("segments", []):
+                        local = shard.get("local")
+                        if not local:
+                            continue
+                        for seg in local.get("segments", []):
                             for vec in seg.get("config", {}).get("vector_data", {}).values():
                                 st = vec.get("storage_type", "unknown")
                                 types[st] = types.get(st, 0) + 1
                     return types
-        except Exception:
-            pass
+            print(f"    Collection '{COLLECTION}' not found, available: {[c.get('id') for c in collections]}")
+        except Exception as e:
+            print(f"    Telemetry error: {e}")
         return {}
 
     def wait_mmap(self, uri: str, timeout=180):

ansible/playbooks/roles/run-transfer-speed/tasks/main.yml

@@ -50,6 +50,11 @@
     OUTPUT_FILENAME: "{{ working_dir }}/results.json"
     WORK_DIR: "{{ working_dir }}"
     PYTHONUNBUFFERED: "1"
+  register: benchmark_output
+
+- name: Show benchmark output
+  debug:
+    msg: "{{ benchmark_output.stdout_lines }}"
 
 - name: Read results
   ansible.builtin.slurp: