gerson
2024-11-01 3cc575268b3042b35e785ffeebf51a6ea3b5e73d
src/stores/chatRoom.ts
@@ -1,13 +1,124 @@
import { computed, ref, watch } from 'vue';
import type { ChatRoomItem } from '../layout/component/sidebar/waterLeftAside/types';
import { Local } from '../utils/storage';
import { computed, ref } from 'vue';
import { getSectionList, getSelectSample, getUserTemplateList } from '../api/ai/chat';
import type { ChatRoomItem } from '../layout/component/sidebar/components/types';
/**
 * Room 关联的一些配置
 */
export type RoomConfig = {
   /** 是否直接调用大模型(通义千问)回答 */
   isAnswerByLLM: boolean;
   /** @description 从首页进去获取的第一个回复,回调函数 */
   firstResCb: any;
   /** @description 当前聊天室的 group_type */
   activeGroupType: string;
};
export type RoomConfigKey = keyof RoomConfig;
export const roomConfig = ref<Record<string, RoomConfig>>(null);
export const setRoomConfig = <T extends RoomConfigKey>(roomId: string, key: T, value: RoomConfig[T]) => {
   if (!roomId) return;
   if (!roomConfig.value) {
      roomConfig.value = {};
   }
   if (!roomConfig.value[roomId]) {
      roomConfig.value[roomId] = {
         [key]: value,
      } as any;
   } else {
      roomConfig.value[roomId][key] = value;
   }
};
export const getRoomConfig = <T extends RoomConfigKey>(roomId: string, key: T) => {
   if (!roomId) return;
   if (!roomConfig.value) {
      return null;
   }
   if (!roomConfig.value[roomId]) {
      return null;
   } else {
      return roomConfig.value[roomId][key];
   }
};
export const chatRoomList = ref<ChatRoomItem[]>([]);
export const activeRoomId = ref(null);
export const activeChatRoom = computed(() => chatRoomList.value?.find((item) => item.id === activeRoomId.value));
export const activeSampleId = ref(null);
export const activeSectionAId= ref(null);
export const activeLLMId= ref(null);
export const activeSectionAId = ref(null);
export const topGroupId = ref(null);
export const activeLLMId = ref(null);
/** @description 当前聊天室 groupType */
export const activeGroupType = computed({
   get: () => {
      const result = getRoomConfig(activeRoomId.value, 'activeGroupType') ?? '业务场景';
      return result;
   },
   set: (value) => {
      setRoomConfig(activeRoomId.value, 'activeGroupType', value);
   },
});
/**
 * 全局使用的 ref
 */
export const sectionAList = ref([]);
//#region ====================== 全局使用数据 ======================
// group 列表
export const sceneGroupList = ref([]);
// groupType 列表
export const groupTypeList = computed(() => Array.from(new Set(sceneGroupList.value.map((item) => item.group_type))));
// 办公/模板 列表
export const exampleSceneList = ref([]);
export const officeList = ref([]);
export const groupTypeMapIcon = {
   办公助手: 'ywicon-bangong',
   知识库: 'ywicon-changyonggongjuzhishisuoyin',
   业务场景: 'ywicon-yewu',
};
//获取场景选择列表
const getSceneGroupList = async () => {
   const res = await getSectionList();
   sceneGroupList.value = res?.groups ?? [];
};
const getSelectListSample = async () => {
   const res1 = getSelectSample({});
   const res2 = getUserTemplateList();
   const [sampleListPromise, templateListPromise] = await Promise.allSettled([res1, res2]);
   const samples = (sampleListPromise as any).value?.samples ?? [];
   const templateSamples = ((templateListPromise as any).value?.templates ?? []).map((item) => ({
      group_id: item.template_group,
      sample_id: item.template_id,
      sample_title: item.template_title,
      sample_question: item.template_value,
      //#region ====================== template 特有字段 ======================
      template_create_time: item.create_time,
      template_type: item.template_type,
      isTemplate: true,
      //#endregion
   }));
   exampleSceneList.value = samples
      .concat(templateSamples)
      .map((item) => ({ ...item, Icon: '/static/images/wave/ChatImg.png', BgColor: randomHexColor() }));
};
//随机生成颜色
const randomHexColor = () => {
   return `#${Math.floor(Math.random() * 16777215)
      .toString(16)
      .padEnd(6, '0')}`;
};
/**
 * 获取全局所有数据
 */
export const getAllData = async () => {
   getSceneGroupList();
   getSelectListSample();
};
//#endregion