yangyin
2024-10-28 be0b1a911b5e36faf8bcc61c452520e97f00cd16
src/stores/chatRoom.ts
@@ -1,18 +1,95 @@
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 type { ChatRoomItem } from '../layout/component/sidebar/components/types';
import { getSectionList } from '../api/ai/chat';
export const chatRoomList = ref<ChatRoomItem[]>(Local.get('chatRoomList'));
/**
 * Room 关联的一些配置
 */
export type RoomConfig = {
   /** 是否直接调用大模型(通义千问)回答 */
   isAnswerByLLM: boolean;
   /** @description 从首页进去获取的第一个回复,回调函数 */
   firstResCb: any;
   /** @description 当前聊天室的 group_type */
   activeGroupType: string;
};
watch(
   () => chatRoomList.value,
   (val) => {
      Local.set('chatRoomList', val);
   },
   {
      deep: true,
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 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 groupTypeMapIcon = {
   办公助手: 'ywicon-bangong',
   知识库: 'ywicon-changyonggongjuzhishisuoyin',
   业务场景: 'ywicon-yewu',
};
const getSceneGroupList = async () => {
   const res = await getSectionList();
   sceneGroupList.value = res?.groups ?? [];
};
/**
 * 获取全局所有数据
 */
export const getAllData = async () => {
   getSceneGroupList();
};
//#endregion