From 2ca1302b870682475d00b44887558fd23bd55078 Mon Sep 17 00:00:00 2001
From: Heiru Wu <heiru.wu@instill.tech>
Date: Wed, 4 Sep 2024 20:38:58 +0800
Subject: [PATCH] chore: update test model_hub

---
 model-hub/model_hub_test.json | 44 +++++++++++++++++------------------
 1 file changed, 22 insertions(+), 22 deletions(-)

diff --git a/model-hub/model_hub_test.json b/model-hub/model_hub_test.json
index 72e26189..d459f870 100644
--- 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",