Add vector name auto-complete in visualisation (#182)

* refactor: Update Autocomplete.js to include vector names in autocomplete suggestions

* chore: Update autocomplete-openapi dependency to version 0.1.3

* fmt

package.json

@@ -25,7 +25,7 @@
     "@uppy/react": "^3.1.2",
     "@uppy/xhr-upload": "^3.3.1",
     "@vitejs/plugin-react": "^4.0.0",
-    "autocomplete-openapi": "^0.1.2",
+    "autocomplete-openapi": "0.1.3",
     "axios": "^1.6.7",
     "chart.js": "^4.3.0",
     "chroma-js": "^2.4.2",

src/components/FilterEditorWindow/config/Autocomplete.js

@@ -1,12 +1,17 @@
 import { OpenapiAutocomplete } from 'autocomplete-openapi/src/autocomplete';
 
-export const autocomplete = async (monaco, qdrantClient) => {
+export const autocomplete = async (monaco, qdrantClient, collectionName) => {
   const response = await fetch(import.meta.env.BASE_URL + './openapi.json');
   const openapi = await response.json();
 
-  let collections = [];
+  const vectorNames = [];
   try {
-    collections = (await qdrantClient.getCollections()).collections.map((c) => c.name);
+    const collectionInfo = await qdrantClient.getCollection(collectionName);
+    Object.keys(collectionInfo.config.params.vectors).map((key) => {
+      if (typeof collectionInfo.config.params.vectors[key] === 'object') {
+        vectorNames.push(key);
+      }
+    });
   } catch (e) {
     console.error(e);
   }
@@ -35,7 +40,7 @@ export const autocomplete = async (monaco, qdrantClient) => {
       vector_name: {
         description: 'Vector field name',
         type: 'string',
-        nullable: true,
+        enum: vectorNames,
       },
       color_by: {
         description: 'Color points by this field',
@@ -46,7 +51,7 @@ export const autocomplete = async (monaco, qdrantClient) => {
   };
   openapi.components.schemas.FilterRequest = FilterRequest;
 
-  const autocomplete = new OpenapiAutocomplete(openapi, collections);
+  const autocomplete = new OpenapiAutocomplete(openapi, []);
 
   return {
     provideCompletionItems: (model, position) => {

src/components/FilterEditorWindow/index.jsx

@@ -81,7 +81,7 @@ const CodeEditorWindow = ({ onChange, code, onChangeResult }) => {
     });
   }
   function handleEditorWillMount(monaco) {
-    autocomplete(monaco, qdrantClient).then((autocomplete) => {
+    autocomplete(monaco, qdrantClient, collectionName).then((autocomplete) => {
       autocompleteRef.current = monaco.languages.registerCompletionItemProvider('custom-language', autocomplete);
     });
   }

src/components/Sidebar/Sidebar.jsx

@@ -83,7 +83,6 @@ export default function Sidebar({ open, version, jwtEnabled }) {
 }
 
 function sidebarItem(title, icon, linkPath, isOpen, enabled = true) {
-  console.log('enabled', enabled);
   return (
     <ListItem key={title} disablePadding sx={{ display: 'block' }}>
       <Tooltip title={title} placement="right" arrow disableHoverListener={isOpen}>