fix

qdrant · Dec 16, 2024 · 587f41c · 587f41c
1 parent 4d8d261
commit 587f41c
Showing 1 changed file with 5 additions and 16 deletions.
diff --git a/src/lib/graph-visualization-helpers.js b/src/lib/graph-visualization-helpers.js
@@ -153,21 +153,10 @@ export const getMinimalSpanningTree = (links, acs = true) => {
   return mstLinks;
 };
 
-//  the function which returns metric for given collection and name of the vector
-
-export const getMetrics = async (qdrantClient, { collectionName }) => {
-  const metricConfig = {};
-  try {
-    const collectionInfo = await qdrantClient.getCollection(collectionName);
-    Object.keys(collectionInfo.config.params.vectors).map((key) => {
-      if (typeof collectionInfo.config.params.vectors[key] === 'object') {
-        metricConfig[key] = collectionInfo.config.params.vectors[key].distance;
-      } else {
-        metricConfig[''] = collectionInfo.config.params.vectors.distance;
-      }
-    });
-    return metricConfig;
-  } catch (e) {
-    console.error(e);
+export const getMetrics = async (qdrantClient, { collectionName, vectorName = null }) => {
+  const collectionInfo = await qdrantClient.getCollection(collectionName);
+  if (vectorName && collectionInfo.config.params.vectors[vectorName]) {
+    return collectionInfo.config.params.vectors[vectorName].distance;
   }
+  return collectionInfo.config.params.vectors.distance;
 };