App.vue 25.4 KB
Newer Older
6
UPDATE  
64104061f23fda247c679fa8 已提交
1
<template>
2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
  <div class="w-full h-screen">
    <!-- toolbar -->
    <div class="w-full h-14 pt-2 border-b-2 shadow-2xl text-center fixed bg-slate-200">
      <a href="#/llm-rep/app/">
        <svg xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 24 24" stroke-width="1.5" stroke="currentColor"
          class="w-4 h-4 absolute left-4 top-5">
          <path stroke-linecap="round" stroke-linejoin="round"
            d="M2.25 12l8.954-8.955c.44-.439 1.152-.439 1.591 0L21.75 12M4.5 9.75v10.125c0 .621.504 1.125 1.125 1.125H9.75v-4.875c0-.621.504-1.125 1.125-1.125h2.25c.621 0 1.125.504 1.125 1.125V21h4.125c.621 0 1.125-.504 1.125-1.125V9.75M8.25 21h8.25" />
        </svg>
      </a>
      <span class="leading-10 font-bold">{{ name }} {{ config?.mode === 'completions' ? '(无上下文)' : '' }}</span>
      <svg xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 24 24" stroke-width="1.5" stroke="currentColor"
        class="w-4 h-4  absolute right-4 top-5 cursor-pointer" @click="cleanHistory">
        <path stroke-linecap="round" stroke-linejoin="round"
          d="M14.74 9l-.346 9m-4.788 0L9.26 9m9.968-3.21c.342.052.682.107 1.022.166m-1.022-.165L18.16 19.673a2.25 2.25 0 01-2.244 2.077H8.084a2.25 2.25 0 01-2.244-2.077L4.772 5.79m14.456 0a48.108 48.108 0 00-3.478-.397m-12 .562c.34-.059.68-.114 1.022-.165m0 0a48.11 48.11 0 013.478-.397m7.5 0v-.916c0-1.18-.91-2.164-2.09-2.201a51.964 51.964 0 00-3.32 0c-1.18.037-2.09 1.022-2.09 2.201v.916m7.5 0a48.667 48.667 0 00-7.5 0" />
      </svg>
    </div>
    <!-- main content -->
    <div class="flex w-full h-full pt-14">
      <!-- left  -->
22 23 24
      <div class="flex-none w-0 md:w-[260px] bg-slate-200 border-r-2 hidden md:block p-8 text-sm">

        <t-form labelAlign="top">
25
          <t-form-item label="GPT服务器类型" name="sdServerType">
26
            <t-select v-model="sdServerType" @change="handleSDServerTypeChange" class="bg-gray-200" style="background: #ccc;font-size:12px;">
27 28 29 30 31 32
              <t-option v-for="(item, index) in sdServerTypeOptions" :key="index" :value="item.value" :label="item.label">
                {{ item.label }}
              </t-option>
            </t-select>
          </t-form-item>
          <t-form-item label="服务器地址" name="sdServerUrl">
33
            <t-textarea placeholder="请输入内容" v-model="sdServerUrl" :autosize="true" @change="saveConfig"/>
34 35 36 37 38
          </t-form-item>
        </t-form>
        <t-card class="bg-orange-100 mt-8 text-gray-600 text-sm" v-if="sdServerType === 'common'">
          公共服务器为临时提供的测试服务器,可能存在需要排队或随时下线的可能。<br/>
          建议使用
39
          <t-tooltip content='购买后上面服务器地址填写 工作台-小羊驼 域名,同时把端口换成8000,并添加"/v1"后缀' theme="primary">
40 41 42 43 44
            <t-link href="https://inscode.csdn.net/gpu?utm_source=sd_app" target="_blank" :suffix-icon="renderSuffixIcon" underline>私有服务器</t-link>
          </t-tooltip>
        </t-card>


