瀏覽代碼

Feat: Display chat content on the agent page #3221 (#8140)

### What problem does this PR solve?

Feat: Display chat content on the agent page #3221

### Type of change


- [x] New Feature (non-breaking change which adds functionality)
tags/v0.19.1
balibabu 4 月之前
父節點
當前提交
9a69d5f367
沒有連結到貢獻者的電子郵件帳戶。

+ 11
- 8
web/src/pages/agent/chat/hooks.ts 查看文件

@@ -52,11 +52,13 @@ export const useSelectNextMessages = () => {
};

function findMessageFromList(eventList: IEventList) {
const event = eventList.find((x) => x.event === MessageEventType.Message) as
| IMessageEvent
| undefined;

return event?.data?.content;
const messageEventList = eventList.filter(
(x) => x.event === MessageEventType.Message,
) as IMessageEvent[];
return {
id: messageEventList[0]?.message_id,
content: messageEventList.map((x) => x.data.content).join(''),
};
}

const useGetBeginNodePrologue = () => {
@@ -127,10 +129,11 @@ export const useSendNextMessage = () => {
);

useEffect(() => {
const message = findMessageFromList(answerList);
if (message) {
const { content, id } = findMessageFromList(answerList);
if (content) {
addNewestAnswer({
answer: message,
answer: content,
id: id,
reference: {
chunks: [],
doc_aggs: [],

+ 18
- 4
web/src/pages/agent/constant.tsx 查看文件

@@ -27,6 +27,10 @@ import { ReactComponent as TemplateIcon } from '@/assets/svg/template.svg';
import { ReactComponent as TuShareIcon } from '@/assets/svg/tushare.svg';
import { ReactComponent as WenCaiIcon } from '@/assets/svg/wencai.svg';
import { ReactComponent as YahooFinanceIcon } from '@/assets/svg/yahoo-finance.svg';
import {
initialKeywordsSimilarityWeightValue,
initialSimilarityThresholdValue,
} from '@/components/similarity-slider';
import { CodeTemplateStrMap, ProgrammingLanguage } from '@/constants/agent';

export enum AgentDialogueMode {
@@ -437,10 +441,20 @@ const initialQueryBaseValues = {
};

export const initialRetrievalValues = {
similarity_threshold: 0.2,
keywords_similarity_weight: 0.3,
top_n: 8,
...initialQueryBaseValues,
query: '',
top_n: 0.2,
top_k: 1024,
kb_ids: [],
rerank_id: '',
empty_response: '',
...initialSimilarityThresholdValue,
...initialKeywordsSimilarityWeightValue,
outputs: {
formalized_content: {
type: 'string',
value: '',
},
},
};

export const initialBeginValues = {

+ 20
- 16
web/src/pages/agent/form/retrieval-form/next.tsx 查看文件

@@ -1,11 +1,7 @@
import { FormContainer } from '@/components/form-container';
import { KnowledgeBaseFormField } from '@/components/knowledge-base-item';
import { RerankFormFields } from '@/components/rerank';
import {
initialKeywordsSimilarityWeightValue,
initialSimilarityThresholdValue,
SimilaritySliderFormField,
} from '@/components/similarity-slider';
import { SimilaritySliderFormField } from '@/components/similarity-slider';
import { TopNFormField } from '@/components/top-n-item';
import {
Form,
@@ -17,11 +13,16 @@ import {
} from '@/components/ui/form';
import { Textarea } from '@/components/ui/textarea';
import { zodResolver } from '@hookform/resolvers/zod';
import { useMemo } from 'react';
import { useForm } from 'react-hook-form';
import { useTranslation } from 'react-i18next';
import { z } from 'zod';
import { initialRetrievalValues } from '../../constant';
import { useWatchFormChange } from '../../hooks/use-watch-form-change';
import { INextOperatorForm } from '../../interface';
import { Output } from '../components/output';
import { QueryVariable } from '../components/query-variable';
import { useValues } from './use-values';

const FormSchema = z.object({
query: z.string().optional(),
@@ -34,25 +35,27 @@ const FormSchema = z.object({
empty_response: z.string(),
});

const defaultValues = {
query: '',
top_n: 0.2,
top_k: 1024,
kb_ids: [],
rerank_id: '',
empty_response: '',
...initialSimilarityThresholdValue,
...initialKeywordsSimilarityWeightValue,
};

const RetrievalForm = ({ node }: INextOperatorForm) => {
const { t } = useTranslation();

const outputList = useMemo(() => {
return [
{
title: 'formalized_content',
type: initialRetrievalValues.outputs.formalized_content.type,
},
];
}, []);

const defaultValues = useValues(node);

const form = useForm({
defaultValues: defaultValues,
resolver: zodResolver(FormSchema),
});

useWatchFormChange(node?.id, form);

return (
<Form {...form}>
<form
@@ -92,6 +95,7 @@ const RetrievalForm = ({ node }: INextOperatorForm) => {
)}
/>
</FormContainer>
<Output list={outputList}></Output>
</form>
</Form>
);

+ 25
- 0
web/src/pages/agent/form/retrieval-form/use-values.ts 查看文件

@@ -0,0 +1,25 @@
import { RAGFlowNodeType } from '@/interfaces/database/flow';
import { isEmpty } from 'lodash';
import { useMemo } from 'react';
import { initialRetrievalValues } from '../../constant';

export function useValues(node?: RAGFlowNodeType) {
const defaultValues = useMemo(
() => ({
...initialRetrievalValues,
}),
[],
);

const values = useMemo(() => {
const formData = node?.data?.form;

if (isEmpty(formData)) {
return defaultValues;
}

return formData;
}, [defaultValues, node?.data?.form]);

return values;
}

+ 1
- 6
web/src/pages/agent/hooks/use-get-begin-query.tsx 查看文件

@@ -141,11 +141,6 @@ export const useBuildComponentIdSelectOptions = (
const options = useMemo(() => {
const query: BeginQuery[] = getBeginNodeDataQuery();
return [
{
label: <span>Component Output</span>,
title: 'Component Output',
options: componentIdOptions,
},
{
label: <span>Begin Input</span>,
title: 'Begin Input',
@@ -156,7 +151,7 @@ export const useBuildComponentIdSelectOptions = (
},
...nodeOutputOptions,
];
}, [componentIdOptions, getBeginNodeDataQuery, nodeOutputOptions]);
}, [getBeginNodeDataQuery, nodeOutputOptions]);

return options;
};

Loading…
取消
儲存