yangyin
2024-11-14 a12818479aac56a0c08206ce4dcf4ede092e9376
src/stores/chatRoom.ts
@@ -1,18 +1,180 @@
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 { CreateHistoryGroup, getSectionList, getSelectSample, getUserTemplateList } from '../api/ai/chat';
import type { ChatRoomItem } from '../layout/component/sidebar/components/types';
import { router } from '../router';
import { gotoRoute } from '../utils/route';
/**
 * Room 关联的一些配置
 */
export type RoomConfig = {
   /** 是否直接调用大模型(通义千问)回答 */
   isAnswerByLLM: boolean;
   /** @description 从首页进去获取的第一个回复,回调函数 */
   firstResCb: any;
   /** @description 当前聊天室的 group_type */
   activeGroupType: string;
};
export const chatRoomList = ref<ChatRoomItem[]>(Local.get('chatRoomList'));
export type RoomConfigKey = keyof RoomConfig;
export const roomConfig = ref<Record<string, RoomConfig>>(null);
watch(
   () => chatRoomList.value,
   (val) => {
      Local.set('chatRoomList', val);
   },
   {
      deep: true,
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([]);
export const isShowLogin = ref(false);
export const isLoginStatus = ref(false);
//#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
/** @description 聊天室展示模式 */
export type ChatMode = 'share' | 'default';
/** @description 聊天室展示模式 */
// export const chatDisplayMode = ref<ChatMode>('default');
export const SHARE_PAGE_NAME = 'ShareAnswer';
export const isSharePage = computed(() => {
   const isShare = router.currentRoute.value.name === SHARE_PAGE_NAME;
   return isShare;
});
export const newChatRoomClick = async () => {
   const res = await CreateHistoryGroup({
      group_title: '新建对话开始',
   });
   const newRoom = {
      id: res.history_group_id,
      isInitial: true,
      title: '新建对话开始',
   };
   if (!chatRoomList.value) {
      chatRoomList.value = [newRoom];
   } else {
      chatRoomList.value.unshift(newRoom);
   }
   activeRoomId.value = newRoom.id;
   gotoAnswerPage(newRoom);
};
export const gotoAnswerPage = (room: ChatRoomItem) => {
   if (room.isInitial) {
      gotoRoute({
         name: 'Home',
         query: {
            id: room.id,
         },
      });
   } else {
      gotoRoute({
         name: 'AskAnswer',
         query: {
            id: room.id,
         },
      });
   }
};
// 是否已经展示引导
export const hadShowFirstGuide = ref(false);
//是否是新老用户
export const isNewOldUser = ref(null);