- 智能思考
-
- ),
- });
+ // 开始处理
+ const handleOpen = (data: any): void => {
+ // const { run_id } = data;
+ // currentTaskId.value = run_id;
+ // conversationList.value.push({
+ // key: run_id,
+ // run_id,
+ // role: INTELLECTUAL_THINKING_ROLE,
+ // content: (
+ //
+ // 智能思考
+ //
+ // ),
+ // });
};
- // 节点更新处理
- const handleNodeUpdate = (data: any) => {
- const { run_id, status, output } = data;
-
- switch (status) {
- case 'TeamRunResponseContent':
- conversationList.value.push({
- run_id,
- content: data,
- role: ANSWER_ROLE,
- messageRender: (item) => (
-
- 
- {item.message}
-
- )
- });
- break;
- case 'TeamRunCompleted':
- conversationList.value.push({
- run_id,
- content: output,
- role: ANSWER_ROLE,
- messageRender: (content: string) => ,
- style: ANSWER_STYLE
- });
- break;
- }
- };
-
- // 最终结果处理
- const handleFinalResult = (data: any) => {
+ // 最终结果处理
+ const handleFileReview = (data: MESSAGE.Answer) => {
const { run_id, output } = data;
showRightView.value = true;
- const _files = output?.files;
- rightViewContent.value = _files?.[0]?.content || '';
+ // const _files = output?.files;
+ // rightViewInfo.value = _files?.[0]?.content || '';
- conversationList.value.push({
- run_id,
- id: currentTaskId.value,
- role: FILE_ROLE,
- content: _files,
- style: ANSWER_STYLE,
- footer: ({ item }: { item: any }) => {
- const nonQuestionElements = conversationList.value.filter((item) => item.role !== QUESTION_ROLE);
- const isLastAnswer = nonQuestionElements[nonQuestionElements.length - 1]?.id === item.id;
-
- // return (
- //
- // onDownload(rightViewContent?.value || '')}>
- //
- // {isLastAnswer && onRefresh && (
- // onRefresh(currentTaskId.value!, conversationList.value.length)}>
- //
- // )}
- //
- // );
- },
- });
+ // conversationList.value.push({
+ // run_id,
+ // role: FILE_ROLE,
+ // content: _files,
+ // style: ANSWER_STYLE,
+ // footer: ({ item }: { item: any }) => {
+ // const nonQuestionElements = conversationList.value.filter((item) => item.role !== QUESTION_ROLE);
+ // const isLastAnswer = nonQuestionElements[nonQuestionElements.length - 1]?.id === item.id;
+ // },
+ // });
};
- // 重置生成状态
+ // 重置生成状态
const resetGenerateStatus = () => {
generateLoading.value = false;
};
- // 错误处理
- const handleError = () => {
- resetGenerateStatus();
- antdMessage.error('连接服务器失败');
- };
+ // // 错误处理
+ // const handleError = () => {
+ // resetGenerateStatus();
+ // antdMessage.error('连接服务器失败');
+ // };
- const onRefresh = (tempId: string, tempIndex: number) => {
- generateLoading.value = true;
- conversationList.value.splice(tempIndex, 1, {
- id: tempId,
- loading: true,
+ const handleTaskStart = (data: MESSAGE.Answer) => {
+ const { run_id } = data;
+ currentTaskId.value = run_id;
+ conversationList.value.push({
+ run_id,
+ content: data,
+ output: data.output,
+ role: ANSWER_ROLE,
+ messageRender: (data: MESSAGE.Answer) => {
+ let outputEleClass: string = `thought-chain-output border-l-#E6E6E8 border-l-1px pl-12px relative left-6px mb-4px`;
+ !isLastAnswer(data) && (outputEleClass += ' hasLine pb-12px pt-4px');
+ return (
+ <>
+
+ 智能思考
+
+
+
+
+ );
+ },
});
};
- // 消息处理主函数
- const handleMessage = (parsedData: { event: string; data: any }) => {
- const { event, data } = parsedData;
- switch (event) {
- case 'start':
- handleStart(data);
+ const onRefresh = (run_id: string) => {
+ generateLoading.value = true;
+ conversationList.value = conversationList.value.filter((item) => item.run_id !== run_id);
+ initSse({ message: senderRef.value?.searchValue });
+ };
+
+ const isLastAnswer = (data: MESSAGE.Answer) => {
+ const { run_id } = data;
+ const nonQuestionElements = conversationList.value.filter(
+ (item) => item.role === ANSWER_ROLE && item.run_id === run_id,
+ );
+ return nonQuestionElements[nonQuestionElements.length - 1]?.run_id === run_id;
+ };
+
+ const isLastTask = (data: MESSAGE.Answer) => {
+ const { run_id } = data;
+ const lastElement = conversationList.value[conversationList.value.length - 1];
+ return lastElement && lastElement.run_id === run_id;
+ };
+
+ // 节点更新处理
+ const handleTaskUpdate = (data: MESSAGE.Answer) => {
+ const { run_id, output } = data;
+
+ const existingItemIndex = conversationList.value.findIndex((item) => item.run_id === run_id);
+
+ if (existingItemIndex !== -1 && output) {
+ const existingItem = conversationList.value[existingItemIndex];
+ existingItem.content.output += output;
+ }
+ };
+
+ const handleTaskEnd = (data: MESSAGE.Answer) => {
+ const { run_id, output, team_session_state } = data;
+
+ const existingItem = conversationList.value.find((item) => item.run_id === run_id);
+ resetGenerateStatus();
+
+ if (existingItem) {
+ existingItem.content.output += output;
+ existingItem.footer = ({ item: any }) => {
+ return (
+
+
+
+
+ {data.node}
+
+ onCopy(existingItem.content.output)}>
+
+
+
+ {isLastTask(data) && (
+ onRefresh(run_id)}>
+
+ )}
+
+ );
+ };
+ }
+
+ if (team_session_state) {
+ handleFileReview(data);
+ }
+ };
+ // 消息处理主函数
+ const handleMessage = (parsedData: { event: string; data: MESSAGE.Answer }) => {
+ const { data } = parsedData;
+ const { status } = data;
+ switch (status) {
+ case EnumTeamRunStatus.TeamRunStarted:
+ handleTaskStart(data);
break;
- case 'node_update':
- handleNodeUpdate(data);
+ case EnumTeamRunStatus.TeamRunResponseContent:
+ handleTaskUpdate(data);
break;
- case 'final_result':
- handleFinalResult(data);
- break;
- case 'end':
- resetGenerateStatus();
- break;
- case 'error':
- handleError();
+ case EnumTeamRunStatus.TeamRunCompleted:
+ handleTaskEnd(data);
break;
default:
break;
@@ -217,11 +275,13 @@ export default function useChatHandler(): UseChatHandlerReturn {
return {
roles,
+ senderRef,
currentTaskId,
handleMessage,
+ handleOpen,
generateLoading,
conversationList,
showRightView,
- rightViewContent,
+ rightViewInfo,
};
-}
\ No newline at end of file
+}
diff --git a/src/components/xt-chat/xt-bubble/style.scss b/src/components/xt-chat/xt-bubble/style.scss
index 3821521..7b01927 100644
--- a/src/components/xt-chat/xt-bubble/style.scss
+++ b/src/components/xt-chat/xt-bubble/style.scss
@@ -28,6 +28,27 @@
flex-direction: row-reverse;
}
+ .thought-chain-item {
+ position: relative;
+ margin: 0;
+ font-size: 14px;
+ list-style: none;
+ .thought-chain-output {
+ position: relative;
+ &.hasLine {
+ &::before {
+ content: '';
+ position: absolute;
+ top: 0;
+ left: 0;
+ width: 100%;
+ height: 100%;
+ border-left: 1px solid #E6E6E8;
+ }
+ }
+ }
+ }
+
.xt-bubble-avatar {
}
diff --git a/src/components/xt-chat/xt-bubble/xt-bubbleList.vue b/src/components/xt-chat/xt-bubble/xt-bubbleList.vue
index 89026b3..8404bcf 100644
--- a/src/components/xt-chat/xt-bubble/xt-bubbleList.vue
+++ b/src/components/xt-chat/xt-bubble/xt-bubbleList.vue
@@ -105,6 +105,7 @@ export default defineComponent({
// 暴露控制方法
const abortTypingByKey = (key: string | number) => {
bubbleRefs.value[key]?.abortTyping?.();
+ console.log('abortTypingByKey----', bubbleRefs.value[key])
};
// 对外暴露能力
expose({
diff --git a/src/types/chat.ts b/src/types/chat.ts
index 2116a91..5e25cc6 100644
--- a/src/types/chat.ts
+++ b/src/types/chat.ts
@@ -1,7 +1,9 @@
declare global {
namespace CHAT {
- export type TInputInfo = {
+ type TInputInfo = {
message: string;
};
}
}
+
+export default {};
diff --git a/src/types/message.ts b/src/types/message.ts
new file mode 100644
index 0000000..fcd7483
--- /dev/null
+++ b/src/types/message.ts
@@ -0,0 +1,19 @@
+declare global {
+ namespace MESSAGE {
+ type TASK_STATUS = 'TeamRunStarted' | 'TeamRunResponseContent' | 'TeamRunCompleted';
+
+ interface Answer {
+ message: string;
+ node: string;
+ output: string;
+ run_id: string;
+ status: TASK_STATUS;
+ extra_data: {
+ type: string,
+ data: Record