45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65
        <t-form labelAlign="top" label-width="100" :style="{display: showProfileSetting ? 'block':'none'}">
            <t-form-item label="机器人头像">
              <!-- <t-input placeholder="请输入内容" v-model="config.robot_img"/> -->
              <t-space direction="vertical" align="center" v-for="imgItem in robotAvatarList" class="mr-2 cursor-pointer hover:bg-blue-600">
                <t-image class="rounded-lg border-2 " :class="{ 'border-blue-800': imgItem.src === config.robot_img }"
                  @click="changeRobotAvatar(imgItem.src)" :src="imgItem.src" fit="cover"
                  :style="{ width: '40px', height: '40px' }" />
                <!-- <span>{{ imgItem.name }}</span> -->
              </t-space>
            </t-form-item>
            <t-form-item label="你的头像">
              <t-space direction="vertical" align="center" v-for="imgItem in userAvatarList" class="mr-2 cursor-pointer hover:bg-blue-600">
                <t-image class="rounded-lg border-2 " :class="{ 'border-blue-800': imgItem.src === config.user_img }"
                  @click="changeUserAvatar(imgItem.src)" :src="imgItem.src" fit="cover"
                  :style="{ width: '40px', height: '40px' }" />
                <!-- <span>{{ imgItem.name }}</span> -->
              </t-space>
            </t-form-item>
            <t-form-item label="机器人对你的称呼" help="多个称呼用逗号隔开">
              <t-input v-model="config.user_call_name" />
            </t-form-item>
66
        </t-form>
67 68 69
      </div>
      <!-- right -->
      <div class="flex-auto bg-slate-300 md:bg-slate-300 p-2 sm:p-8">
70 71 72

        <content/>

73
        <div class="w-full bg-slate-200 h-full m-auto relative container max-w-6xl rounded-xl">
74
          <div class="w-full h-full pb-24 p-4 overflow-y-auto overflow-x-hidden" ref="messageList">
75
            <div v-for="(item,index) in message" class="mb-8">
76 77 78 79
              <div v-if="item.user === 'User'">
                <div class="flex flex-row-reverse">
                  <t-image class="rounded-lg" v-if="config.user_img" :src="config.user_img" fit="cover"
                    :style="{ width: '40px', height: '40px' }"></t-image>
80
                  <div class="bg-green-400 text-gray-700  p-4 mx-2 w-fit max-w-2xl 2xl:max-w-4xl rounded-lg cursor-pointer text-sm" @click="copyAsPrompt(item.message)" v-html="item.message">
81
                   
82 83
                  </div>
                </div>
6
UPDATE  
64104061f23fda247c679fa8 已提交
84

85 86 87 88 89
              </div>
              <div v-else class="flex">
                <t-image class="rounded-lg" v-if="config.robot_img" :src="config.robot_img" fit="cover"
                  :style="{ width: '40px', height: '40px' }"></t-image>
                <div class="bg-slate-50 text-gray-700 p-4 mx-2 w-fit max-w-2xl 2xl:max-w-4xl rounded-lg text-wrapper">
90 91
                  <template v-if="item?.viewMode??'chat' === 'image'">
                    <img v-if="item.img" :src="item.img"/>
6
622aa39c1f9b166ab1a38c05 已提交
92
                    <img v-else src="/loading.gif" class="w-[512px]"/>
93
                  </template>
94 95 96 97 98
                  <template v-else-if="item.message && item.message.length > 5">
                    <p class="text-sm">
                      {{ item.message }}
                    </p>
                    <!-- <markdown-it-vue class="md-body" content="123" :options="mdOptions" :key="index" v-if="item.message"/> -->
99 100

                  </template>
W
weixin_44463441 已提交
101
                  <p class="text-sm" v-else>思考中...</p>
102
                  
103 104 105
                </div>
              </div>
            </div>
106
            <div class="text-center" v-if="mode !== 'draw'">
