(idd, vector_db_name, vector)
| 172 | |
| 173 | |
| 174 | def ann_vector(idd, vector_db_name, vector): |
| 175 | url = f"{base_url}/collections/{vector_db_name}/search/dense" |
| 176 | data = {"query_vector": vector, "top_k": 5} # Corrected body |
| 177 | response = requests.post( |
| 178 | url, headers=generate_headers(), data=json.dumps(data), verify=False |
| 179 | ) |
| 180 | if response.status_code != 200: |
| 181 | print(f"Error response: {response.text}") |
| 182 | raise Exception(f"Failed to search vector: {response.status_code}") |
| 183 | result = response.json() |
| 184 | |
| 185 | return (idd, result) |
| 186 | |
| 187 | |
| 188 | # Function to generate a random vector with given constraints |
no test coverage detected