Feat: Enable logical operators in metadata. #11387 #11376 (#11442)

### What problem does this PR solve?

Feat: Enable logical operators in metadata. #11387  #11376
### Type of change


- [x] New Feature (non-breaking change which adds functionality)
This commit is contained in:
balibabu
2025-11-21 16:21:27 +08:00
committed by GitHub
parent 249296e417
commit 13299197b8
6 changed files with 119 additions and 80 deletions

View File

@ -14,6 +14,7 @@ type MetadataFilterProps = {
export const MetadataFilterSchema = { export const MetadataFilterSchema = {
meta_data_filter: z meta_data_filter: z
.object({ .object({
logic: z.string().optional(),
method: z.string().optional(), method: z.string().optional(),
manual: z manual: z
.array( .array(

View File

@ -15,14 +15,17 @@ import {
} from '@/components/ui/form'; } from '@/components/ui/form';
import { Input } from '@/components/ui/input'; import { Input } from '@/components/ui/input';
import { Separator } from '@/components/ui/separator'; import { Separator } from '@/components/ui/separator';
import { SwitchOperatorOptions } from '@/constants/agent'; import { SwitchLogicOperator, SwitchOperatorOptions } from '@/constants/agent';
import { useBuildSwitchOperatorOptions } from '@/hooks/logic-hooks/use-build-operator-options'; import { useBuildSwitchOperatorOptions } from '@/hooks/logic-hooks/use-build-operator-options';
import { useBuildSwitchLogicOperatorOptions } from '@/hooks/logic-hooks/use-build-options';
import { useFetchKnowledgeMetadata } from '@/hooks/use-knowledge-request'; import { useFetchKnowledgeMetadata } from '@/hooks/use-knowledge-request';
import { PromptEditor } from '@/pages/agent/form/components/prompt-editor'; import { PromptEditor } from '@/pages/agent/form/components/prompt-editor';
import { Plus, X } from 'lucide-react'; import { Plus, X } from 'lucide-react';
import { useCallback } from 'react'; import { useCallback } from 'react';
import { useFieldArray, useFormContext } from 'react-hook-form'; import { useFieldArray, useFormContext } from 'react-hook-form';
import { useTranslation } from 'react-i18next'; import { useTranslation } from 'react-i18next';
import { RAGFlowFormItem } from '../ragflow-form';
import { RAGFlowSelect } from '../ui/select';
export function MetadataFilterConditions({ export function MetadataFilterConditions({
kbIds, kbIds,
@ -36,10 +39,13 @@ export function MetadataFilterConditions({
const { t } = useTranslation(); const { t } = useTranslation();
const form = useFormContext(); const form = useFormContext();
const name = prefix + 'meta_data_filter.manual'; const name = prefix + 'meta_data_filter.manual';
const logic = prefix + 'meta_data_filter.logic';
const metadata = useFetchKnowledgeMetadata(kbIds); const metadata = useFetchKnowledgeMetadata(kbIds);
const switchOperatorOptions = useBuildSwitchOperatorOptions(); const switchOperatorOptions = useBuildSwitchOperatorOptions();
const switchLogicOperatorOptions = useBuildSwitchLogicOperatorOptions();
const { fields, remove, append } = useFieldArray({ const { fields, remove, append } = useFieldArray({
name, name,
control: form.control, control: form.control,
@ -47,13 +53,14 @@ export function MetadataFilterConditions({
const add = useCallback( const add = useCallback(
(key: string) => () => { (key: string) => () => {
form.setValue(logic, SwitchLogicOperator.And);
append({ append({
key, key,
value: '', value: '',
op: SwitchOperatorOptions[0].value, op: SwitchOperatorOptions[0].value,
}); });
}, },
[append], [append, form, logic],
); );
return ( return (
@ -77,11 +84,28 @@ export function MetadataFilterConditions({
</DropdownMenuContent> </DropdownMenuContent>
</DropdownMenu> </DropdownMenu>
</div> </div>
<div className="space-y-5"> <section className="flex">
{fields.length > 1 && (
<div className="relative min-w-14">
<RAGFlowFormItem
name={logic}
className="absolute top-1/2 -translate-y-1/2 right-1 left-0 z-10 bg-bg-base"
>
<RAGFlowSelect
options={switchLogicOperatorOptions}
triggerClassName="w-full text-xs px-1 py-0 h-6"
></RAGFlowSelect>
</RAGFlowFormItem>
<div className="absolute border-l border-y w-5 right-0 top-4 bottom-4 rounded-l-lg"></div>
</div>
)}
<div className="space-y-5 flex-1">
{fields.map((field, index) => { {fields.map((field, index) => {
const typeField = `${name}.${index}.key`; const typeField = `${name}.${index}.key`;
return ( return (
<div key={field.id} className="flex w-full items-center gap-2"> <section key={field.id} className="flex gap-2">
<div className="w-full space-y-2">
<div className="flex items-center gap-1">
<FormField <FormField
control={form.control} control={form.control}
name={typeField} name={typeField}
@ -97,7 +121,7 @@ export function MetadataFilterConditions({
</FormItem> </FormItem>
)} )}
/> />
<Separator className="w-3 text-text-secondary" /> <Separator className="w-1 text-text-secondary" />
<FormField <FormField
control={form.control} control={form.control}
name={`${name}.${index}.op`} name={`${name}.${index}.op`}
@ -113,7 +137,7 @@ export function MetadataFilterConditions({
</FormItem> </FormItem>
)} )}
/> />
<Separator className="w-3 text-text-secondary" /> </div>
<FormField <FormField
control={form.control} control={form.control}
name={`${name}.${index}.value`} name={`${name}.${index}.value`}
@ -137,13 +161,15 @@ export function MetadataFilterConditions({
</FormItem> </FormItem>
)} )}
/> />
</div>
<Button variant={'ghost'} onClick={() => remove(index)}> <Button variant={'ghost'} onClick={() => remove(index)}>
<X className="text-text-sub-title-invert " /> <X className="text-text-sub-title-invert " />
</Button> </Button>
</div> </section>
); );
})} })}
</div> </div>
</section> </section>
</section>
); );
} }

View File

@ -179,3 +179,8 @@ export enum JsonSchemaDataType {
Array = 'array', Array = 'array',
Object = 'object', Object = 'object',
} }
export enum SwitchLogicOperator {
And = 'and',
Or = 'or',
}

View File

@ -0,0 +1,12 @@
import { SwitchLogicOperator } from '@/constants/agent';
import { buildOptions } from '@/utils/form';
import { useTranslation } from 'react-i18next';
export function useBuildSwitchLogicOperatorOptions() {
const { t } = useTranslation();
return buildOptions(
SwitchLogicOperator,
t,
'flow.switchLogicOperatorOptions',
);
}

View File

@ -10,6 +10,7 @@ import {
JsonSchemaDataType, JsonSchemaDataType,
Operator, Operator,
ProgrammingLanguage, ProgrammingLanguage,
SwitchLogicOperator,
SwitchOperatorOptions, SwitchOperatorOptions,
initialLlmBaseValues, initialLlmBaseValues,
} from '@/constants/agent'; } from '@/constants/agent';
@ -51,8 +52,6 @@ import {
export const BeginId = 'begin'; export const BeginId = 'begin';
export const SwitchLogicOperatorOptions = ['and', 'or'];
export const CommonOperatorList = Object.values(Operator).filter( export const CommonOperatorList = Object.values(Operator).filter(
(x) => x !== Operator.Note, (x) => x !== Operator.Note,
); );
@ -308,7 +307,7 @@ export const initialExeSqlValues = {
export const initialSwitchValues = { export const initialSwitchValues = {
conditions: [ conditions: [
{ {
logical_operator: SwitchLogicOperatorOptions[0], logical_operator: SwitchLogicOperator.And,
items: [ items: [
{ {
operator: SwitchOperatorOptions[0].value, operator: SwitchOperatorOptions[0].value,

View File

@ -11,16 +11,17 @@ import {
import { RAGFlowSelect } from '@/components/ui/select'; import { RAGFlowSelect } from '@/components/ui/select';
import { Separator } from '@/components/ui/separator'; import { Separator } from '@/components/ui/separator';
import { Textarea } from '@/components/ui/textarea'; import { Textarea } from '@/components/ui/textarea';
import { SwitchLogicOperator } from '@/constants/agent';
import { useBuildSwitchOperatorOptions } from '@/hooks/logic-hooks/use-build-operator-options'; import { useBuildSwitchOperatorOptions } from '@/hooks/logic-hooks/use-build-operator-options';
import { useBuildSwitchLogicOperatorOptions } from '@/hooks/logic-hooks/use-build-options';
import { cn } from '@/lib/utils'; import { cn } from '@/lib/utils';
import { zodResolver } from '@hookform/resolvers/zod'; import { zodResolver } from '@hookform/resolvers/zod';
import { t } from 'i18next'; import { t } from 'i18next';
import { X } from 'lucide-react'; import { X } from 'lucide-react';
import { memo, useCallback, useMemo } from 'react'; import { memo, useCallback } from 'react';
import { useFieldArray, useForm, useFormContext } from 'react-hook-form'; import { useFieldArray, useForm, useFormContext } from 'react-hook-form';
import { useTranslation } from 'react-i18next'; import { useTranslation } from 'react-i18next';
import { z } from 'zod'; import { z } from 'zod';
import { SwitchLogicOperatorOptions } from '../../constant';
import { IOperatorForm } from '../../interface'; import { IOperatorForm } from '../../interface';
import { FormWrapper } from '../components/form-wrapper'; import { FormWrapper } from '../components/form-wrapper';
import { QueryVariable } from '../components/query-variable'; import { QueryVariable } from '../components/query-variable';
@ -185,12 +186,7 @@ function SwitchForm({ node }: IOperatorForm) {
control: form.control, control: form.control,
}); });
const switchLogicOperatorOptions = useMemo(() => { const switchLogicOperatorOptions = useBuildSwitchLogicOperatorOptions();
return SwitchLogicOperatorOptions.map((x) => ({
value: x,
label: t(`flow.switchLogicOperatorOptions.${x}`),
}));
}, [t]);
useWatchFormChange(node?.id, form); useWatchFormChange(node?.id, form);
@ -253,7 +249,7 @@ function SwitchForm({ node }: IOperatorForm) {
<BlockButton <BlockButton
onClick={() => onClick={() =>
append({ append({
logical_operator: SwitchLogicOperatorOptions[0], logical_operator: SwitchLogicOperator.And,
[ItemKey]: [ [ItemKey]: [
{ {
operator: switchOperatorOptions[0].value, operator: switchOperatorOptions[0].value,