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