model-config.vue 28 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819
  1. <template>
  2. <template v-if="state.indexMethod.show">
  3. <div class="__czr-title_1 mb-[0.5rem]">索引方式</div>
  4. <div class="flex w-full gap-[1rem]">
  5. <div
  6. class="__hover h-[8.38rem] flex-1 rounded-[var(--czr-gap)] bg-[#ffffff] px-[1.5rem] py-[1rem]"
  7. style="border: var(--czr-border); border-width: 0.13rem"
  8. :class="{ 'index-method-active': state.indexMethod.value === '高质量' }"
  9. @click="() => (state.indexMethod.value = '高质量')"
  10. >
  11. <div class="flex items-center">
  12. <img
  13. src="@/assets/images/model-icon-1.png"
  14. class="mr-[var(--czr-gap)] h-[3.25rem] w-[3.25rem]"
  15. />
  16. <div class="text-[1.25rem] font-bold text-[#2E3238]">高质量</div>
  17. <img
  18. src="@/assets/images/model-icon-3.png"
  19. class="h-[1.27rem] w-[2.88rem]"
  20. />
  21. </div>
  22. <div
  23. class="mt-[0.5rem] text-[0.88rem] text-[#606266]"
  24. style="line-height: 1.3rem"
  25. >
  26. 调用嵌入模型来处理文档以实现更精确的检索,可以帮助大语言模型生成高质量的回答。
  27. </div>
  28. </div>
  29. <!-- <div-->
  30. <!-- class="__hover h-[8.38rem] flex-1 rounded-[var(&#45;&#45;czr-gap)] bg-[#ffffff] px-[1.5rem] py-[1rem]"-->
  31. <!-- style="border: var(&#45;&#45;czr-border); border-width: 0.13rem"-->
  32. <!-- :class="{ 'index-method-active': state.indexMethod.value === 2 }"-->
  33. <!-- @click="() => state.indexMethod.value = 2"-->
  34. <!-- >-->
  35. <!-- <div class="flex items-center">-->
  36. <!-- <img-->
  37. <!-- src="@/assets/images/model-icon-2.png"-->
  38. <!-- class="mr-[var(&#45;&#45;czr-gap)] h-[3.25rem] w-[3.25rem]"-->
  39. <!-- />-->
  40. <!-- <div class="text-[1.25rem] font-bold text-[#2E3238]">经济</div>-->
  41. <!-- </div>-->
  42. <!-- <div-->
  43. <!-- class="mt-[0.5rem] text-[0.88rem] text-[#606266]"-->
  44. <!-- style="line-height: 1.3rem"-->
  45. <!-- >-->
  46. <!-- 每个块使用 10 个关键词进行检索,不消耗 tokens,但会降低检索准确性。-->
  47. <!-- </div>-->
  48. <!-- </div>-->
  49. </div>
  50. <div
  51. v-if="state.indexMethod.value === '高质量'"
  52. class="mt-[var(--czr-gap)] flex h-[2.5rem] w-full items-center gap-[0.5rem] rounded-[0.25rem] px-[0.75rem] text-[0.8rem] text-[#666666]"
  53. style="
  54. border: var(--czr-border);
  55. border-color: var(--czr-warning-color);
  56. background-color: rgba(var(--czr-warning-color-rgb), 0.1);
  57. "
  58. >
  59. <SvgIcon
  60. name="czr_tip"
  61. color="var(--czr-warning-color)"
  62. />使用高质量模式进行嵌入后,无法切换回经济模式。
  63. </div>
  64. </template>
  65. <template v-if="state.embedding.show">
  66. <div class="__czr-title_1 mb-[0.5rem]">Embedding 模型</div>
  67. <CzrFormColumn
  68. required
  69. class="__czr-table-form-column"
  70. :span="24"
  71. label="模型类型"
  72. v-model:param="state.embedding.value"
  73. link="select"
  74. :options="state.optionsEmbedding"
  75. :clearable="false"
  76. />
  77. </template>
  78. <template v-if="state.searchMethod.show">
  79. <div class="__czr-title_1">检索方式</div>
  80. <div
  81. class="mt-[1rem] w-full cursor-pointer rounded-[var(--czr-gap)] bg-[#ffffff]"
  82. style="border: var(--czr-border); border-width: 0.13rem"
  83. :class="{
  84. 'search-method-active':
  85. state.searchMethod.value === SearchMethodType.Vector,
  86. }"
  87. @click="() => (state.searchMethod.value = SearchMethodType.Vector)"
  88. >
  89. <div class="h-full w-full px-[1.5rem] py-[1rem]">
  90. <div class="flex items-center">
  91. <img
  92. src="@/assets/images/model-icon-4.png"
  93. class="mr-[var(--czr-gap)] h-[3.25rem] w-[3.25rem]"
  94. />
  95. <div class="text-[1.25rem] font-bold text-[#2E3238]">向量检索</div>
  96. </div>
  97. <div
  98. class="mt-[0.5rem] text-[0.88rem] text-[#606266]"
  99. style="line-height: 1.3rem"
  100. >
  101. 通过生成查询嵌入并查询与其向量表示最相似的文本分段
  102. </div>
  103. <template v-if="state.searchMethod.value === SearchMethodType.Vector">
  104. <div
  105. class="mt-[1rem] flex flex-col gap-[0.5rem] rounded-[0.25rem] bg-[#ffffff] p-[var(--czr-gap)] text-[1rem] font-bold text-[#2E3238]"
  106. >
  107. <div class="flex items-center gap-[0.5rem]">
  108. <el-switch
  109. size="small"
  110. v-model="state.searchMethod[SearchMethodType.Vector].isRerank"
  111. :active-value="1"
  112. :inactive-value="0"
  113. />
  114. Rerank 模型
  115. <el-tooltip
  116. content="重排序模型将根据候选文档列表与用户问题语义匹配度进行重新排序,从而改进语义排序的结果。"
  117. placement="top"
  118. >
  119. <SvgIcon name="czr_tip" size="14" />
  120. </el-tooltip>
  121. </div>
  122. <template
  123. v-if="state.searchMethod[SearchMethodType.Vector].isRerank"
  124. >
  125. <CzrFormColumn
  126. required
  127. class="__czr-table-form-column"
  128. :span="24"
  129. label-width="0px"
  130. v-model:param="
  131. state.searchMethod[SearchMethodType.Vector].rerank
  132. "
  133. link="select"
  134. :options="state.optionsRerank"
  135. :clearable="false"
  136. default-error-msg="请选择Rerank模型"
  137. />
  138. </template>
  139. <div class="flex gap-[1rem]">
  140. <div class="flex-1">
  141. <div class="flex h-[1rem] items-center gap-[0.5rem]">
  142. TOP K
  143. <el-tooltip
  144. content="用于筛选与用户问题相似度最高的文本片段。系统同时会根据选用模型上下文窗口大小动态调整分段数量。"
  145. placement="top"
  146. >
  147. <SvgIcon name="czr_tip" size="14" />
  148. </el-tooltip>
  149. </div>
  150. <div class="mt-[1rem]">
  151. <el-slider
  152. v-model="state.searchMethod[SearchMethodType.Vector].topK"
  153. show-input
  154. class="re-slider"
  155. :min="1"
  156. :max="10"
  157. />
  158. </div>
  159. </div>
  160. <div class="flex-1">
  161. <div class="flex h-[1rem] items-center gap-[0.5rem]">
  162. <el-switch
  163. size="small"
  164. v-model="
  165. state.searchMethod[SearchMethodType.Vector].isScore
  166. "
  167. :active-value="1"
  168. :inactive-value="0"
  169. />
  170. Score 阈值
  171. <el-tooltip
  172. content="重排序模型将根据候选文档列表与用户问题语义匹配度进行重新排序,从而改进语义排序的结果。"
  173. placement="top"
  174. >
  175. <SvgIcon name="czr_tip" size="14" />
  176. </el-tooltip>
  177. </div>
  178. <div class="mt-[1rem]">
  179. <el-slider
  180. v-model="state.searchMethod[SearchMethodType.Vector].score"
  181. show-input
  182. class="re-slider"
  183. :max="1"
  184. :min="0"
  185. :step="0.1"
  186. :disabled="
  187. !state.searchMethod[SearchMethodType.Vector].isScore
  188. "
  189. />
  190. </div>
  191. </div>
  192. </div>
  193. </div>
  194. </template>
  195. </div>
  196. </div>
  197. <div
  198. class="mt-[1rem] w-full cursor-pointer rounded-[var(--czr-gap)] bg-[#ffffff]"
  199. style="border: var(--czr-border); border-width: 0.13rem"
  200. :class="{
  201. 'search-method-active':
  202. state.searchMethod.value === SearchMethodType.Global,
  203. }"
  204. @click="() => (state.searchMethod.value = SearchMethodType.Global)"
  205. >
  206. <div class="h-full w-full px-[1.5rem] py-[1rem]">
  207. <div class="flex items-center">
  208. <img
  209. src="@/assets/images/model-icon-5.png"
  210. class="mr-[var(--czr-gap)] h-[3.25rem] w-[3.25rem]"
  211. />
  212. <div class="text-[1.25rem] font-bold text-[#2E3238]">全文检索</div>
  213. </div>
  214. <div
  215. class="mt-[0.5rem] text-[0.88rem] text-[#606266]"
  216. style="line-height: 1.3rem"
  217. >
  218. 索引文档中的所有词汇,从而允许用户查询任意词汇,并返回包含这些词汇的文本片段
  219. </div>
  220. <template v-if="state.searchMethod.value === SearchMethodType.Global">
  221. <div
  222. class="mt-[1rem] flex flex-col gap-[0.5rem] rounded-[0.25rem] bg-[#ffffff] p-[var(--czr-gap)] text-[1rem] font-bold text-[#2E3238]"
  223. >
  224. <div class="flex items-center gap-[0.5rem]">
  225. <el-switch
  226. size="small"
  227. v-model="state.searchMethod[SearchMethodType.Global].isRerank"
  228. :active-value="1"
  229. :inactive-value="0"
  230. />
  231. Rerank 模型
  232. <el-tooltip
  233. content="用于筛选与用户问题相似度最高的文本片段。系统同时会根据选用模型上下文窗口大小动态调整分段数量。"
  234. placement="top"
  235. >
  236. <SvgIcon name="czr_tip" size="14" />
  237. </el-tooltip>
  238. </div>
  239. <template
  240. v-if="state.searchMethod[SearchMethodType.Global].isRerank"
  241. >
  242. <CzrFormColumn
  243. class="__czr-table-form-column"
  244. :span="24"
  245. label-width="0px"
  246. v-model:param="
  247. state.searchMethod[SearchMethodType.Global].rerank
  248. "
  249. link="select"
  250. :options="state.optionsRerank"
  251. :clearable="false"
  252. default-error-msg="请选择Rerank模型"
  253. />
  254. </template>
  255. <div class="flex gap-[1rem]">
  256. <div class="flex-1">
  257. <div class="flex h-[1rem] items-center gap-[0.5rem]">
  258. TOP K
  259. <el-tooltip
  260. content="用于筛选与用户问题相似度最高的文本片段。系统同时会根据选用模型上下文窗口大小动态调整分段数量。"
  261. placement="top"
  262. >
  263. <SvgIcon name="czr_tip" size="14" />
  264. </el-tooltip>
  265. </div>
  266. <div class="mt-[1rem]">
  267. <el-slider
  268. v-model="state.searchMethod[SearchMethodType.Global].topK"
  269. show-input
  270. class="re-slider"
  271. :min="1"
  272. :max="10"
  273. />
  274. </div>
  275. </div>
  276. <div
  277. class="flex-1"
  278. v-if="state.searchMethod[SearchMethodType.Global].isRerank"
  279. >
  280. <div class="flex h-[1rem] items-center gap-[0.5rem]">
  281. <el-switch
  282. size="small"
  283. v-model="
  284. state.searchMethod[SearchMethodType.Global].isScore
  285. "
  286. :active-value="1"
  287. :inactive-value="0"
  288. />
  289. Score 阈值
  290. <el-tooltip
  291. content="用于设置文本片段筛选的相似度阈值。"
  292. placement="top"
  293. >
  294. <SvgIcon name="czr_tip" size="14" />
  295. </el-tooltip>
  296. </div>
  297. <div class="mt-[1rem]">
  298. <el-slider
  299. v-model="state.searchMethod[SearchMethodType.Global].score"
  300. show-input
  301. class="re-slider"
  302. :max="1"
  303. :min="0"
  304. :step="0.1"
  305. :disabled="
  306. !state.searchMethod[SearchMethodType.Global].isScore
  307. "
  308. />
  309. </div>
  310. </div>
  311. </div>
  312. </div>
  313. </template>
  314. </div>
  315. </div>
  316. <div
  317. class="mt-[1rem] w-full cursor-pointer rounded-[var(--czr-gap)] bg-[#ffffff]"
  318. style="border: var(--czr-border); border-width: 0.13rem"
  319. :class="{
  320. 'search-method-active':
  321. state.searchMethod.value === SearchMethodType.Mix,
  322. }"
  323. @click="() => (state.searchMethod.value = SearchMethodType.Mix)"
  324. >
  325. <div class="h-full w-full px-[1.5rem] py-[1rem]">
  326. <div class="flex items-center">
  327. <img
  328. src="@/assets/images/model-icon-6.png"
  329. class="mr-[var(--czr-gap)] h-[3.25rem] w-[3.25rem]"
  330. />
  331. <div class="text-[1.25rem] font-bold text-[#2E3238]">混合检索</div>
  332. <img
  333. src="@/assets/images/model-icon-3.png"
  334. class="h-[1.27rem] w-[2.88rem]"
  335. />
  336. </div>
  337. <div
  338. class="mt-[0.5rem] text-[0.88rem] text-[#606266]"
  339. style="line-height: 1.3rem"
  340. >
  341. 同时执行全文检索和向量检索,并应用重排序步骤,从两类查询结果中选择匹配用户问题的最佳结果,用户可以选择设置权重或配置重新排序模型。
  342. </div>
  343. <template v-if="state.searchMethod.value === SearchMethodType.Mix">
  344. <div class="mt-[1rem] flex w-full gap-[1rem]">
  345. <div
  346. class="__hover h-[8.38rem] flex-1 rounded-[var(--czr-gap)] bg-[#ffffff] px-[1.5rem] py-[1rem]"
  347. style="border: var(--czr-border); border-width: 0.13rem"
  348. :class="{
  349. 'index-method-active':
  350. state.searchMethod[SearchMethodType.Mix].indexMethod ===
  351. 'weightSetting',
  352. }"
  353. @click="
  354. () =>
  355. (state.searchMethod[SearchMethodType.Mix].indexMethod =
  356. 'weightSetting')
  357. "
  358. >
  359. <div class="flex items-center">
  360. <img
  361. src="@/assets/images/model-icon-8.png"
  362. class="mr-[var(--czr-gap)] h-[3.25rem] w-[3.25rem]"
  363. />
  364. <div class="text-[1.25rem] font-bold text-[#2E3238]">
  365. 权重设置
  366. </div>
  367. <img
  368. src="@/assets/images/model-icon-3.png"
  369. class="h-[1.27rem] w-[2.88rem]"
  370. />
  371. </div>
  372. <div
  373. class="mt-[0.5rem] text-[0.88rem] text-[#606266]"
  374. style="line-height: 1.3rem"
  375. >
  376. 通过调整分配的权重,重新排序策略确定是优先进行语义匹配还是关键字匹配。
  377. </div>
  378. </div>
  379. <div
  380. class="__hover h-[8.38rem] flex-1 rounded-[var(--czr-gap)] bg-[#ffffff] px-[1.5rem] py-[1rem]"
  381. style="border: var(--czr-border); border-width: 0.13rem"
  382. :class="{
  383. 'index-method-active':
  384. state.searchMethod[SearchMethodType.Mix].indexMethod ===
  385. 'rerank',
  386. }"
  387. @click="
  388. () =>
  389. (state.searchMethod[SearchMethodType.Mix].indexMethod =
  390. 'rerank')
  391. "
  392. >
  393. <div class="flex items-center">
  394. <img
  395. src="@/assets/images/model-icon-9.png"
  396. class="mr-[var(--czr-gap)] h-[3.25rem] w-[3.25rem]"
  397. />
  398. <div class="text-[1.25rem] font-bold text-[#2E3238]">
  399. Rerank 模型
  400. </div>
  401. </div>
  402. <div
  403. class="mt-[0.5rem] text-[0.88rem] text-[#606266]"
  404. style="line-height: 1.3rem"
  405. >
  406. 重排序模型将根据候选文档列表与用户问题语义匹配度进行重新排序,从而改进语义排序的结果。
  407. </div>
  408. </div>
  409. </div>
  410. <div
  411. class="mt-[1rem] flex flex-col gap-[0.5rem] rounded-[0.25rem] bg-[#ffffff] p-[var(--czr-gap)] text-[1rem] font-bold text-[#2E3238]"
  412. >
  413. <template
  414. v-if="
  415. state.searchMethod[SearchMethodType.Mix].indexMethod ===
  416. 'weightSetting'
  417. "
  418. >
  419. <div>
  420. <el-slider
  421. v-model="state.searchMethod[SearchMethodType.Mix].weight"
  422. :min="0"
  423. :max="1"
  424. :step="0.1"
  425. class="weight-slider"
  426. />
  427. <div class="flex justify-between">
  428. <div class="text-[var(--czr-main-color)]">
  429. 语义 {{ state.searchMethod[SearchMethodType.Mix].weight }}
  430. </div>
  431. <div class="text-[var(--czr-success-color)]">
  432. {{
  433. (
  434. 1 - state.searchMethod[SearchMethodType.Mix].weight
  435. ).toFixed(1)
  436. }}
  437. 关键词
  438. </div>
  439. </div>
  440. </div>
  441. </template>
  442. <template
  443. v-else-if="
  444. state.searchMethod[SearchMethodType.Mix].indexMethod ===
  445. 'rerank'
  446. "
  447. >
  448. <CzrFormColumn
  449. class="__czr-table-form-column"
  450. :span="24"
  451. label-width="0px"
  452. v-model:param="state.searchMethod[SearchMethodType.Mix].rerank"
  453. link="select"
  454. :options="state.optionsRerank"
  455. :clearable="false"
  456. default-error-msg="请选择Rerank模型"
  457. />
  458. </template>
  459. </div>
  460. <div
  461. class="mt-[1rem] flex flex-col gap-[0.5rem] rounded-[0.25rem] bg-[#ffffff] p-[var(--czr-gap)] text-[1rem] font-bold text-[#2E3238]"
  462. >
  463. <div class="flex gap-[1rem]">
  464. <div class="flex-1">
  465. <div class="flex h-[1rem] items-center gap-[0.5rem]">
  466. TOP K
  467. <el-tooltip
  468. content="用于筛选与用户问题相似度最高的文本片段。系统同时会根据选用模型上下文窗口大小动态调整分段数量。"
  469. placement="top"
  470. >
  471. <SvgIcon name="czr_tip" size="14" />
  472. </el-tooltip>
  473. </div>
  474. <div class="mt-[1rem]">
  475. <el-slider
  476. v-model="state.searchMethod[SearchMethodType.Mix].topK"
  477. show-input
  478. class="re-slider"
  479. :min="1"
  480. :max="10"
  481. />
  482. </div>
  483. </div>
  484. <div class="flex-1">
  485. <div class="flex h-[1rem] items-center gap-[0.5rem]">
  486. <el-switch
  487. size="small"
  488. v-model="state.searchMethod[SearchMethodType.Mix].isScore"
  489. :active-value="1"
  490. :inactive-value="0"
  491. />
  492. Score 阈值
  493. <el-tooltip
  494. content="用于设置文本片段筛选的相似度阈值。"
  495. placement="top"
  496. >
  497. <SvgIcon name="czr_tip" size="14" />
  498. </el-tooltip>
  499. </div>
  500. <div class="mt-[1rem]">
  501. <el-slider
  502. v-model="state.searchMethod[SearchMethodType.Mix].score"
  503. show-input
  504. class="re-slider"
  505. :max="1"
  506. :min="0"
  507. :step="0.1"
  508. :disabled="
  509. !state.searchMethod[SearchMethodType.Mix].isScore
  510. "
  511. />
  512. </div>
  513. </div>
  514. </div>
  515. </div>
  516. </template>
  517. </div>
  518. </div>
  519. </template>
  520. </template>
  521. <script setup lang="ts">
  522. import {
  523. computed,
  524. getCurrentInstance,
  525. onMounted,
  526. reactive,
  527. ref,
  528. watch,
  529. } from 'vue'
  530. import { useDictionaryStore } from '@/stores'
  531. import { pluginGetListByType, pluginInstanceByType } from '@/api/modules/model'
  532. import { ElMessage } from 'element-plus'
  533. const DictionaryStore = useDictionaryStore()
  534. const emit = defineEmits([])
  535. const props = defineProps({})
  536. const { proxy }: any = getCurrentInstance()
  537. enum SearchMethodType {
  538. Vector = 'VECTOR',
  539. Global = 'FULL_TEXT',
  540. Mix = 'HYBRID',
  541. }
  542. const state: any = reactive({
  543. optionsEmbedding: [],
  544. optionsRerank: [],
  545. indexMethod: {
  546. show: true,
  547. value: '高质量',
  548. },
  549. embedding: {
  550. show: true,
  551. value: '',
  552. },
  553. searchMethod: {
  554. show: true,
  555. value: SearchMethodType.Vector,
  556. [SearchMethodType.Vector]: {
  557. isRerank: 1,
  558. rerank: '',
  559. topK: 5,
  560. isScore: 1,
  561. score: 0.5,
  562. },
  563. [SearchMethodType.Global]: {
  564. isRerank: 1,
  565. rerank: '',
  566. topK: 5,
  567. isScore: 1,
  568. score: 0.5,
  569. },
  570. [SearchMethodType.Mix]: {
  571. indexMethod: 1,
  572. weight: 0.7,
  573. rerank: '',
  574. topK: 5,
  575. isScore: 1,
  576. score: 0.5,
  577. },
  578. },
  579. })
  580. const optionsEmbeddingMapCpt = computed(() => {
  581. const map = new Map()
  582. state.optionsEmbedding.forEach((v) => {
  583. map.set(v.value, v)
  584. })
  585. return map
  586. })
  587. const optionsRerankMapCpt = computed(() => {
  588. const map = new Map()
  589. state.optionsRerank.forEach((v) => {
  590. map.set(v.value, v)
  591. })
  592. return map
  593. })
  594. const reset = () => {
  595. state.indexMethod = {
  596. show: true,
  597. value: '高质量',
  598. }
  599. state.embedding = {
  600. show: true,
  601. value: '',
  602. }
  603. state.searchMethod = {
  604. show: true,
  605. value: SearchMethodType.Vector,
  606. [SearchMethodType.Vector]: {
  607. isRerank: 1,
  608. rerank: '',
  609. topK: 5,
  610. isScore: 1,
  611. score: 0.5,
  612. },
  613. [SearchMethodType.Global]: {
  614. isRerank: 1,
  615. rerank: '',
  616. topK: 5,
  617. isScore: 1,
  618. score: 0.5,
  619. },
  620. [SearchMethodType.Mix]: {
  621. indexMethod: 1,
  622. weight: 0.7,
  623. rerank: '',
  624. topK: 5,
  625. isScore: 1,
  626. score: 0.5,
  627. },
  628. }
  629. }
  630. const getData = () => {
  631. const result = {
  632. indexingTechnique: state.indexMethod.value,
  633. embeddingModelId: state.embedding.value,
  634. embeddingModel: optionsEmbeddingMapCpt.value.get(state.embedding.value)
  635. .label,
  636. indexConfig: {
  637. type: state.searchMethod.value,
  638. },
  639. }
  640. switch (state.searchMethod.value) {
  641. case SearchMethodType.Vector:
  642. {
  643. result.indexConfig.isRerank =
  644. state.searchMethod[SearchMethodType.Vector].isRerank
  645. result.indexConfig.rerankTypeId =
  646. state.searchMethod[SearchMethodType.Vector].rerank
  647. result.indexConfig.rerankType = optionsRerankMapCpt.value.get(
  648. state.searchMethod[SearchMethodType.Vector].rerank,
  649. ).label
  650. result.indexConfig.topK =
  651. state.searchMethod[SearchMethodType.Vector].topK
  652. result.indexConfig.isScore =
  653. state.searchMethod[SearchMethodType.Vector].isScore
  654. result.indexConfig.score =
  655. state.searchMethod[SearchMethodType.Vector].score
  656. }
  657. break
  658. case SearchMethodType.Global:
  659. {
  660. result.indexConfig.isRerank =
  661. state.searchMethod[SearchMethodType.Global].isRerank
  662. result.indexConfig.rerankTypeId =
  663. state.searchMethod[SearchMethodType.Global].rerank
  664. result.indexConfig.rerankType = optionsRerankMapCpt.value.get(
  665. state.searchMethod[SearchMethodType.Global].rerank,
  666. ).label
  667. result.indexConfig.topK =
  668. state.searchMethod[SearchMethodType.Global].topK
  669. result.indexConfig.isScore =
  670. state.searchMethod[SearchMethodType.Global].isScore
  671. result.indexConfig.score =
  672. state.searchMethod[SearchMethodType.Global].score
  673. }
  674. break
  675. case SearchMethodType.Mix:
  676. {
  677. result.indexConfig.weightRerank =
  678. state.searchMethod[SearchMethodType.Mix].indexMethod
  679. result.indexConfig.semantics =
  680. state.searchMethod[SearchMethodType.Mix].weight
  681. result.indexConfig.keyword =
  682. 1 - state.searchMethod[SearchMethodType.Mix].weight
  683. result.indexConfig.rerankTypeId =
  684. state.searchMethod[SearchMethodType.Mix].rerank
  685. result.indexConfig.rerankType = optionsRerankMapCpt.value.get(
  686. state.searchMethod[SearchMethodType.Mix].rerank,
  687. ).label
  688. result.indexConfig.topK = state.searchMethod[SearchMethodType.Mix].topK
  689. result.indexConfig.isScore =
  690. state.searchMethod[SearchMethodType.Mix].isScore
  691. result.indexConfig.score =
  692. state.searchMethod[SearchMethodType.Mix].score
  693. }
  694. break
  695. }
  696. return result
  697. }
  698. const init = ({ indexingTechnique, embeddingModelId, indexConfig }) => {
  699. reset()
  700. state.indexMethod.value = indexingTechnique
  701. state.embedding.value = embeddingModelId
  702. state.searchMethod.value = indexConfig.type
  703. switch (state.searchMethod.value) {
  704. case SearchMethodType.Vector:
  705. {
  706. state.searchMethod[SearchMethodType.Vector].isRerank = Number(
  707. indexConfig.isRerank,
  708. )
  709. state.searchMethod[SearchMethodType.Vector].rerank =
  710. indexConfig.rerankTypeId
  711. state.searchMethod[SearchMethodType.Vector].topK = Number(
  712. indexConfig.topK,
  713. )
  714. state.searchMethod[SearchMethodType.Vector].isScore = Number(
  715. indexConfig.isScore,
  716. )
  717. state.searchMethod[SearchMethodType.Vector].score = Number(
  718. indexConfig.score,
  719. )
  720. }
  721. break
  722. case SearchMethodType.Global:
  723. {
  724. state.searchMethod[SearchMethodType.Global].isRerank = Number(
  725. indexConfig.isRerank,
  726. )
  727. state.searchMethod[SearchMethodType.Global].rerank =
  728. indexConfig.rerankTypeId
  729. state.searchMethod[SearchMethodType.Global].topK = Number(
  730. indexConfig.topK,
  731. )
  732. state.searchMethod[SearchMethodType.Global].isScore = Number(
  733. indexConfig.isScore,
  734. )
  735. state.searchMethod[SearchMethodType.Global].score = Number(
  736. indexConfig.score,
  737. )
  738. }
  739. break
  740. case SearchMethodType.Mix:
  741. {
  742. state.searchMethod[SearchMethodType.Mix].indexMethod =
  743. indexConfig.weightRerank
  744. state.searchMethod[SearchMethodType.Mix].weight = Number(
  745. indexConfig.semantics,
  746. )
  747. state.searchMethod[SearchMethodType.Mix].rerank =
  748. indexConfig.rerankTypeId
  749. state.searchMethod[SearchMethodType.Mix].topK = Number(indexConfig.topK)
  750. state.searchMethod[SearchMethodType.Mix].isScore = Number(
  751. indexConfig.isScore,
  752. )
  753. state.searchMethod[SearchMethodType.Mix].score = Number(
  754. indexConfig.score,
  755. )
  756. }
  757. break
  758. }
  759. }
  760. onMounted(() => {
  761. initDictionary()
  762. })
  763. defineExpose({
  764. reset,
  765. getData,
  766. init,
  767. })
  768. const initDictionary = () => {
  769. pluginGetListByType({ type: 'EMBEDDING' })
  770. .then(({ data }: any) => {
  771. state.optionsEmbedding = data.map((v) => {
  772. v.label = v.name
  773. v.value = v.id
  774. return v
  775. })
  776. })
  777. .catch(() => {})
  778. pluginGetListByType({ type: 'RERANK' })
  779. .then(({ data }: any) => {
  780. state.optionsRerank = data.map((v) => {
  781. v.label = v.name
  782. v.value = v.id
  783. return v
  784. })
  785. })
  786. .catch(() => {})
  787. }
  788. </script>
  789. <style lang="scss" scoped>
  790. .index-method-active {
  791. box-shadow: 0rem 0.25rem 0.63rem 0rem rgba(40, 83, 247, 0.05);
  792. border-color: var(--czr-main-color) !important;
  793. background-color: rgba(var(--czr-main-color-rgb), 0.1);
  794. }
  795. .search-method-active {
  796. box-shadow: 0rem 0.25rem 0.63rem 0rem rgba(40, 83, 247, 0.05);
  797. border-color: var(--czr-main-color) !important;
  798. background-color: rgba(var(--czr-main-color-rgb), 0.1);
  799. background-image: url('@/assets/images/model-icon-7.png');
  800. background-size: 100% 100%;
  801. background-repeat: no-repeat;
  802. }
  803. :deep(.re-slider) {
  804. flex-direction: row-reverse;
  805. .el-slider__runway {
  806. margin-left: 10px;
  807. margin-right: 0;
  808. }
  809. }
  810. :deep(.weight-slider) {
  811. .el-slider__runway {
  812. background-color: var(--czr-success-color);
  813. }
  814. }
  815. </style>