| 31 | |
| 32 | |
| 33 | def create_db(name: str, description: str | None = None): |
| 34 | client._ensure_session() |
| 35 | |
| 36 | url = f"{client.base_url}/collections" |
| 37 | data = { |
| 38 | "name": name, |
| 39 | "description": description, |
| 40 | "key_value_options": { "enabled": True }, |
| 41 | "config": {"max_vectors": None, "replication_factor": None}, |
| 42 | "store_raw_text": False, |
| 43 | } |
| 44 | |
| 45 | response = requests.post( |
| 46 | url, |
| 47 | headers=client._get_headers(), |
| 48 | data=json.dumps(data), |
| 49 | verify=client.verify_ssl, |
| 50 | ) |
| 51 | |
| 52 | if response.status_code not in [200, 201]: |
| 53 | raise Exception(f"Failed to create collection: {response.text}") |
| 54 | |
| 55 | |
| 56 | def create_index(name): |