Feat: The agent can only retrieve content from the knowledge base or memory. #4213 (#12224)

### What problem does this PR solve? Feat: The agent can only retrieve content from the knowledge base or memory. #4213 ### Type of change - [x] New Feature (non-breaking change which adds functionality)
2025-12-26 17:16:52 +08:00 · 2025-12-26 12:10:13 +08:00
parent 6044314811
commit cbcbbc41af
6 changed files with 90 additions and 23 deletions
--- a/web/src/layouts/next-header.tsx
+++ b/web/src/layouts/next-header.tsx
@ -79,7 +79,7 @@ export function Header() {
      { path: Routes.Chats, name: t('header.chat'), icon: MessageSquareText },
      { path: Routes.Searches, name: t('header.search'), icon: Search },
      { path: Routes.Agents, name: t('header.flow'), icon: Cpu },
-      { path: Routes.Memories, name: t('header.Memories'), icon: Cpu },
+      { path: Routes.Memories, name: t('header.memories'), icon: Cpu },
      { path: Routes.Files, name: t('header.fileManager'), icon: File },
    ],
    [t],
--- a/web/src/locales/en.ts
+++ b/web/src/locales/en.ts
@ -99,7 +99,7 @@ export default {
      search: 'Search',
      welcome: 'Welcome to',
      dataset: 'Dataset',
-      Memories: 'Memory',
+      memories: 'Memory',
    },
    memories: {
      llmTooltip:
@ -2169,7 +2169,7 @@ Important structured information may include: names, dates, locations, events, k
        agentStatus: 'Agent status:',
      },
      saveToMemory: 'Save to memory',
