fix: 模型管理代码修改

zhangxiao

2024-08-15 f74f264d875b003730484f3fe7cb242c7f91294e

 src/views/dmx/model/components/addPageModel.vue

@@ -17,35 +17,33 @@
      :style="{ width: '90%', margin: '0 auto' }"
      layout="vertical"
    >
      <div v-if="isType !== 1">
        <a-form-item field="model_type" label="模型类型">
          <a-select v-model="form.model_type" placeholder="请选择">
            <a-option value="chat">chat</a-option>
            <a-option value="embedding">embedding</a-option>
          </a-select>
        </a-form-item>
        <a-form-item field="llm_name" label="模型名称">
          <a-input v-model="form.llm_name" placeholder="请输入模型名称" />
        </a-form-item>

        <!-- <a-form-item field="volc_ak" label="火山 ACCESS_KEY">
          <a-input v-model="form.volc_ak" placeholder="请输入火山 ACCESS_KEY" />
          <a-button
            type="primary"
            @click="handleClick"
            style="margin-left: 10px"
          >
            <template #icon>
              <icon-plus />
            </template>
          </a-button>
        </a-form-item>
        <a-form-item field="volc_sk" label="火山 SECRET_KEY">
          <a-input v-model="form.volc_sk" placeholder="请输入火山 SECRET_KEY" />
        </a-form-item> -->
        <a-form-item field="api_base" label="基础URL">
          <a-input v-model="form.api_base" placeholder="请输入基础URL" />
        </a-form-item>
      <div>
        <div v-for="(item, index) in form.formData.params" :key="index">
          <div v-if="item.value_type === 'select'">
            <a-form-item
              :field="item.key"
              :label="item.name"
              :required="item.required"
            >
              <a-select v-model="form[item.key]" placeholder="请选择">
                <a-option
                  v-for="(list, v) in item.options"
                  :value="list"
                  :label="v"
                ></a-option>
              </a-select>
            </a-form-item>
          </div>
          <div v-else-if="item.value_type === 'input'">
            <a-form-item
              :field="item.key"
              :label="item.name"
              :required="item.required"
            >
              <a-input v-model="form[item.key]" placeholder="请输入" />
            </a-form-item>
          </div>
        </div>
        <!-- <a-form-item
          field="raptor"
          label="是否支持 Vision"
@@ -54,17 +52,22 @@
          <a-switch v-model="form.raptor" />
        </a-form-item> -->
        <a-form-item field="logo" label="模型图片">
          <a-space direction="vertical" :style="{ width: '100%' }">
          <a-space
            direction="vertical"
            v-model="form.logo"
            :style="{ width: '100%' }"
          >
            <Upload
              :action="uploadAction"
              :limit="1"
              :url="form.urlS"
              @handleRemove="deleteRemove"
              @update:fileList="updateFileList"
              @success="handleSuccess"
            ></Upload>
          </a-space>
        </a-form-item>
        <a-form-item field="all_params" label="增加其它参数配置">
        <!-- <a-form-item field="all_params" label="增加其它参数配置">
          <a-button
            type="primary"
            @click="handleAddClick"
@@ -74,7 +77,7 @@
              <icon-plus />
            </template>
          </a-button>
        </a-form-item>
        </a-form-item> -->

        <a-form-item v-for="(item, index) in form.addFomList" :key="index">
          <div class="addInput">
@@ -98,26 +101,6 @@
          </div>
        </a-form-item>
      </div>
      <div v-else>
        <a-form-item field="name" label="API-Key">
          <a-input v-model="form.name" placeholder="API-Key" />
        </a-form-item>
        <a-form-item field="describe" label="Base-Url">
          <a-input v-model="form.name" placeholder="Base-Url" />
        </a-form-item>

        <a-form-item field="section" label="模型图片">
          <a-space direction="vertical" :style="{ width: '100%' }">
            <Upload
              :action="uploadAction"
              :limit="1"
              :url="form.urlS"
              @update:fileList="updateFileList"
              @success="handleSuccess"
            ></Upload>
          </a-space>
        </a-form-item>
      </div>

      <a-form-item>
        <div style="width: 100%; text-align: right">
@@ -133,13 +116,13 @@