107 108 109 110 111 112 113 114 115 116
              <div v-if="loading"  @click="stop" class="-mt-4 px-5 py-1 m-auto w-fit cursor-pointer hover:bg-slate-300 border-dotted border-[1px] border-slate-400 rounded-md">
                <svg xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 24 24" stroke-width="1.5" stroke="currentColor" class="animate-spin w-4 h-4 inline-block">
                  <path stroke-linecap="round" stroke-linejoin="round" d="M21 12a9 9 0 11-18 0 9 9 0 0118 0z" />
                  <path stroke-linecap="round" stroke-linejoin="round" d="M9 9.563C9 9.252 9.252 9 9.563 9h4.874c.311 0 .563.252.563.563v4.874c0 .311-.252.563-.563.563H9.564A.562.562 0 019 14.437V9.564z" />
                </svg>
                <span class="ml-2 text-[12px]">停止</span>
              </div>
             
            </div>
          </div>
117
          <div class="w-full absolute bottom-0 bg-slate-100 h-22 py-2 px-2 rounded flex flex-col  text-sm">
118 119
            
            <div class="w-full h-10 flex">
W
weixin_44463441 已提交
120
                <input v-model="prompt" class="flex-auto ring-0	outline-0	pl-2 bg-slate-200 h-10 border-0" v-on:keyup.enter="query" placeholder="提示词" />
121

122
                <t-dropdown :options="options" @click="clickHandler" class="ml-4 mt-1" v-if="mode === 'draw'">
123 124 125 126 127
                  <t-button theme="default" variant="text" shape="square"> 
                    <t-icon :name="modeImage" size="24" class="w-6 h-6 text-cyan-800 font-bold hover:text-cyan-600 cursor-pointer"/>
                  </t-button>
                </t-dropdown>
              
128
                <!-- <svg xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 24 24" stroke-width="1.5" stroke="currentColor" 
129 130 131 132
                @click="showMoreSetting=!showMoreSetting"
                class="w-6 mx-2 ml-4 mt-2 h-6 flex-none text-cyan-800 font-bold hover:text-cyan-600 cursor-pointer">
                  <path stroke-linecap="round" stroke-linejoin="round" d="M10.343 3.94c.09-.542.56-.94 1.11-.94h1.093c.55 0 1.02.398 1.11.94l.149.894c.07.424.384.764.78.93.398.164.855.142 1.205-.108l.737-.527a1.125 1.125 0 011.45.12l.773.774c.39.389.44 1.002.12 1.45l-.527.737c-.25.35-.272.806-.107 1.204.165.397.505.71.93.78l.893.15c.543.09.94.56.94 1.109v1.094c0 .55-.397 1.02-.94 1.11l-.893.149c-.425.07-.765.383-.93.78-.165.398-.143.854.107 1.204l.527.738c.32.447.269 1.06-.12 1.45l-.774.773a1.125 1.125 0 01-1.449.12l-.738-.527c-.35-.25-.806-.272-1.203-.107-.397.165-.71.505-.781.929l-.149.894c-.09.542-.56.94-1.11.94h-1.094c-.55 0-1.019-.398-1.11-.94l-.148-.894c-.071-.424-.384-.764-.781-.93-.398-.164-.854-.142-1.204.108l-.738.527c-.447.32-1.06.269-1.45-.12l-.773-.774a1.125 1.125 0 01-.12-1.45l.527-.737c.25-.35.273-.806.108-1.204-.165-.397-.505-.71-.93-.78l-.894-.15c-.542-.09-.94-.56-.94-1.109v-1.094c0-.55.398-1.02.94-1.11l.894-.149c.424-.07.765-.383.93-.78.165-.398.143-.854-.107-1.204l-.527-.738a1.125 1.125 0 01.12-1.45l.773-.773a1.125 1.125 0 011.45-.12l.737.527c.35.25.807.272 1.204.107.397-.165.71-.505.78-.929l.15-.894z" />
                  <path stroke-linecap="round" stroke-linejoin="round" d="M15 12a3 3 0 11-6 0 3 3 0 016 0z" />
