Format a vector for better readability
(vector)
| 15 | |
| 16 | |
| 17 | def format_vector(vector): |
| 18 | """Format a vector for better readability""" |
| 19 | # Handle Vector class instances |
| 20 | if hasattr(vector, "dense_values"): |
| 21 | dense_values = vector.dense_values |
| 22 | if dense_values: |
| 23 | # Show first 5 dimensions and total count |
| 24 | preview = dense_values[:5] |
| 25 | formatted = { |
| 26 | "id": vector.id, |
| 27 | "dense_values": f"{preview}... (total: {len(dense_values)} dimensions)", |
| 28 | } |
| 29 | else: |
| 30 | formatted = {"id": vector.id, "dense_values": "[]"} |
| 31 | return json.dumps(formatted, indent=2) |
| 32 | # Handle dictionary format |
| 33 | elif isinstance(vector, dict): |
| 34 | dense_values = vector.get("dense_values", []) |
| 35 | if dense_values: |
| 36 | # Show first 5 dimensions and total count |
| 37 | preview = dense_values[:5] |
| 38 | formatted = { |
| 39 | "id": vector.get("id", "N/A"), |
| 40 | "dense_values": f"{preview}... (total: {len(dense_values)} dimensions)", |
| 41 | } |
| 42 | else: |
| 43 | formatted = {"id": vector.get("id", "N/A"), "dense_values": "[]"} |
| 44 | return json.dumps(formatted, indent=2) |
| 45 | return str(vector) |
| 46 | |
| 47 | |
| 48 | def generate_random_vector(id: int, dimension: int) -> dict: |