<script lang="ts" setup>
  import { reactive, ref, computed, watch, watchEffect, onMounted } from 'vue';
  import { addLlm, getLlmDetail, editLlm } from '@/api/model';
  import { addLlm, getLlmDetail, editLlm, getFactoryDetail } from '@/api/model';
  import { Modal, Message } from '@arco-design/web-vue';
  import message from '@arco-design/web-vue/es/message';
  import { resolveUnref } from '@vueuse/core';
  import { useUserStore } from '@/store';
  import { userModelState } from '@/store';

  const userStore = useUserStore();
  const modelStore = userModelState();

  const addPageModelVisible = defineModel('show');
  const loading = ref(false);
@@ -168,10 +151,8 @@
  const form = reactive({
    size: 'medium',
    name: '',
    formData: { logo: '' },
    llm_factory: '',
    model_type: '',
    province: 'haidian',
    options: [],
    llm_name: '',
    api_base: '',
    volc_ak: '',
@@ -182,51 +163,26 @@
    slider: 5,
    score: 5,
    switch: false,
    multiSelect: ['section one'],
    treeSelect: '',
    raptor: false,
    addFomList: [],
    addFomList: <any[]>[],
  });
  const form_ref = ref(null);

  const form_ref = ref();

  const rules = {
    model_type: [
    logo: [
      {
        required: true,
        message: '请选择模型类型',
      },
    ],
    llm_name: [
      {
        required: true,
        message: '名称不允许为空',
      },
    ],
    volc_sk: [
      {
        required: true,
        message: '请输入volc_sk',
      },
    ],
    volc_ak: [
      {
        required: true,
        message: '请输入volc_ak',
      },
    ],
    api_base: [
      {
        required: true,
        message: '请输入api_base',
        message: '模型图片不能为空',
      },
    ],
  };

  const uploadAction = '/api/v1/llm/upload'; // 替换为你的上传API
  const fileList = ref([]);
  const imageUrls = ref([]);
  const imageUrls = ref<any[]>([]);
  const uploadUrl = ref([]);
  const httpUrl = localStorage.getItem('httpUrl');
  const httpUrl = modelStore.hrefUrl;

  const updateFileList = (newFileList) => {
    fileList.value = newFileList;
@@ -234,33 +190,34 @@

  const handleSuccess = (urls) => {
    uploadUrl.value = urls;
    form.logo = urls[0];

    const urlsArr = urls.map((url) => {
      return httpUrl + url;
    });
    imageUrls.value = urlsArr; // 拿到上传的图片地址
  };
  const deleteRemove = () => {
    form.logo = '';
    uploadUrl.value = [];
    imageUrls.value = [];
  };

  const emit = defineEmits(['refresh-parent']);
  const handleSubmit = ({ values, errors }) => {
    const all_params = form.addFomList.map((item) => {
      return item.name + ':' + item.value;
    });
    console.log(form, 9999);
    form_ref.value
      ?.validate()
      .then(async (res) => {
        if (!res) {
          form.formData.logo = uploadUrl.value[0] || '';
          form.llm_factory = editList.value.llm_factory;
          delete form.formData;
          if (props.title == '编辑') {
            console.log(imageUrls.value, 999);
            const data = await editLlm({
              llm_factory: editList.value.llm_factory,
              llm_name: form.llm_name,
              model_type: form.model_type,
              volc_ak: form.volc_ak,
              volc_sk: form.volc_sk,
              api_base: form.api_base,
              all_params: all_params,
              logo: uploadUrl.value[0],
            });
            const data = await editLlm(form);

            if ((data as any).code == 200) {
              message.success('添加成功');
@@ -270,16 +227,7 @@
              message.error('添加失败');
            }
          } else {
            const data = await addLlm({
              llm_factory: editList.value.llm_factory,
              llm_name: form.llm_name,
              model_type: form.model_type,
              volc_ak: form.volc_ak,
              volc_sk: form.volc_sk,
              api_base: form.api_base,
              all_params: all_params,
              logo: uploadUrl.value[0],
            });
            const data = await addLlm(form);

            if ((data as any).code == 200) {
              message.success('添加成功');
@@ -372,6 +320,15 @@

    imageUrls.value.push(httpUrl + data.data.logo);
  };
  const getFactory = async () => {
    const { data } = await getFactoryDetail({
      factory_name: editList.value.llm_factory,
    });
    form.formData = data;
  };
  onMounted(() => {
    getFactory();
  });

  watchEffect(() => {
    if (props.editList.name) {