(vector_db_name, vectors)
| 165 | |
| 166 | |
| 167 | def batch_ann_search(vector_db_name, vectors): |
| 168 | url = f"{base_url}/collections/{vector_db_name}/search/batch-dense" |
| 169 | data = {"queries": [{"vector": vector} for vector in vectors], "top_k": 5} |
| 170 | response = requests.post( |
| 171 | url, headers=generate_headers(), data=json.dumps(data), verify=False |
| 172 | ) |
| 173 | if response.status_code != 200: |
| 174 | print(f"Error response: {response.text}") |
| 175 | raise Exception(f"Failed to batch search vector: {response.status_code}") |
| 176 | result = response.json() |
| 177 | return result |
| 178 | |
| 179 | |
| 180 | # Function to generate a random vector with given constraints |
nothing calls this directly
no test coverage detected