-      memory: 'Memory',
+      retrievalFrom: 'Retrieval from',
    },
    llmTools: {
      bad_calculator: {
--- a/web/src/locales/zh.ts
+++ b/web/src/locales/zh.ts
@ -93,7 +93,7 @@ export default {
      search: '搜索',
      welcome: '欢迎来到',
      dataset: '知识库',
-      Memories: '记忆',
+      memories: '记忆',
    },
    memories: {
      llmTooltip: '分析对话内容，提取关键信息，并生成结构化的记忆摘要。',
@ -1970,7 +1970,7 @@ Tokenizer 会根据所选方式将内容存储为对应的数据结构。`,
        agentStatus: '智能体状态：',
      },
      saveToMemory: '保存到Memory',
-      memory: 'Memory',
+      retrievalFrom: '检索来源',
    },
    footer: {
      profile: 'All rights reserved @ React',
--- a/web/src/pages/agent/constant/index.tsx
+++ b/web/src/pages/agent/constant/index.tsx
@ -79,6 +79,11 @@ export const DataOperationsOperatorOptions = [

 export const SwitchElseTo = 'end_cpn_ids';

+export enum RetrievalFrom {
+  Dataset = 'dataset',
+  Memory = 'memory',
+}
+
 export const initialRetrievalValues = {
  query: AgentGlobalsSysQueryWithBrace,
  top_n: 8,
@ -91,6 +96,7 @@ export const initialRetrievalValues = {
  use_kg: false,
  toc_enhance: false,
  cross_languages: [],
+  retrieval_from: RetrievalFrom.Dataset,
  outputs: {
    formalized_content: {
      type: 'string',
--- a/web/src/pages/agent/form/retrieval-form/next.tsx
+++ b/web/src/pages/agent/form/retrieval-form/next.tsx
@ -20,14 +20,20 @@ import {
  FormLabel,
  FormMessage,
 } from '@/components/ui/form';
+import { Radio } from '@/components/ui/radio';
 import { Textarea } from '@/components/ui/textarea';
 import { UseKnowledgeGraphFormField } from '@/components/use-knowledge-graph-item';
 import { zodResolver } from '@hookform/resolvers/zod';
 import { memo, useMemo } from 'react';
-import { useForm, useFormContext } from 'react-hook-form';
+import {
+  UseFormReturn,
+  useForm,
+  useFormContext,
+  useWatch,
+} from 'react-hook-form';
 import { useTranslation } from 'react-i18next';
 import { z } from 'zod';
-import { initialRetrievalValues } from '../../constant';
+import { RetrievalFrom, initialRetrievalValues } from '../../constant';
 import { useWatchFormChange } from '../../hooks/use-watch-form-change';
 import { INextOperatorForm } from '../../interface';
 import { FormWrapper } from '../components/form-wrapper';
@ -55,6 +61,42 @@ export const FormSchema = z.object({
  ...RetrievalPartialSchema,
 });

+export function MemoryDatasetForm() {
+  const { t } = useTranslation();
+  const form = useFormContext();
+  const retrievalFrom = useWatch({
+    control: form.control,
+    name: 'retrieval_from',
+  });
+
+  return (
+    <>
+      <RAGFlowFormItem name="retrieval_from" label={t('flow.retrievalFrom')}>
+        <Radio.Group>
+          <Radio value={RetrievalFrom.Dataset}>
+            {t('knowledgeDetails.dataset')}
+          </Radio>
+          <Radio value={RetrievalFrom.Memory}>{t('header.memories')}</Radio>
+        </Radio.Group>
+      </RAGFlowFormItem>
+      {retrievalFrom === RetrievalFrom.Memory ? (
+        <MemoriesFormField label={t('header.memories')}></MemoriesFormField>
+      ) : (
+        <KnowledgeBaseFormField showVariable></KnowledgeBaseFormField>
+      )}
+    </>
+  );
+}
+
+export function useHideKnowledgeGraphField(form: UseFormReturn<any>) {
+  const retrievalFrom = useWatch({
+    control: form.control,
+    name: 'retrieval_from',
+  });
+
+  return retrievalFrom === RetrievalFrom.Memory;
+}
+
 export function EmptyResponseField() {
  const { t } = useTranslation();
  const form = useFormContext();
@ -106,6 +148,8 @@ function RetrievalForm({ node }: INextOperatorForm) {
    resolver: zodResolver(FormSchema),
  });

+  const hideKnowledgeGraphField = useHideKnowledgeGraphField(form);
+
  useWatchFormChange(node?.id, form);

  return (
@ -114,8 +158,7 @@ function RetrievalForm({ node }: INextOperatorForm) {
        <RAGFlowFormItem name="query" label={t('flow.query')}>
          <PromptEditor></PromptEditor>
        </RAGFlowFormItem>
-        <KnowledgeBaseFormField showVariable></KnowledgeBaseFormField>
-        <MemoriesFormField label={t('flow.memory')}></MemoriesFormField>
+        <MemoryDatasetForm></MemoryDatasetForm>
        <Collapse title={<div>{t('flow.advancedSettings')}</div>}>
          <FormContainer>
            <SimilaritySliderFormField
@ -123,12 +166,20 @@ function RetrievalForm({ node }: INextOperatorForm) {
              isTooltipShown
            ></SimilaritySliderFormField>
            <TopNFormField></TopNFormField>
-            <RerankFormFields></RerankFormFields>
-            <MetadataFilter canReference></MetadataFilter>
+            {hideKnowledgeGraphField || (
+              <>
+                <RerankFormFields></RerankFormFields>
+                <MetadataFilter canReference></MetadataFilter>
+              </>
+            )}
            <EmptyResponseField></EmptyResponseField>
-            <CrossLanguageFormField name="cross_languages"></CrossLanguageFormField>
-            <UseKnowledgeGraphFormField name="use_kg"></UseKnowledgeGraphFormField>
-            <TOCEnhanceFormField name="toc_enhance"></TOCEnhanceFormField>
+            {hideKnowledgeGraphField || (
+              <>
+                <CrossLanguageFormField name="cross_languages"></CrossLanguageFormField>
+                <UseKnowledgeGraphFormField name="use_kg"></UseKnowledgeGraphFormField>
+                <TOCEnhanceFormField name="toc_enhance"></TOCEnhanceFormField>
+              </>
+            )}
          </FormContainer>
        </Collapse>
        <Output list={outputList}></Output>
--- a/web/src/pages/agent/form/tool-form/retrieval-form/index.tsx
+++ b/web/src/pages/agent/form/tool-form/retrieval-form/index.tsx
@ -1,8 +1,6 @@
 import { Collapse } from '@/components/collapse';
 import { CrossLanguageFormField } from '@/components/cross-language-form-field';
 import { FormContainer } from '@/components/form-container';
-import { KnowledgeBaseFormField } from '@/components/knowledge-base-item';
-import { MemoriesFormField } from '@/components/memories-form-field';
 import { MetadataFilter } from '@/components/metadata-filter';
 import { RerankFormFields } from '@/components/rerank';
 import { SimilaritySliderFormField } from '@/components/similarity-slider';
@ -18,7 +16,9 @@ import { DescriptionField } from '../../components/description-field';
 import { FormWrapper } from '../../components/form-wrapper';
 import {
  EmptyResponseField,
+  MemoryDatasetForm,
  RetrievalPartialSchema,
+  useHideKnowledgeGraphField,
 } from '../../retrieval-form/next';
 import { useValues } from '../use-values';
 import { useWatchFormChange } from '../use-watch-change';
@ -36,14 +36,15 @@ const RetrievalForm = () => {
    resolver: zodResolver(FormSchema),
  });

+  const hideKnowledgeGraphField = useHideKnowledgeGraphField(form);
+
  useWatchFormChange(form);

  return (
    <Form {...form}>
      <FormWrapper>
        <DescriptionField></DescriptionField>
-        <KnowledgeBaseFormField showVariable></KnowledgeBaseFormField>
-        <MemoriesFormField label={t('flow.memory')}></MemoriesFormField>
+        <MemoryDatasetForm></MemoryDatasetForm>
        <Collapse title={<div>{t('flow.advancedSettings')}</div>}>
          <FormContainer>
            <SimilaritySliderFormField
@ -51,12 +52,21 @@ const RetrievalForm = () => {
              isTooltipShown
            ></SimilaritySliderFormField>
            <TopNFormField></TopNFormField>
-            <RerankFormFields></RerankFormFields>
-            <MetadataFilter canReference></MetadataFilter>
+            {hideKnowledgeGraphField || (
+              <>
+                <RerankFormFields></RerankFormFields>
+                <MetadataFilter canReference></MetadataFilter>
+              </>
+            )}
+
            <EmptyResponseField></EmptyResponseField>
-            <CrossLanguageFormField name="cross_languages"></CrossLanguageFormField>
-            <UseKnowledgeGraphFormField name="use_kg"></UseKnowledgeGraphFormField>
-            <TOCEnhanceFormField name="toc_enhance"></TOCEnhanceFormField>
+            {hideKnowledgeGraphField || (
+              <>
+                <CrossLanguageFormField name="cross_languages"></CrossLanguageFormField>
+                <UseKnowledgeGraphFormField name="use_kg"></UseKnowledgeGraphFormField>
+                <TOCEnhanceFormField name="toc_enhance"></TOCEnhanceFormField>
+              </>
+            )}
          </FormContainer>
        </Collapse>
      </FormWrapper>