LaiTool/src/main/setting/gptSetting.js

import path from 'path'
import os from 'os'
import { define } from '../../define/define'
import { errorMessage, successMessage } from '../Public/generalTools'
import axios from 'axios'
import { ServiceBase } from '../../define/db/service/serviceBase'
import { isEmpty } from 'lodash'
import { ValidateJson } from '../../define/Tools/validate'

export class GptSetting extends ServiceBase {
  constructor() {
    super()
    axios.defaults.baseURL = define.serverUrl
  }

  /**
   * 获取服务强上面的提示词相关的数据
   */
  async InitServerGptOptions() {
    try {
      // 获取提示词类型数据
      let promptRes = await axios.get('/api/Prompt/GetPromptType/100/1')
      let promptType = promptRes.data.data

      // 获取提示词数据
      let prompt = await axios.get('/api/Prompt/GetPromptString/all/100/1')
      let promptData = prompt.data.data

      let gptOptions = {
        promptType: promptType,
        promptData: promptData
      }
      return successMessage(gptOptions, '请求成功', 'GptSetting_InitServerGptOptions')
    } catch (error) {
      return errorMessage(
        '获取服务端提示词数据错误，错误信息如下：' + error.toString(),
        'GptSetting_InitServerGptOptions'
      )
    }
  }

  // 获取软件设置里面的AI设置，没有的话设置初始值
  async GetAISetting() {
    try {
      await this.InitService()
      let res = this.softService.GetSoftWarePropertyData('aiSetting')
      if (isEmpty(res)) {
        // 没有数据需要额外初始化
        res.data = {
          laiapi: {
            gpt_url: 'https://laitool.net/v1/chat/completions',
            api_key: undefined,
            model: undefined
          },
          kimi: {
            gpt_url: 'https://api.moonshot.cn/v1/chat/completions',
            api_key: undefined,
            model: undefined
          },
          doubao: {
            gpt_url: 'https://ark.cn-beijing.volces.com/api/v3/chat/completions',
            api_key: undefined,
            model: undefined
          }
        }
      } else {
        let tryP = ValidateJson(res)
        if (!tryP) {
          throw new Error('AI设置的数据格式不正确')
        }
        res = JSON.parse(res)
      }
      return successMessage(res, '请求成功', 'GptSetting_GetAISetting')
    } catch (error) {
      return errorMessage(
        '获取软件设置里面的AI设置错误，错误信息如下：' + error.toString(),
        'GptSetting_GetAISetting'
      )
    }
  }

  /**
   * 保存软件中的AI设置
   * @param {*} value
   */
  async SaveAISetting(value) {
    try {
      await this.InitService()
      let res = this.softService.SaveSoftwarePropertyData('aiSetting', JSON.stringify(value))
      return res
    } catch (error) {
      return errorMessage(
        '保存软件设置里面的AI设置错误，错误信息如下：' + error.toString(),
        'GptSetting_SaveAISetting'
      )
    }
  }
}