in storage-reverse-image-search/functions/src/common/vertex.ts [228:257]
export async function queryIndex(
queries: Query[],
searchResults: number,
endpoint: string,
indexEndpoint: string
) {
try {
const accessToken = await getAccessToken();
const response = await axios.post(
`https://${endpoint}/v1beta1/projects/${config.projectId}/locations/${config.location}/indexEndpoints/${indexEndpoint}:findNeighbors`,
{
queries: queries.map(query => query.toVertexQuery()),
deployed_index_id: `ext_${config.instanceId.replace(/-/g, '_')}_index`,
neighbor_count: searchResults,
},
{
headers: {
'Content-Type': 'application/json',
Authorization: `Bearer ${accessToken}`,
},
}
);
return response.data;
} catch (error) {
console.error('Error calling the endpoint:', error);
throw error;
}
}