133
                </svg> -->
134

135 136 137 138 139 140 141
                <svg xmlns="http://www.w3.org/2000/svg" fill="none" viewBox="0 0 24 24" stroke-width="1.5"
                  stroke="currentColor"
                  class="w-6 mx-2 ml-4 mt-2 h-6 flex-none text-cyan-800 font-bold hover:text-cyan-600 cursor-pointer"
                  @click="query">
                  <path stroke-linecap="round" stroke-linejoin="round"
                    d="M6 12L3.269 3.126A59.768 59.768 0 0121.485 12 59.77 59.77 0 013.27 20.876L5.999 12zm0 0h7.5" />
                </svg>
142
            </div>
143
            <div class="w-full h-10 mt-2 flex gap-2" v-if="mode === 'draw'">
144
                <input v-model="negative_prompt" class="flex-auto ring-0	outline-0	pl-2 bg-slate-200 h-10" v-on:keyup.enter="query" placeholder="逆向提示词" />
145 146 147 148 149 150 151
                <div class="flex-none w-40 ring-0	outline-0	pl-2  h-10 flex justify-between">
                  <input v-model="width" class="ring-0	outline-0	pl-2 bg-slate-200 h-10 w-16" placeholder="width"/> 
                  <span class="h-10 leading-10">*</span>
                  <input v-model="height" class="ring-0	outline-0	pl-2 bg-slate-200 h-10 w-16" placeholder="height"/>
                </div>
                <input v-model="steps" class="flex-none w-30 ring-0	outline-0	pl-2 bg-slate-200 h-10 w-16" placeholder="steps"/>
                <input v-model="sampler" class="flex-none w-40 ring-0	outline-0	pl-2 bg-slate-200 h-10 w-16 text-small" placeholder="sampler"/>
152
            </div>
153 154 155
          </div>
        </div>
      </div>
6
UPDATE  
64104061f23fda247c679fa8 已提交
156 157
    </div>

158
  </div>
6
UPDATE  
64104061f23fda247c679fa8 已提交
159
</template>
160
<script>
6
UPDATE  
64104061f23fda247c679fa8 已提交
161

162 163 164
// import { Configuration, OpenAIApi } from "openai"
import OpenAI from './js/openai.js'
import Config from './js/config.js'
165 166
import InsCodeAI from './js/inscodeai.js'

167
import SDApi from './js/sd.js'
168
import StorageApi from './js/storage.js'
169
import PodsApi from './js/pod.js'
170
import RepEs from './js/repEs.js'
171 172
import { MessagePlugin } from 'tdesign-vue-next';

173 174
import Content from './components/content/index.vue'

6
UPDATE  
64104061f23fda247c679fa8 已提交
175

176 177 178
// import MarkdownItVue from 'markdown-it-vue'
// import 'markdown-it-vue/dist/markdown-it-vue.css'
// https://github.com/ravenq/markdown-it-vue
179 180
import MarkdownItVue from 'markdown-it-vue'
import 'markdown-it-vue/dist/markdown-it-vue.css'
6
UPDATE  
64104061f23fda247c679fa8 已提交
181

182
// const DataServe = new Data()
6
UPDATE  
64104061f23fda247c679fa8 已提交
183

