feat(schema): add standardized AI tasks schema (#1070)

Because - We are going to standardize all AI tasks and unify them between Instill Model and Instill VDP This commit - add AI Task schema resolves INS-5456 resolves INS-5463 resolves INS-5464 resolves INS-5474
instill-ai · Sep 5, 2024 · 78133b5 · 78133b5
1 parent cac94c9
commit 78133b5
Show file tree

Hide file tree

Showing 3 changed files with 1,849 additions and 22 deletions.
diff --git a/model-hub/model_hub_test.json b/model-hub/model_hub_test.json
@@ -25,19 +25,6 @@
     "version": "latest",
     "configuration": {}
   },
-  {
-    "id": "dummy-image-to-image",
-    "owner_type": "users",
-    "owner_id": "admin",
-    "description": "A image to image model for integration-test",
-    "model_definition": "model-definitions/container",
-    "visibility": "VISIBILITY_PUBLIC",
-    "task": "TASK_IMAGE_TO_IMAGE",
-    "region": "REGION_LOCAL",
-    "hardware": "CPU",
-    "version": "latest",
-    "configuration": {}
-  },
   {
     "id": "dummy-instance-segmentation",
     "owner_type": "users",
@@ -78,26 +65,26 @@
     "configuration": {}
   },
   {
-    "id": "dummy-text-generation",
+    "id": "dummy-completion",
     "owner_type": "users",
     "owner_id": "admin",
-    "description": "A text generation model for integration-test",
+    "description": "A completion model for integration-test",
     "model_definition": "model-definitions/container",
     "visibility": "VISIBILITY_PUBLIC",
-    "task": "TASK_TEXT_GENERATION",
+    "task": "TASK_COMPLETION",
     "region": "REGION_LOCAL",
     "hardware": "CPU",
     "version": "latest",
     "configuration": {}
   },
   {
-    "id": "dummy-text-generation-chat",
+    "id": "dummy-chat",
     "owner_type": "users",
     "owner_id": "admin",
-    "description": "A text generation chat model for integration-test",
+    "description": "A chat model for integration-test",
     "model_definition": "model-definitions/container",
     "visibility": "VISIBILITY_PUBLIC",
-    "task": "TASK_TEXT_GENERATION_CHAT",
+    "task": "TASK_CHAT",
     "region": "REGION_LOCAL",
     "hardware": "CPU",
     "version": "latest",
@@ -117,13 +104,26 @@
     "configuration": {}
   },
   {
-    "id": "dummy-visual-question-answering",
+    "id": "dummy-multimodal-chat",
+    "owner_type": "users",
+    "owner_id": "admin",
+    "description": "A multimodal chat model for integration-test",
+    "model_definition": "model-definitions/container",
+    "visibility": "VISIBILITY_PUBLIC",
+    "task": "TASK_CHAT",
+    "region": "REGION_LOCAL",
+    "hardware": "CPU",
+    "version": "latest",
+    "configuration": {}
+  },
+  {
+    "id": "dummy-text-embedding",
     "owner_type": "users",
     "owner_id": "admin",
-    "description": "A visual question answering model for integration-test",
+    "description": "A embedding model for integration-test",
     "model_definition": "model-definitions/container",
     "visibility": "VISIBILITY_PUBLIC",
-    "task": "TASK_VISUAL_QUESTION_ANSWERING",
+    "task": "TASK_EMBEDDING",
     "region": "REGION_LOCAL",
     "hardware": "CPU",
     "version": "latest",