adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(web): workflow add opening_statement

Browse Source
This commit is contained in:
zhaoying
2026-04-02 19:32:47 +08:00
parent b20a65ce29
commit d60cb423a4
3 changed files with 73 additions and 44 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
web/src/views/ApplicationConfig/TestChat/index.tsx
View File

@@ -2,7 +2,7 @@
* @Author: ZhaoYing
* @Date: 2026-03-13 17:27:52
* @Last Modified by: ZhaoYing
* @Last Modified time: 2026-03-31 16:04:15
* @Last Modified time: 2026-04-02 17:58:07
*/
import { type FC, useState, useRef, useEffect } from 'react'
import { useTranslation } from 'react-i18next'
@@ -63,6 +63,12 @@ interface NodeData {
state: Record<string, any>;
status?: 'completed' | 'failed';
audio_url?: string;
citations?: {
document_id: string;
file_name: string;
knowledge_id: string;
score: string;
}[]
}
const TestChat: FC<TestChatProps> = ({
@@ -111,8 +117,7 @@ const TestChat: FC<TestChatProps> = ({
}
}])
}
let initVariables: Variable[] = []
switch (application.type) {
@@ -162,7 +167,7 @@ const TestChat: FC<TestChatProps> = ({
}])
}
const updateAssistantMessage = (content: string, audio_url?: string, audio_status?: string, citations?: any[]) => {
const updateAssistantMessage = (content: string, audio_url?: string, audio_status?: string, citations?: NodeData['citations']) => {
setChatList(prev => {
const newList = [...prev]
const lastMsg = newList[newList.length - 1]
@@ -281,12 +286,7 @@ const TestChat: FC<TestChatProps> = ({
data.map(item => {
const { conversation_id, content, message_length, audio_url, citations } = item.data as {
conversation_id: string, content: string, message_length: number; audio_url?: string;
citations?: {
document_id: string;
file_name: string;
knowledge_id: string;
score: string;
}[]
citations?: NodeData['citations']
};
switch (item.event) {
case 'start':
@@ -344,15 +344,15 @@ const TestChat: FC<TestChatProps> = ({
})
}
const handleWorkflowSend = () => {
if (loading || !application || !message || !message?.trim()) return
const handleWorkflowSend = (msg?: string) => {
if (loading || !application || !((message && message?.trim() !== '') || (msg && msg?.trim() !== ''))) return
const files = (toolbarRef.current?.getFiles() || []).filter(item => !['uploading', 'error'].includes(item.status))
const variables = toolbarRef.current?.getVariables() || []
const { isCanSend, params } = buildVariableParams(variables)
if (!isCanSend) return
setLoading(true)
addUserMessage(message, files)
addUserMessage((msg || message) as string, files)
addAssistantMessage()
toolbarRef.current?.setFiles([])
setFileList([])
@@ -361,7 +361,7 @@ const TestChat: FC<TestChatProps> = ({
draftRun(
application.id,
formatParams(message, conversationId, files, params),
formatParams((msg || message) as string, conversationId, files, params),
handleWorkflowStreamMessage
)
.catch((error) => {
@@ -383,7 +383,7 @@ const TestChat: FC<TestChatProps> = ({
const handleWorkflowStreamMessage = (data: SSEMessage[]) => {
data.forEach(item => {
const { content, conversation_id } = item.data as NodeData;
const { content, conversation_id, citations } = item.data as NodeData;
switch (item.event) {
// Append streaming text chunks to assistant message
case 'message':
@@ -412,6 +412,9 @@ const TestChat: FC<TestChatProps> = ({
// Mark workflow as complete
case 'workflow_end':
updateWorkflowEndMessage(item.data as NodeData)
if (citations && citations.length > 0) {
updateWorkflowEndMessage(item.data as NodeData, citations)
}
setStreamLoading(false)
setLoading(false)
break
@@ -536,7 +539,7 @@ const TestChat: FC<TestChatProps> = ({
})
}
const updateWorkflowEndMessage = (data: NodeData) => {
const updateWorkflowEndMessage = (data: NodeData, citations?: NodeData['citations']) => {
const { error, status } = data;
setChatList(prev => {
const newList = [...prev]
@@ -547,6 +550,10 @@ const TestChat: FC<TestChatProps> = ({
status,
error,
content: newList[lastIndex].content === '' ? null : newList[lastIndex].content,
meta_data: {
...newList[lastIndex].meta_data || {},
citations
}
}
}
return newList

50
web/src/views/ApplicationConfig/components/FeaturesConfig/FeaturesConfigModal.tsx
View File

@@ -2,7 +2,7 @@
* @Author: ZhaoYing
* @Date: 2026-02-03 16:27:56
* @Last Modified by: ZhaoYing
* @Last Modified time: 2026-03-27 17:32:10
* @Last Modified time: 2026-04-02 17:49:51
*/
/**
* Copy Application Modal
@@ -116,24 +116,24 @@ const FeaturesConfigModal = forwardRef<FeaturesConfigModalRef, FeaturesConfigMod
layout="vertical"
>
<Flex vertical gap={12}>
<div className="rb:relative rb:border rb:border-[#DFE4ED] rb:p-3 rb:rounded-lg rb:bg-[#f5f7fc]">
<SwitchFormItem
title={t('application.opening_statement')}
name={['opening_statement', "enabled"]}
desc={values?.opening_statement?.enabled ? undefined : t('application.opening_statement_desc')}
/>
{values?.opening_statement?.enabled && (() => {
const statement = values.opening_statement?.statement
return statement && statement.trim() !== '' ? <>
<div className="rb:bg-white rb:rounded-lg rb:py-1 rb:px-3 rb:mb-1">
{statement}
</div>
<Button block onClick={handleOpenStatementSettings}>{t('application.editOpeningStatement')}</Button>
</> : <Button block onClick={handleOpenStatementSettings}>{t('application.editOpeningStatement')}</Button>
})()}
<Form.Item name="opening_statement" hidden />
</div>
{source !== 'workflow' && <>
<div className="rb:relative rb:border rb:border-[#DFE4ED] rb:p-3 rb:rounded-lg rb:bg-[#f5f7fc]">
<SwitchFormItem
title={t('application.opening_statement')}
name={['opening_statement', "enabled"]}
desc={values?.opening_statement?.enabled ? undefined : t('application.opening_statement_desc')}
/>
{values?.opening_statement?.enabled && (() => {
const statement = values.opening_statement?.statement
return statement && statement.trim() !== '' ? <>
<div className="rb:bg-white rb:rounded-lg rb:py-1 rb:px-3 rb:mb-1">
{statement}
</div>
<Button block onClick={handleOpenStatementSettings}>{t('application.editOpeningStatement')}</Button>
</> : <Button block onClick={handleOpenStatementSettings}>{t('application.editOpeningStatement')}</Button>
})()}
<Form.Item name="opening_statement" hidden />
</div>
<div className="rb:relative rb:border rb:border-[#DFE4ED] rb:p-3 rb:rounded-lg rb:bg-[#f5f7fc]">
<SwitchFormItem
title={t(`memoryConversation.web_search`)}
@@ -148,14 +148,14 @@ const FeaturesConfigModal = forwardRef<FeaturesConfigModalRef, FeaturesConfigMod
desc={t('application.text_to_speech_desc')}
/>
</div>
<div className="rb:relative rb:border rb:border-[#DFE4ED] rb:p-3 rb:rounded-lg rb:bg-[#f5f7fc]">
<SwitchFormItem
title={t(`application.citation`)}
name={['citation', "enabled"]}
desc={t('application.citation_desc')}
/>
</div>
</>}
<div className="rb:relative rb:border rb:border-[#DFE4ED] rb:p-3 rb:rounded-lg rb:bg-[#f5f7fc]">
<SwitchFormItem
title={t(`application.citation`)}
name={['citation', "enabled"]}
desc={t('application.citation_desc')}
/>
</div>
<div className="rb:relative rb:border rb:border-[#DFE4ED] rb:p-3 rb:rounded-lg rb:bg-[#f5f7fc]">
<SwitchFormItem

28
web/src/views/Workflow/components/Chat/Chat.tsx
View File

@@ -2,7 +2,7 @@
* @Author: ZhaoYing
* @Date: 2026-02-06 21:10:56
* @Last Modified by: ZhaoYing
* @Last Modified time: 2026-03-27 17:30:47
* @Last Modified time: 2026-04-02 18:01:09
*/
/**
* Workflow Chat Component
@@ -66,6 +66,17 @@ const Chat = forwardRef<ChatRef, { appId: string; graphRef: GraphRef; data: Work
*/
const handleOpen = () => {
setOpen(true)
if (features?.opening_statement?.statement && features?.opening_statement?.statement.trim() !== '') {
setChatList(prev => [...prev, {
role: 'assistant',
created_at: Date.now(),
content: features?.opening_statement?.statement,
meta_data: {
suggested_questions: features?.opening_statement?.suggested_questions || []
}
}])
}
}
useEffect(() => {
@@ -164,7 +175,7 @@ const Chat = forwardRef<ChatRef, { appId: string; graphRef: GraphRef; data: Work
*/
const handleStreamMessage = (data: SSEMessage[]) => {
data.forEach(item => {
const { content, conversation_id, node_id, cycle_id, cycle_idx, input, output, error, elapsed_time, status } = item.data as {
const { content, conversation_id, node_id, cycle_id, cycle_idx, input, output, error, elapsed_time, status, citations } = item.data as {
content: string;
conversation_id: string | null;
cycle_id: string;
@@ -177,7 +188,13 @@ const Chat = forwardRef<ChatRef, { appId: string; graphRef: GraphRef; data: Work
elapsed_time?: string;
error?: any;
state: Record<string, any>;
status?: 'completed' | 'failed'
status?: 'completed' | 'failed',
citations?: {
document_id: string;
file_name: string;
knowledge_id: string;
score: string;
}[]
};
const node = graphRef.current?.getNodes().find(n => n.id === node_id);
@@ -312,6 +329,10 @@ const Chat = forwardRef<ChatRef, { appId: string; graphRef: GraphRef; data: Work
status,
error,
content: newList[lastIndex].content === '' ? null : newList[lastIndex].content,
meta_data: {
...newList[lastIndex].meta_data || {},
citations
}
}
}
return newList
@@ -421,6 +442,7 @@ const Chat = forwardRef<ChatRef, { appId: string; graphRef: GraphRef; data: Work
renderRuntime={(item, index) => {
return <Runtime item={item} index={index} />
}}
onSend={handleSend}
/>
<Flex align="center" gap={10} className="rb:relative rb:m-4! rb:mb-1!">
<ChatInput