Merge pull request #172 from anjanaw/dev

retain test
isee4xai · Jan 22, 2024 · 1d5a70e · 1d5a70e
2 parents c8e698e + 00fd780
commit 1d5a70e
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 0 deletions.
diff --git a/src/pages/designuser/usecases/Create.tsx b/src/pages/designuser/usecases/Create.tsx
@@ -9,6 +9,7 @@ import {
   api_dataset_upload,
   api_delete,
   api_get,
+  api_retain,
   api_get_casestructure,
   api_get_model_instance_count,
   api_get_model_prediction,
@@ -669,6 +670,18 @@ const Create: React.FC<Params> = (props) => {
                   defaultChecked={usecase.published}
                   checked={usecase.published}
                 />
+                <Button
+                  type="primary"
+                  style={{ margin: '0 1rem' }}
+                  onClick={async () => {
+                    console.log('Retain Started!');
+                    await api_retain(usecase._id || '');
+                  }}
+                  htmlType="button"
+                  icon={<SaveOutlined />}
+                >
+                  Retain
+                </Button>
                 <Button
                   type="primary"
                   style={{ margin: '0 1rem' }}

diff --git a/src/services/isee/usecases.ts b/src/services/isee/usecases.ts
@@ -480,6 +480,27 @@ export async function api_persona_query_set_default(
   }
 }
 
+export async function api_retain(usecaseId: string) {
+  if (!usecaseId) return usecaseId;
+
+  try {
+    const data = await fetch(`${BASE_URL}/cbr/${usecaseId}/retain`, {
+      method: 'POST',
+      headers: {
+        'Content-Type': 'application/json',
+        'x-access-token': getToken(),
+      },
+    });
+    const result = await data.json();
+    if (result.message) {
+      message.error("Case retain Failed! Make sure you have a completed the use case.")
+      return []
+    };
+    return result;
+  } catch (error) {
+    return [];
+  }
+}
 
 export async function api_usecase_analytics(usecaseId: string | undefined, version: string | undefined) {
   if (!usecaseId || !version) return usecaseId;