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> = (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 @@ /> - + @@ -114,10 +118,12 @@