Upsert multiple key-value pairs in a single request
(name, txn_id, items: list[tuple[str, str]])
| 70 | raise Exception(f"Failed to upsert vectors: {response.text}") |
| 71 | |
| 72 | def bulk_upsert(name, txn_id, items: list[tuple[str, str]]): |
| 73 | """Upsert multiple key-value pairs in a single request""" |
| 74 | url = f"{client.base_url}/collections/{name}/transactions/{txn_id}/upsert" |
| 75 | data = {"vectors": [{"id": k, "bytes": list(v.encode("utf-8"))} for k, v in items]} |
| 76 | response = requests.post(url, headers=client._get_headers(), data=json.dumps(data), verify=client.verify_ssl) |
| 77 | if response.status_code not in [200, 204]: |
| 78 | raise Exception(f"Failed to bulk upsert: {response.text}") |
| 79 | |
| 80 | def lookup(name, key): |
| 81 | url = f"{client.base_url}/collections/{name}/search/key-value" |
no test coverage detected