From 2e5f9be6013f8da9d5c4f1702309256fad23d920 Mon Sep 17 00:00:00 2001
From: Germey <germey@acedata.cloud>
Date: Sun, 3 Mar 2024 15:03:49 +0800
Subject: [PATCH] fix actions

---
 .../midjourney/tasks/TaskBriefList.vue        |  9 +--
 .../midjourney/tasks/TaskFullList.vue         | 17 ++++--
 .../midjourney/tasks/TaskPreview.vue          | 47 ++++++---------
 src/components/order/Create.vue               | 11 +---
 src/operators/midjourney.ts                   |  3 +-
 src/pages/chatdoc/Conversation.vue            | 15 +----
 src/pages/chatdoc/Index.vue                   |  4 +-
 src/pages/midjourney/History.vue              | 12 ++--
 src/pages/midjourney/Index.vue                | 12 ++--
 src/store/chatdoc/models.ts                   |  2 +-
 src/store/chatdoc/mutations.ts                |  7 ++-
 src/store/chatdoc/state.ts                    |  9 ++-
 src/store/index.ts                            |  6 +-
 src/store/midjourney/actions.ts               | 60 +++++++++++++------
 src/store/midjourney/mutations.ts             |  7 ++-
 src/store/midjourney/state.ts                 |  9 ++-
 16 files changed, 128 insertions(+), 102 deletions(-)

diff --git a/src/components/midjourney/tasks/TaskBriefList.vue b/src/components/midjourney/tasks/TaskBriefList.vue
index f75b58df..455c84fc 100644
--- a/src/components/midjourney/tasks/TaskBriefList.vue
+++ b/src/components/midjourney/tasks/TaskBriefList.vue
@@ -14,7 +14,7 @@
   </div>
   <div v-else-if="tasks.length > 0" class="tasks">
     <div v-for="(task, taskKey) in tasks" :key="taskKey" class="task">
-      <task-preview :full="false" :model-value="task" :applications="applications" @custom="$emit('custom', $event)" />
+      <task-preview :full="false" :model-value="task" @custom="$emit('custom', $event)" />
     </div>
     <el-button type="primary" class="btn mb-4" @click="onLoadHistory">{{ $t('midjourney.button.history') }}</el-button>
   </div>
