Add Label ID

tnc-ca-geo · Jan 11, 2024 · dd6c6b5 · dd6c6b5
1 parent 01fdf5d
commit dd6c6b5
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/src/api/db/models/Image.js b/src/api/db/models/Image.js
@@ -476,6 +476,8 @@ export class ImageModel {
       return await retry(async () => {
         console.log('ImageModel.createInternalLabels - creating label: ', JSON.stringify(label));
 
+        label.type = 'ml';
+
         // find image, create label record
         const image = await ImageModel.queryById(label.imageId, context);
         if (isLabelDupe(image, label)) throw new DuplicateLabelError();
@@ -545,6 +547,8 @@ export class ImageModel {
       return await retry(async () => {
         console.log('ImageModel.createLabels - creating label: ', JSON.stringify(label));
 
+        label.type = 'manual';
+
         // find image, create label record
         const image = await ImageModel.queryById(label.imageId, context);
         const project = await ProjectModel.queryById(image.projectId);

diff --git a/src/api/type-defs/objects/Label.js b/src/api/type-defs/objects/Label.js
@@ -8,7 +8,7 @@ export default `
   type Label {
     _id: ID!
     type: String!
-    category: String!
+    labelId: String!
     conf: Float
     bbox: [Float!]!
     labeledDate: Date!