184 185 186
export default {
  name: 'LlmApp',
  components: {
187
    MarkdownItVue, Content
188 189 190 191 192 193 194
  },
  data() {
    return {
      id: 0,
      name: '加载中...',
      userAvatarList: [],
      robotAvatarList: [],
195
      mode: 'chat',
196
      modeImage: 'image', // image chat chart-bubble
197 198 199 200 201 202 203 204 205 206 207
      message: [
        // {"user": "User", "message": "创建一个用户表,要求分区"},
        // {"user": "AI", "message": "在GaussDB数据库中创建一个带有分区的用户表可以使用以下语句:在GaussDB数据库中创建一个带有分区的用户表可以使用以下语句:在GaussDB数据库中创建一个带有分区的用户表可以使用以下语句:在GaussDB数据库中创建一个带有分区的用户表可以使用以下语句:"},
        // {"user": "User", "message": "create table user(id int primary key, name varchar)将这个建表语句翻译成GaussDB建表语句create table user(id int primary key, name varchar)将这个建表语句翻译成GaussDB建表语句create table user(id int primary key, name varchar)将这个建表语句翻译成GaussDB建表语句create table user(id int primary key, name varchar)将这个建表语句翻译成GaussDB建表语句create table user(id int primary key, name varchar)将这个建表语句翻译成GaussDB建表语句"},
        // {"user": "AI", "message": "在GaussDB数据库中,创建一个带有主键约束的用户表可以使用以下语句:"},
      ],
      client: null,
      indexClient: null,
      prompt: '',
      config: {
      },
208 209 210 211 212
      sdServerType: 'common',
      sdServerTypeOptions: [
        { label: '公共服务器', value: 'common' },
        { label: '私有服务器', value: 'private' },
      ],
213
      sdServerUrl: '',
214 215 216 217 218
      loading: false,
      maxHistory: 200,
      speaking: false, // 是否正在输出语音
      autoSpeech: false, // 是否自动开始语音
      speech: null, // 语音object
219 220 221 222 223
      options: [
        { content: '绘图模式', value: 'draw' },
        { content: '聊天模式', value: 'chat' },
        { content: '自动模式', value: 'auto' },
      ],
224
      showProfileSetting: false, //是否显示角色设置模块
225 226 227 228
      width: 512,
      height: 512,
      steps: 20,
      sampler: 'DPM++ SDE Karras',
229
      showMoreSetting: false,
230
      default_prompt: '1girl, (ulzzang-6500:0.7), kpop idol, yae miko, detached sleeves, bare shoulders, pink hair, long hair, cleavage,japanese clothes,breast,best quality, (painting:1.5), (hair ornament:1.35), jewelry, purple eyes, earrings, breasts, torii, cherry blossoms, lantern light, depth of field, detailed face, face focus, ribbon_trim, (looking at viewer:1.25), nontraditional miko, shiny skin, long sleeves, smile, thick lips, game cg,  east asian architecture, (blurry background:1.2), sitting, upper body <lora:YaeMiko_Test:0.45>,<lora:mix4:0.5>',
6
622aa39c1f9b166ab1a38c05 已提交
231
      negative_prompt: 'nfsw, bright lantern, brightness, (nipples:1.2), pussy, EasyNegative, (worst quality:2), (low quality:2), (normal quality:2), lowres, normal quality, ((monochrome)), ((grayscale)), skin spots, acnes, skin blemishes, age spot, glans,extra fingers, fewer fingers, strange fingers, bad hand, bare thighs,hand,bad finger',
232 233 234 235
      history: {
        name: 'ai',
        key: 'history',
      },
236 237 238 239 240 241 242 243 244 245 246
      mdOptions: {
        markdownIt: {
          linkify: true
        },
        linkAttributes: {
          attrs: {
            target: '_blank',
            rel: 'noopener'
          }
        }
      }
247 248 249 250 251 252 253 254 255 256 257 258 259 260 261 262 263 264 265 266 267 268 269 270 271 272 273 274 275 276 277 278 279 280 281 282 283 284 285
    }
  },
  methods: {
    hideLeftMenu() {

    },
    changeUserAvatar(imgSrc) {
      this.config.user_img = imgSrc
    },
    changeRobotAvatar(imgSrc) {
      this.config.robot_img = imgSrc
    },
    getAvatar() {
      let that = this
      DataServe.find('t_llm_rep_chat_img', { resoure_sub_type: 'user_avatar' }).then(res => {
        if (res && res.status === 200 && res.data && res.data.code === 200 && res.data.data) {
          const data = res.data.data
          for (let item of data) {
            that.userAvatarList.push({
              src: JSON.parse(item.ext).img_src,
              name: item.name
            })
          }
        }
      })

      DataServe.find('t_llm_rep_chat_img', { resoure_sub_type: 'robot_avatar' }).then(res => {
        if (res && res.status === 200 && res.data && res.data.code === 200 && res.data.data) {
          const data = res.data.data
          for (let item of data) {
            that.robotAvatarList.push({
              src: JSON.parse(item.ext).img_src,
              name: item.name
            })
          }
        }
      })
    },
    initAIClient () {
286 287
      // this.client = new OpenAI(this.config)
      this.client = new InsCodeAI(this.config)
288 289 290 291 292 293 294 295 296 297 298 299 300 301 302
    },
    initIndexClient () {
      if (this.config?.index_url) {
        const index_type = this.config?.index_type
        //confluence
        if (index_type === 'es') {
          this.indexClient = new RepEs(this.config)
        } else if(index_type === 'confluence') {
          this.indexClient = new RepConfluence(this.config)
        } else {
          this.indexClient = new RepSimple(this.config)
        }
      } 
    },
    query() {
6
622aa39c1f9b166ab1a38c05 已提交
303
      
304 305 306 307 308 309 310 311 312 313 314
      if (this.loading) {
        MessagePlugin.warning({ content: '正在执行中,请稍等!', placement: 'center' })
        return
      }
      if (this.prompt === '') {

        MessagePlugin.warning({ content: '提示词不能为空!', placement: 'center' })
        console.info('提示词不能为空')
        return
      }
      
315 316 317
      this.loading = true


318 319
      if (this.mode === 'draw') {
        this.draw()
320
      } else {
321

322 323 324 325 326 327 328 329 330 331 332 333

        const _this = this
        const messages = this.message
        const currentMsg = {
          "user": "AI", "message": ''
        }
        const newPrompt = this.prompt
        this.prompt = ''
        messages.push({ "user": "User", "message": newPrompt })
        messages.push(currentMsg)
        _this.$refs.messageList.scrollTop = _this.$refs.messageList.scrollHeight;
        
334
        if (this.indexClient) {
335 336
          this.indexClient.query(newPrompt).then(res => {
            this.getAnswer(res, newPrompt, currentMsg)
337
          }).catch((err) => {
338
            this.getAnswer(null, newPrompt, currentMsg)
339
          })
340 341
        }else {
          this.getAnswer(null, newPrompt, currentMsg)
342
        }
343
      }
344 345 346 347 348 349 350 351 352
      // if (this.indexClient) {
      //   this.indexClient.query(this.prompt).then(res => {
      //     this.getAnswer(res)
      //   }).catch((err) => {
      //     this.getAnswer()
      //   })
      // } else {
      //   this.getAnswer()
      // }
6
UPDATE  
64104061f23fda247c679fa8 已提交
353

354 355 356 357 358 359 360
    },
    scrollBottom () {
      const _this = this
      this.$nextTick(() => {
        _this.$refs.messageList.scrollTop = _this.$refs.messageList.scrollHeight + 20;
      })
      
361
    },
362 363
    draw () {

6
622aa39c1f9b166ab1a38c05 已提交
364
      this.loading = true
365 366 367 368
      const currentMsg = {
        "user": "AI", "message": '', img: null, viewMode: 'image'
      }
      const query = this.prompt
369
      const negative_prompt = this.negative_prompt
370 371
      const _this = this
      const messages = this.message
372 373 374 375 376
      const content = 'prompt: <br/>' + 
                        this.prompt + 
                        (negative_prompt ? '<br/><br/>' + 'negative_prompt: </br>' + negative_prompt : '') + '<br/><br/>' +
                        'width: ' + this.width + ' height: ' + this.height + '<br/>steps: ' + this.steps +'<br/>sampler: ' + this.sampler 
      messages.push({ "user": "User", "message": content })
377 378 379

      messages.push(currentMsg)

380
      // this.prompt = ''
381
      this.scrollBottom()
382 383 384 385 386
      const config = JSON.parse(JSON.stringify(this.config)) 
      config.width = this.width
      config.height = this.height
      config.steps = this.steps
      config.sampler = this.sampler
387
      config.sd_api = this.sdServerUrl
388 389

      SDApi.draw(config, query, negative_prompt).then(res => {
390 391

        _this.loading = false
392
        
393 394 395 396 397
        console.info('close')
        currentMsg.img = res

        messages.splice(messages.length - 1, 1)
        messages.push(currentMsg)
398
        _this.scrollBottom()
399
        _this.saveHistory()
400 401
        

402 403 404 405
      }).catch(err => {
        _this.loading = false
      })
    },
406 407 408 409
    getAnswer(context = null, prompt = null, currentMsg = null) {


      const newPrompt = prompt ? prompt : this.prompt
410
      const _this = this
411
      this.config.api_url = this.sdServerUrl
412
      const messages = this.message
413
      this.scrollBottom()
414 415 416 417 418 419 420
      this.client.createCompletion(
        newPrompt,
        this.message,
        context,
        {
          onmessage: (msg, isPart) => {
            // _this.set(_this.message, id, currentMsg)
421
            // console.info(msg)
422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437 438 439 440 441 442 443 444 445 446 447 448 449
            if (isPart) {
              currentMsg.message += msg
            } else {
              currentMsg.message = msg
            }

            messages.splice(messages.length - 1, 1)
            messages.push(currentMsg)
            _this.$refs.messageList.scrollTop = _this.$refs.messageList.scrollHeight;
            // console.info(currentMsg.message)
          },
          onclose: () => {
            _this.loading = false
            _this.saveHistory()
            console.info('close')
            if(_this.autoSpeech) {
              _this.speak()
            }
          },
          onerror: (err) => {
            _this.loading = false
            console.info(err)
          }
        })
    },
    getAppInfo() {


450
      const data = Config.getData().data
451 452 453
      this.name = data.name
      document.title = this.name
      if (data.ext) {
454
        const config = data.ext
455
        
456
        this.config = config
W
weixin_44463441 已提交
457
        
458 459 460 461
        console.info(config)
      }
    },
    recoveryHistory() {
462

463
      StorageApi.get(this.history.name).then(data => {
464
        if (data && data.data) {
465 466 467 468 469
          const historyMessage = JSON.parse(data.data)
          if (historyMessage !== null && historyMessage.length > 0) {
            this.message = historyMessage
          }
          
470 471
        }
      })      
472 473 474 475 476 477 478 479 480
    },
    saveHistory() {
      if (this.message.length > this.maxHistory) {
        const tmpHistory = []
        const start = this.message.length - this.maxHistory
        const end = this.message.length
        for (let id = start; id < end; id++) {
          tmpHistory.push(this.message[id])
        }
481
        StorageApi.put(this.history.name, JSON.stringify(tmpHistory))
482
      } else {
483
        StorageApi.put(this.history.name, JSON.stringify(this.message))
484 485 486 487 488 489 490 491 492 493 494 495 496 497 498 499 500 501 502 503 504 505 506 507 508 509 510 511 512 513 514 515
      }

    },
    cleanHistory() {
      this.message = []
      this.saveHistory()
    },
    initSpeaker () {
      window.speechSynthesis.cancel()
      this.speech = new SpeechSynthesisUtterance();
      this.speech.onend = () => {
        console.log("语音播报结束")
        this.speaking = false
      }
    },
    speak () {
       this.speaking = true
        const content = this.message[this.message.length - 1].message
        if (content && content.length > 1) {
          // var utterThis = new SpeechSynthesisUtterance(text);
          this.speech.text = content;
          console.info('speak ' + content)
          // speech.lang = 'zh';//汉语
          window.speechSynthesis.speak(this.speech);
        }
        
    },
    stop () {
      this.client.close()
    },
    copyAsPrompt (message) {
      this.prompt = message
516
    },
517
    handleSDServerTypeChange (type) {
518 519
      
      if (type === 'common') {
520
        
521
        this.sdServerUrl = this.config.common_gpt_url
522
      }
523
      this.saveConfig()
524 525
    },
    recoverConfig () {
526
      const config = localStorage.getItem('ai-docs-config')
527
      
528 529 530 531
      if (config) {
        const configJSON = JSON.parse(config)
        this.config = configJSON
        this.sdServerUrl = configJSON.sdServerUrl
532 533 534 535 536
        this.sdServerType = configJSON.sdServerType??'common'

        const data = Config.getData().data
        this.name = data.name
        document.title = this.name
6
622aa39c1f9b166ab1a38c05 已提交
537

538
        
W
weixin_44463441 已提交
539 540 541 542 543
        if (data.ext) {
          if (data.ext.common_gpt_url && this.sdServerType === 'common') {
            this.config.common_gpt_url = data.ext.common_gpt_url
            this.sdServerUrl = data.ext.common_gpt_url
          }
544
          
W
weixin_44463441 已提交
545 546 547
          if (data.ext.api_prompt_prefix) {
            this.config.api_prompt_prefix = data.ext.api_prompt_prefix
          }
548
          this.config.prompt_template = data.ext.prompt_template
W
weixin_44463441 已提交
549
          this.config.welcome_text = data.ext.welcome_text
W
weixin_44463441 已提交
550
          this.config.temperature = data.ext?.temperature??0.7
551
        }
W
weixin_44463441 已提交
552
        
6
622aa39c1f9b166ab1a38c05 已提交
553

554
      } else {
555
        
556
        this.getAppInfo()
557
        this.sdServerUrl = this.config.common_gpt_url
558
        this.sdServerType = 'common'
559 560 561 562
        
      }
    },
    saveConfig () {
563
      
564 565 566
      const config = JSON.parse(JSON.stringify(this.config))
      config.sdServerType = this.sdServerType
      config.sdServerUrl = this.sdServerUrl
567
      config.sdServerType = this.sdServerType
568
      localStorage.setItem('ai-docs-config', JSON.stringify(config))
569 570 571 572 573 574 575
    },
    getPods () {

      PodsApi.getPods().then(res => {


      })
W
weixin_44463441 已提交
576 577
    },
    initChat() {
578
      
W
weixin_44463441 已提交
579 580 581 582 583 584 585 586 587 588 589 590 591 592
      this.initAIClient()
      this.initIndexClient()
      const config = this.config

      if (config.default_prompt) {
        this.prompt = config.default_prompt
      }
      if (config?.auto_speech??false) {
        this.autoSpeech = true
        this.initSpeaker()
      }
      if (config?.welcome_text && config?.welcome_text.length > 0 && this.message.length === 0) {
        this.message.push({
          "user": "AI", 
593 594
          "message": config.welcome_text,
          "ignore": true,
W
weixin_44463441 已提交
595 596 597 598
        })
      }
      this.showProfileSetting = config?.show_profile_setting??false

599 600 601 602
    }
  },
  mounted() {
    this.id = this.$route?.params?.id
603 604 605
    StorageApi.init(this.history.name).then(res => {
      this.recoveryHistory()
    })
606
    this.recoverConfig()
607
    // this.getAvatar()
608
    if (this.mode === 'chat') {
W
weixin_44463441 已提交
609
      this.initChat()
610
    }
611
  } 
612 613 614 615 616 617 618 619 620 621 622

};
</script>


<style>
.text-wrapper {
  white-space: pre-wrap;
}
.vuepress-markdown-body {
  padding: 16px !important;
6
UPDATE  
64104061f23fda247c679fa8 已提交
623 624
}
</style>
625