@@ -44,17 +44,18 @@ export default defineComponent({
   },
   computed: {
     loading() {
-      return this.$store.state.midjourney.getImagineTasksStatus === Status.Request;
+      return this.$store.state.midjourney.status.getApplication === Status.Request;
     },
     tasks() {
       return this.$store.state.midjourney.imagineTasks;
     },
-    applications() {
-      return this.$store.state.midjourney.applications;
+    application() {
+      return this.$store.state.midjourney.application;
     }
   },
   async mounted() {
     await this.$store.dispatch('midjourney/setImagineTasks', undefined);
+    // @ts-ignore
     this.job = setInterval(() => {
       this.getImagineTasks();
     }, 5000);
diff --git a/src/components/midjourney/tasks/TaskFullList.vue b/src/components/midjourney/tasks/TaskFullList.vue
index 414e88fc..2ce860bf 100644
--- a/src/components/midjourney/tasks/TaskFullList.vue
+++ b/src/components/midjourney/tasks/TaskFullList.vue
@@ -1,7 +1,7 @@
 <template>
   <div v-if="tasks" class="tasks">
     <div v-for="(task, taskKey) in tasks" :key="taskKey" class="task">
-      <task-preview :full="true" :model-value="task" :applications="applications" @custom="onCustom($event)" />
+      <task-preview :full="true" :model-value="task" @custom="onCustom($event)" />
     </div>
   </div>
   <div v-else class="tasks">
@@ -26,6 +26,10 @@ import Pagination from '@/components/common/Pagination.vue';
 import { ElCard, ElSkeleton, ElSkeletonItem } from 'element-plus';
 import { Status } from '@/models';
 
+interface IData {
+  job: number | undefined;
+}
+
 export default defineComponent({
   name: 'TaskFullList',
   components: {
@@ -36,9 +40,9 @@ export default defineComponent({
     ElCard
   },
   emits: ['update:modelValue', 'custom'],
-  data() {
+  data(): IData {
     return {
-      job: 0
+      job: undefined
     };
   },
   computed: {
@@ -55,13 +59,13 @@ export default defineComponent({
       return 12;
     },
     loading() {
-      return this.$store.state.midjourney.getImagineTasksStatus === Status.Request;
+      return this.$store.state.midjourney.status.getApplication === Status.Request;
     },
     tasks() {
       return this.$store.state.midjourney.imagineTasks;
     },
-    applications() {
-      return this.$store.state.midjourney.applications;
+    application() {
+      return this.$store.state.midjourney.application;
     }
   },
   watch: {
@@ -74,6 +78,7 @@ export default defineComponent({
   },
   async mounted() {
     await this.$store.dispatch('midjourney/setImagineTasks', undefined);
+    // @ts-ignore
     this.job = setInterval(() => {
       this.getImagineTasks();
     }, 5000);
diff --git a/src/components/midjourney/tasks/TaskPreview.vue b/src/components/midjourney/tasks/TaskPreview.vue
index 479748a1..8bbbf312 100644
--- a/src/components/midjourney/tasks/TaskPreview.vue
+++ b/src/components/midjourney/tasks/TaskPreview.vue
@@ -13,9 +13,9 @@
         </el-tag>
       </div>
       <div class="right">
-        <el-tag v-if="channel" type="info" class="channel">
-          <font-awesome-icon :class="{ icon: true, [channel.name]: true }" :icon="channel?.icon" />
-          {{ channel?.displayName }}
+        <el-tag v-if="mode" type="info" class="channel">
+          <font-awesome-icon :class="{ icon: true, [mode.name]: true }" :icon="mode?.icon" />
+          {{ mode?.displayName }}
         </el-tag>
       </div>
     </div>
@@ -103,7 +103,7 @@
 <script lang="ts">
 import { defineComponent } from 'vue';
 import { ElImage, ElTag, ElButton, ElTooltip, ElSkeleton, ElAlert } from 'element-plus';
-import { IApplication, IMidjourneyImagineTask, MidjourneyImagineAction, MidjourneyImagineState } from '@/models';
+import { IMidjourneyImagineTask, MidjourneyImagineAction, MidjourneyImagineState } from '@/models';
 import { FontAwesomeIcon } from '@fortawesome/vue-fontawesome';
 import CopyToClipboard from '@/components/common/CopyToClipboard.vue';
 
@@ -130,10 +130,6 @@ export default defineComponent({
       type: Object as () => IMidjourneyImagineTask | undefined,
       required: true
     },
-    applications: {
-      type: Object as () => IApplication[] | undefined,
-      required: true
-    },
     full: {
       type: Boolean,
       default: false
@@ -153,14 +149,16 @@ export default defineComponent({
         [MidjourneyImagineAction.VARIATION2]: 'V2',
         [MidjourneyImagineAction.VARIATION3]: 'V3',
         [MidjourneyImagineAction.VARIATION4]: 'V4',
-        [MidjourneyImagineAction.HIGH_VARIATION]: 'Vary (Strong)',
-        [MidjourneyImagineAction.LOW_VARIATION]: 'Vary (Subtle)',
+        [MidjourneyImagineAction.VARIATION_STRONG]: 'Vary (Strong)',
+        [MidjourneyImagineAction.VARIATION_SUBTLE]: 'Vary (Subtle)',
+        [MidjourneyImagineAction.UPSCALE_CREATIVE]: 'Upscale (Creative)',
+        [MidjourneyImagineAction.UPSCALE_SUBTLE]: 'Upscale (Subtle)',
         [MidjourneyImagineAction.ZOOM_OUT_2X]: 'Zoom Out 2x',
         [MidjourneyImagineAction.ZOOM_OUT_1_5X]: 'Zoom Out 1.5x',
-        [MidjourneyImagineAction.UPSCALE4_2x]: 'Upsample 2x',
-        [MidjourneyImagineAction.UPSAMPLE_4X]: 'Upsample 4x',
-        [MidjourneyImagineAction.REDO_UPSCALE_2X]: 'Redo Upsample 2x',
-        [MidjourneyImagineAction.REDO_UPSAMPLE_4X]: 'Redo Upsample 4x',
+        [MidjourneyImagineAction.UPSCALE_2X]: 'Upscale 2x',
+        [MidjourneyImagineAction.UPSCALE_4X]: 'Upscale 4x',
+        [MidjourneyImagineAction.REDO_UPSCALE_2X]: 'Redo Upscale 2x',
+        [MidjourneyImagineAction.REDO_UPSCALE_4X]: 'Redo Upscale 4x',
         [MidjourneyImagineAction.SQUARE]: 'Make Square',
         [MidjourneyImagineAction.PAN_LEFT]: '⬅️',
         [MidjourneyImagineAction.PAN_UP]: '⬆️',
@@ -182,10 +180,12 @@ export default defineComponent({
         [MidjourneyImagineAction.VARIATION_SUBTLE]: this.$t('midjourney.description.variation_subtle'),
         [MidjourneyImagineAction.ZOOM_OUT_2X]: this.$t('midjourney.description.zoom_out_2x'),
         [MidjourneyImagineAction.ZOOM_OUT_1_5X]: this.$t('midjourney.description.zoom_out_1_5x'),
-        [MidjourneyImagineAction.UPSCALE_2x]: this.$t('midjourney.description.upscale_2x'),
+        [MidjourneyImagineAction.UPSCALE_2X]: this.$t('midjourney.description.upscale_2x'),
         [MidjourneyImagineAction.UPSCALE_4X]: this.$t('midjourney.description.upscale_4x'),
+        [MidjourneyImagineAction.UPSCALE_SUBTLE]: this.$t('midjourney.description.upscale_subtle'),
+        [MidjourneyImagineAction.UPSCALE_CREATIVE]: this.$t('midjourney.description.upscale_creative'),
         [MidjourneyImagineAction.REDO_UPSCALE_2X]: this.$t('midjourney.description.redo_upscale_2x'),
-        [MidjourneyImagineAction.REDO_UPSAMPLE_4X]: this.$t('midjourney.description.redo_upscale_4x'),
+        [MidjourneyImagineAction.REDO_UPSCALE_4X]: this.$t('midjourney.description.redo_upscale_4x'),
         [MidjourneyImagineAction.SQUARE]: this.$t('midjourney.description.square'),
         [MidjourneyImagineAction.PAN_LEFT]: this.$t('midjourney.description.pan_left'),
         [MidjourneyImagineAction.PAN_UP]: this.$t('midjourney.description.pan_up'),
@@ -197,19 +197,10 @@ export default defineComponent({
   },
   computed: {
     application() {
-      return this.applications?.find((application) => {
-        return application.id === this.modelValue?.request?.application_id;
-      });
+      return this.$store.state.midjourney.application;
     },
-    channel() {
-      // if (this.application?.api_id === API_ID_MIDJOURNEY_FAST) {
-      //   return MIDJOURNEY_CHANNEL_FAST;
-      // } else if (this.application?.api_id === API_ID_MIDJOURNEY_RELAX) {
-      //   return MIDJOURNEY_CHANNEL_RELAX;
-      // } else if (this.application?.api_id === API_ID_MIDJOURNEY_TURBO) {
-      //   return MIDJOURNEY_CHANNEL_TURBO;
-      // }
-      return undefined;
+    mode() {
+      return this.$store.state.midjourney.mode;
     }
   },
   methods: {
diff --git a/src/components/order/Create.vue b/src/components/order/Create.vue
index 6971d76c..67895e21 100644
--- a/src/components/order/Create.vue
+++ b/src/components/order/Create.vue
@@ -46,7 +46,7 @@
 </template>
 
 <script lang="ts">
-import { IApplication, IApplicationType, IOrderDetailResponse, IPackage, orderOperator } from '@/operators';
+import { IApplication, IApplicationType, IOrderDetailResponse, IPackage } from '@/models';
 import { defineComponent } from 'vue';
 import { ElMessage } from 'element-plus';
 import { ROUTE_CONSOLE_ORDER_DETAIL } from '@/router';
@@ -92,17 +92,12 @@ export default defineComponent({
   },
   computed: {
     price() {
-      if (this.application?.type === IApplicationType.API) {
-        if (this.application.api?.price && this.form.amount) {
-          return this.form.amount * this.application.api?.price;
-        }
-      }
       return 0;
     },
     package() {
       if (this.application?.type === IApplicationType.API) {
-        if (this.application.api?.packages && this.form.packageId) {
-          const filterPackages = this.application.api?.packages.filter((item) => item.id === this.form.packageId);
+        if (this.application.service?.packages && this.form.packageId) {
+          const filterPackages = this.application.service?.packages.filter((item) => item.id === this.form.packageId);
           if (filterPackages.length > 0) {
             return filterPackages[0];
           }
diff --git a/src/operators/midjourney.ts b/src/operators/midjourney.ts
index 483b9dc2..68d47fcc 100644
--- a/src/operators/midjourney.ts
+++ b/src/operators/midjourney.ts
@@ -70,7 +70,8 @@ class MidjourneyOperator {
     return await axios.post('/midjourney/imagine', data, {
       headers: {
         authorization: `Bearer ${options.token}`,
-        'content-type': 'application/json'
+        'content-type': 'application/json',
+        accept: 'application/json'
       },
       baseURL: BASE_URL_API
     });
diff --git a/src/pages/chatdoc/Conversation.vue b/src/pages/chatdoc/Conversation.vue
index 1be50844..6f7e1b18 100644
--- a/src/pages/chatdoc/Conversation.vue
+++ b/src/pages/chatdoc/Conversation.vue
@@ -57,13 +57,7 @@ import axios from 'axios';
 import { isJSONString } from '@/utils/is';
 import { Status } from '@/models';
 import ApplicationStatus from '@/components/application/Status.vue';
-import {
-  IApplication,
-  IChatdocChatResponse,
-  IChatdocConversation,
-  IChatdocMessageState,
-  IChatdocRepository
-} from '@/models';
+import { IChatdocChatResponse, IChatdocConversation, IChatdocMessageState, IChatdocRepository } from '@/models';
 
 export default defineComponent({
   name: 'ChatdocChat',
@@ -80,7 +74,6 @@ export default defineComponent({
       repositoryId: this.$route.params.repositoryId.toString(),
       answering: false,
       question: ''
-      // references: []
     };
   },
   computed: {
@@ -149,14 +142,12 @@ export default defineComponent({
       }, 0);
     },
     async onFetchAnswer() {
-      const token = this.application?.credential?.token;
-      const endpoint = this.application?.api?.endpoint;
-      const path = this.application?.api?.path;
+      const token = this.application?.credentials?.[0]?.token;
       const question = this.question;
       log(this.onFetchAnswer, 'validated', question);
       // reset question and references
       this.question = '';
-      if (!token || !endpoint || !question || !path) {
+      if (!token || !question) {
         console.error('no token or endpoint or question');
         this.messages.push({
           error: {
diff --git a/src/pages/chatdoc/Index.vue b/src/pages/chatdoc/Index.vue
index 310f3c10..fef26d37 100644
--- a/src/pages/chatdoc/Index.vue
+++ b/src/pages/chatdoc/Index.vue
@@ -92,7 +92,7 @@ export default defineComponent({
       return this.$store.state.chatdoc.repositories;
     },
     needApply() {
-      return this.$store.state.chatdoc.getApplicationStatus === Status.Success && !this.application;
+      return this.$store.state.chatdoc.status.getApplication === Status.Success && !this.application;
     },
     application() {
       return this.$store.state.chatdoc.application;
@@ -101,7 +101,7 @@ export default defineComponent({
       return this.$store.state.chatdoc.service;
     },
     initializing() {
-      return this.$store.state.chatdoc.getApplicationStatus === Status.Request;
+      return this.$store.state.chatdoc.status.getApplication === Status.Request;
     }
   },
   async mounted() {},
diff --git a/src/pages/midjourney/History.vue b/src/pages/midjourney/History.vue
index 43a15384..6811a717 100644
--- a/src/pages/midjourney/History.vue
+++ b/src/pages/midjourney/History.vue
@@ -83,18 +83,14 @@ export default defineComponent({
       await this.$store.dispatch('midjourney/getApplication');
     },
     async onStartTask(request: IMidjourneyImagineRequest) {
-      const token = this.application?.credential?.token;
-      const endpoint = this.application?.api?.endpoint;
-      const path = this.application?.api?.path;
-      if (!token || !endpoint || !path) {
-        console.error('no token or endpoint or question');
+      const token = this.application?.credentials?.[0]?.token;
+      if (!token) {
+        console.error('no token found');
         return;
       }
       midjourneyOperator
         .imagine(request, {
-          token,
-          endpoint,
-          path
+          token
         })
         .then(() => {
           ElMessage.success(this.$t('midjourney.message.startTaskSuccess'));
diff --git a/src/pages/midjourney/Index.vue b/src/pages/midjourney/Index.vue
index 808c4344..2a466992 100644
--- a/src/pages/midjourney/Index.vue
+++ b/src/pages/midjourney/Index.vue
@@ -5,10 +5,10 @@
     </template>
     <template #operation>
       <mode-selector class="mb-4" />
-      <api-status
+      <application-status
         :initializing="initializing"
         :application="application"
-        :api-id="channel.apiId"
+        :service="service"
         :need-apply="needApply"
         class="mb-4"
         @refresh="onGetApplications"
@@ -41,7 +41,7 @@ import { ElButton, ElMessage } from 'element-plus';
 import ModeSelector from '@/components/midjourney/ModeSelector.vue';
 import ReferenceImage from '@/components/midjourney/ReferenceImage.vue';
 import { applicationOperator, midjourneyOperator } from '@/operators';
-import ApiStatus from '@/components/common/ApiStatus.vue';
+import ApplicationStatus from '@/components/application/Status.vue';
 import TaskBriefList from '@/components/midjourney/tasks/TaskBriefList.vue';
 import FinalPrompt from '@/components/midjourney/FinalPrompt.vue';
 import { ERROR_CODE_DUPLICATION } from '@/constants/errorCode';
@@ -67,7 +67,7 @@ export default defineComponent({
     ElementsSelector,
     IgnoreSelector,
     ElButton,
-    ApiStatus,
+    ApplicationStatus,
     TaskBriefList,
     FinalPrompt,
     Layout
@@ -174,9 +174,7 @@ export default defineComponent({
       }
       midjourneyOperator
         .imagine(request, {
-          token,
-          endpoint,
-          path
+          token
         })
         .then(() => {
           ElMessage.success(this.$t('midjourney.message.startTaskSuccess'));
diff --git a/src/store/chatdoc/models.ts b/src/store/chatdoc/models.ts
index 9ee9328d..aa2c4a5b 100644
--- a/src/store/chatdoc/models.ts
+++ b/src/store/chatdoc/models.ts
@@ -7,6 +7,6 @@ export interface IChatdocState {
   status: {
     getService: Status;
     getApplication: Status;
-    getConversations: Status;
+    getRepositories: Status;
   };
 }
diff --git a/src/store/chatdoc/mutations.ts b/src/store/chatdoc/mutations.ts
index 0e8cb797..27a83521 100644
--- a/src/store/chatdoc/mutations.ts
+++ b/src/store/chatdoc/mutations.ts
@@ -1,4 +1,4 @@
-import { IApplication, IChatdocRepository } from '@/models';
+import { IApplication, IChatdocRepository, IService } from '@/models';
 import { IChatdocState } from './models';
 import { log } from '@/utils';
 
@@ -7,6 +7,10 @@ export const resetAll = (state: IChatdocState): void => {
   state.repositories = [];
 };
 
+export const setService = (state: IChatdocState, payload: IService): void => {
+  state.service = payload;
+};
+
 export const setApplication = (state: IChatdocState, payload: IApplication): void => {
   state.application = payload;
 };
@@ -54,6 +58,7 @@ export const setRepository = (state: IChatdocState, payload: IChatdocRepository)
 };
 
 export default {
+  setService,
   setApplication,
   setRepositories,
   setRepository,
diff --git a/src/store/chatdoc/state.ts b/src/store/chatdoc/state.ts
index 96bd31a2..33f2b99e 100644
--- a/src/store/chatdoc/state.ts
+++ b/src/store/chatdoc/state.ts
@@ -1,8 +1,15 @@
+import { Status } from '@/models';
 import { IChatdocState } from './models';
 
 export default (): IChatdocState => {
   return {
+    service: undefined,
     application: undefined,
-    repositories: undefined
+    repositories: undefined,
+    status: {
+      getService: Status.None,
+      getApplication: Status.None,
+      getRepositories: Status.None
+    }
   };
 };
diff --git a/src/store/index.ts b/src/store/index.ts
index afeb697e..8c514e80 100644
--- a/src/store/index.ts
+++ b/src/store/index.ts
@@ -12,9 +12,9 @@ import persistRoot from './common/persist';
 const store = createStore({
   ...root,
   modules: {
-    midjourney,
-    chat,
-    chatdoc
+    midjourney: midjourney,
+    chat: chat,
+    chatdoc: chatdoc
   },
   plugins: [
     createPersistedState({
diff --git a/src/store/midjourney/actions.ts b/src/store/midjourney/actions.ts
index 13c3f0a0..2e389378 100644
--- a/src/store/midjourney/actions.ts
+++ b/src/store/midjourney/actions.ts
@@ -30,17 +30,27 @@ export const setApplication = ({ commit }: any, payload: IApplication[]) => {
 
 export const getApplication = async ({
   commit,
+  state,
   rootState
 }: ActionContext<IMidjourneyState, IRootState>): Promise<IApplication> => {
   log(getApplication, 'start to get application for midjourney');
-  commit('setGetApplicationsStatus', Status.Request);
-  const { data: applications } = await applicationOperator.getAll({
-    user_id: rootState.user.id,
-    service_id: MIDJOURNEY_SERVICE_ID
+  return new Promise(async (resolve, reject) => {
+    state.status.getApplication = Status.Request;
+    applicationOperator
+      .getAll({
+        user_id: rootState.user.id,
+        service_id: MIDJOURNEY_SERVICE_ID
+      })
+      .then((response) => {
+        state.status.getApplication = Status.Success;
+        commit('setApplication', response.data.items[0]);
+        resolve(response.data.items[0]);
+      })
+      .catch((error) => {
+        state.status.getApplication = Status.Error;
+        reject(error);
+      });
   });
-  log(getApplication, 'get application for midjourney success', applications);
-  commit('setApplication', applications?.items?.[0]);
-  return applications.items?.[0];
 };
 
 export const setImagineTasks = ({ commit }: any, payload: IMidjourneyImagineTask[]) => {
@@ -66,19 +76,33 @@ export const getService = async ({ commit, state }: ActionContext<IMidjourneySta
 };
 
 export const getImagineTasks = async (
-  { commit, state, rootState }: ActionContext<IMidjourneyState, IRootState>,
+  { commit, state }: ActionContext<IMidjourneyState, IRootState>,
   { offset, limit }: { offset?: number; limit?: number }
 ): Promise<IMidjourneyImagineTask[]> => {
-  log(getImagineTasks, 'start to get imagine tasks', offset, limit);
-  commit('setGetImagineTasksStatus', Status.Request);
-  const { data: tasks } = (
-    await midjourneyOperator.tasks({
-      applicationId: state.application?.id
-    })
-  ).data;
-  // commit('setImagineTasksTotal', count);
-  commit('setImagineTasks', tasks);
-  return tasks;
+  return new Promise(async (resolve, reject) => {
+    log(getImagineTasks, 'start to get imagine tasks', offset, limit);
+    const application = state.application;
+    const token = application?.credentials?.[0]?.token;
+    if (!token) {
+      return reject('no token');
+    }
+    midjourneyOperator
+      .tasks(
+        {
+          applicationId: state.application?.id
+        },
+        {
+          token
+        }
+      )
+      .then((response) => {
+        commit('setImagineTasks', response.data);
+        resolve(response.data);
+      })
+      .catch((error) => {
+        return reject(error);
+      });
+  });
 };
 
 export default {
diff --git a/src/store/midjourney/mutations.ts b/src/store/midjourney/mutations.ts
index 6ba13742..a4d4fc3b 100644
--- a/src/store/midjourney/mutations.ts
+++ b/src/store/midjourney/mutations.ts
@@ -1,4 +1,4 @@
-import { IApplication, IMidjourneyMode, IMidjourneyPreset } from '@/models';
+import { IApplication, IMidjourneyMode, IMidjourneyPreset, IService } from '@/models';
 import { IMidjourneyState } from './models';
 
 export const resetAll = (state: IMidjourneyState): void => {
@@ -7,6 +7,10 @@ export const resetAll = (state: IMidjourneyState): void => {
   state.imagineTasks = [];
 };
 
+export const setService = (state: IMidjourneyState, payload: IService): void => {
+  state.service = payload;
+};
+
 export const setApplication = (state: IMidjourneyState, payload: IApplication): void => {
   state.application = payload;
 };
@@ -30,6 +34,7 @@ export const setImagineTasksTotal = (state: IMidjourneyState, payload: number):
 export default {
   setApplication,
   setPreset,
+  setService,
   setMode,
   setImagineTasks,
   setImagineTasksTotal,
diff --git a/src/store/midjourney/state.ts b/src/store/midjourney/state.ts
index 706e75a2..7d483278 100644
--- a/src/store/midjourney/state.ts
+++ b/src/store/midjourney/state.ts
@@ -1,12 +1,19 @@
 import { MIDJOURNEY_MODE_FAST } from '@/constants';
 import { IMidjourneyState } from './models';
+import { Status } from '@/models';
 
 export default (): IMidjourneyState => {
   return {
+    service: undefined,
     application: undefined,
     imagineTasks: undefined,
     imagineTasksTotal: undefined,
     preset: {},
-    mode: MIDJOURNEY_MODE_FAST
+    mode: MIDJOURNEY_MODE_FAST,
+    status: {
+      getService: Status.None,
+      getApplication: Status.None,
+      getImagineTasks: Status.None
+    }
   };
 };