Fix error handling and allow UUID or Int IDs

CHANGELOG.md

@@ -5,6 +5,10 @@ This changelog documents changes between release tags.
 ## [Unreleased]
 Upcoming changes for the next versioned release.
 
+## [0.1.6] - 2024-03-22
+* Fix error handling
+* Allow String (UUID) OR Integer IDs by introspecting the ID type.
+
 ## [0.1.5] - 2024-03-13
 * Fix connector-metadata.yaml
 

connector-definition/connector-metadata.yaml

@@ -1,13 +1,13 @@
 packagingDefinition:
   type: PrebuiltDockerImage
-  dockerImage: ghcr.io/hasura/ndc-qdrant:v0.1.5
+  dockerImage: ghcr.io/hasura/ndc-qdrant:v0.1.6
 supportedEnvironmentVariables:
   - name: QDRANT_URL
     description: The url for the Qdrant database
   - name: QDRANT_API_KEY
     description: The Qdrant API Key
 commands:
-  update: docker run --rm -e QDRANT_URL="$QDRANT_URL" -e QDRANT_API_KEY="$QDRANT_API_KEY" -v "$HASURA_PLUGIN_CONNECTOR_CONTEXT_PATH":/etc/connector ghcr.io/hasura/ndc-qdrant:v0.1.5 update
+  update: docker run --rm -e QDRANT_URL="$QDRANT_URL" -e QDRANT_API_KEY="$QDRANT_API_KEY" -v "$HASURA_PLUGIN_CONNECTOR_CONTEXT_PATH":/etc/connector ghcr.io/hasura/ndc-qdrant:v0.1.6 update
 dockerComposeWatch:
   - path: ./
     target: /etc/connector

generate-config.ts

@@ -30,23 +30,72 @@ async function main() {
       with_payload: true,
     });
     let fieldDict = {};
+    let baseFields = {};
+    let insertFields = {};
     if (records.length > 0) {
+      console.log(records);
       const recordPayload = records[0].payload;
       fieldDict = insertion(cn, recordPayload!, objectTypes);
+      if (typeof records[0].id === "number"){
+        baseFields = {
+          id: {
+            description: null,
+            type: {
+              type: "named",
+              name: "Int",
+            },
+          },
+          ...BASE_FIELDS
+        };
+        insertFields = {
+          id: {
+            description: null,
+            type: {
+              type: "named",
+              name: "Int",
+            },
+          },
+          ...INSERT_FIELDS
+        };
+      } else {
+        baseFields = {
+          id: {
+            description: null,
+            type: {
+              type: "named",
+              name: "String",
+            },
+          },
+          ...BASE_FIELDS
+        };
+        insertFields = {
+          id: {
+            description: null,
+            type: {
+              type: "named",
+              name: "String",
+            },
+          },
+          ...INSERT_FIELDS
+        };
+      }
     }
+    console.log(fieldDict);
+
     objectTypes[cn] = {
       description: null,
       fields: {
         ...fieldDict,
-        ...BASE_FIELDS,
+        ...baseFields,
       },
     };
 
+    // Need to handle that here as well for insert fields.
     objectTypes[`${cn}_InsertType`] = {
       description: null,
       fields: {
         ...fieldDict,
-        ...INSERT_FIELDS
+        ...insertFields
       }
     }
   }

package.json

@@ -1,6 +1,6 @@
 {
   "name": "ndc-qdrant",
-  "version": "0.1.5",
+  "version": "0.1.6",
   "main": "index.js",
   "scripts": {},
   "dependencies": {

src/constants.ts

@@ -125,13 +125,6 @@ export const SCALAR_TYPES: { [key: string]: ScalarType } = {
 };
 
 export const INSERT_FIELDS: Record<string, ObjectField> = {
-  id: {
-    description: null,
-    type: {
-      type: "named",
-      name: ID_FIELD_TYPE,
-    },
-  },
   vector: {
     description: null,
     type: {
@@ -145,13 +138,6 @@ export const INSERT_FIELDS: Record<string, ObjectField> = {
 };
 
 export const BASE_FIELDS: Record<string, ObjectField> = {
-  id: {
-    description: null,
-    type: {
-      type: "named",
-      name: ID_FIELD_TYPE,
-    },
-  },
   score: {
     description: null,
     type: {