From 2023f649533fd586175e7d952de9860aad8678e9 Mon Sep 17 00:00:00 2001
From: wangdan-fit2cloud <dan.wang@fit2cloud.com>
Date: Thu, 23 Nov 2023 14:30:28 +0800
Subject: [PATCH] =?UTF-8?q?feat:=20model=E5=BB=BA=E6=96=87=E4=BB=B6?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 ui/src/api/model.ts                           | 13 +++++++++-
 ui/src/api/type/model.ts                      |  6 +++++
 ui/src/stores/index.ts                        |  2 ++
 ui/src/stores/modules/model.ts                | 23 ++++++++++++++++
 .../views/application/CreateApplication.vue   | 26 ++++++++++++++++---
 5 files changed, 66 insertions(+), 4 deletions(-)
 create mode 100644 ui/src/api/type/model.ts
 create mode 100644 ui/src/stores/modules/model.ts

diff --git a/ui/src/api/model.ts b/ui/src/api/model.ts
index 9d665c32c..fb3962492 100644
--- a/ui/src/api/model.ts
+++ b/ui/src/api/model.ts
@@ -1,6 +1,17 @@
 import { Result } from '@/request/Result'
 import { get, post, del, put } from '@/request/index'
+import type { modelRequest } from '@/api/type/model'
 const prefix = '/model'
 const prefix_provider = '/provider'
 
-export default {}
+/**
+ * 获得模型列表
+ * @params 参数 name, model_type, model_name
+ */
+const getModel: (data?: modelRequest) => Promise<Result<any>> = (data) => {
+  return get(`${prefix}`, data)
+}
+
+export default {
+  getModel
+}
diff --git a/ui/src/api/type/model.ts b/ui/src/api/type/model.ts
new file mode 100644
index 000000000..e33c4bf32
--- /dev/null
+++ b/ui/src/api/type/model.ts
@@ -0,0 +1,6 @@
+interface modelRequest {
+  name: string
+  model_type: string
+  model_name: string
+}
+export type { modelRequest }
diff --git a/ui/src/stores/index.ts b/ui/src/stores/index.ts
index 0c33fe59f..bec30d953 100644
--- a/ui/src/stores/index.ts
+++ b/ui/src/stores/index.ts
@@ -4,11 +4,13 @@ export { store }
 import useUserStore from './modules/user'
 import useDatasetStore from './modules/dataset'
 import useParagraphStore from './modules/paragraph'
+import useModelStore from './modules/model'
 
 const useStore = () => ({
   user: useUserStore(),
   dataset: useDatasetStore(),
   paragraph:  useParagraphStore(),
+  model:  useModelStore(),
 })
 
 export default useStore
diff --git a/ui/src/stores/modules/model.ts b/ui/src/stores/modules/model.ts
new file mode 100644
index 000000000..7847d6de7
--- /dev/null
+++ b/ui/src/stores/modules/model.ts
@@ -0,0 +1,23 @@
+import { defineStore } from 'pinia'
+import modelApi from '@/api/model'
+import type { modelRequest } from '@/api/type/model'
+const useModelStore = defineStore({
+  id: 'model',
+  state: () => ({}),
+  actions: {
+    async asyncGetModel(data?: modelRequest) {
+      return new Promise((resolve, reject) => {
+        modelApi
+          .getModel(data)
+          .then((res) => {
+            resolve(res)
+          })
+          .catch((error) => {
+            reject(error)
+          })
+      })
+    }
+  }
+})
+
+export default useModelStore
diff --git a/ui/src/views/application/CreateApplication.vue b/ui/src/views/application/CreateApplication.vue
index 9f135b3cc..b5ca4942b 100644
--- a/ui/src/views/application/CreateApplication.vue
+++ b/ui/src/views/application/CreateApplication.vue
@@ -35,7 +35,11 @@
                 />
               </el-form-item>
               <el-form-item label="选择模型" prop="model_id">
-                <el-select v-model="applicationForm.model_id" placeholder="请选择模型" style="width: 100%;">
+                <el-select
+                  v-model="applicationForm.model_id"
+                  placeholder="请选择模型"
+                  style="width: 100%"
+                >
                   <el-option label="Zone one" value="shanghai" />
                   <el-option label="Zone two" value="beijing" />
                 </el-select>
@@ -114,10 +118,12 @@
   </LayoutContainer>
 </template>
 <script setup lang="ts">
-import { reactive, ref, watch } from 'vue'
+import { reactive, ref, watch, onMounted } from 'vue'
 import AiDialog from '@/components/ai-dialog/index.vue'
 import type { FormInstance, FormRules } from 'element-plus'
-import type { ApplicationFormType } from '@/api/application'
+import type { ApplicationFormType } from '@/api/type/application'
+import useStore from '@/stores'
+const { model } = useStore()
 
 const applicationFormRef = ref<FormInstance>()
 
@@ -147,6 +153,20 @@ const rules = reactive<FormRules<ApplicationFormType>>({
 watch(exampleList.value, () => {
   applicationForm.example = exampleList.value.filter((v) => v)
 })
+
+function getModel() {
+  loading.value = true
+  model.asyncGetModel()
+    .then((res) => {
+      loading.value = false
+    })
+    .catch(() => {
+      loading.value = false
+    })
+}
+onMounted(() => {
+  getModel()
+})
 </script>
 <style lang="scss" scoped>
 .create-application {