Skip to content

Commit

Permalink
Feat: Add ChatBasicSetting component infiniflow#3221 (infiniflow#4876)
Browse files Browse the repository at this point in the history
### What problem does this PR solve?

Feat: Add ChatBasicSetting component infiniflow#3221

### Type of change


- [x] New Feature (non-breaking change which adds functionality)
  • Loading branch information
cike8899 authored Feb 11, 2025
1 parent ca16480 commit 521d25d
Show file tree
Hide file tree
Showing 8 changed files with 297 additions and 5 deletions.
55 changes: 53 additions & 2 deletions web/src/components/knowledge-base-item.tsx
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,12 @@ import { DocumentParserType } from '@/constants/knowledge';
import { useTranslate } from '@/hooks/common-hooks';
import { useFetchKnowledgeList } from '@/hooks/knowledge-hooks';
import { UserOutlined } from '@ant-design/icons';
import { Avatar, Form, Select, Space } from 'antd';
import { Avatar as AntAvatar, Form, Select, Space } from 'antd';
import { Book } from 'lucide-react';
import { useFormContext } from 'react-hook-form';
import { Avatar, AvatarFallback, AvatarImage } from './ui/avatar';
import { FormControl, FormField, FormItem, FormLabel } from './ui/form';
import { MultiSelect } from './ui/multi-select';

const KnowledgeBaseItem = () => {
const { t } = useTranslate('chat');
Expand All @@ -16,7 +21,7 @@ const KnowledgeBaseItem = () => {
const knowledgeOptions = filteredKnowledgeList.map((x) => ({
label: (
<Space>
<Avatar size={20} icon={<UserOutlined />} src={x.avatar} />
<AntAvatar size={20} icon={<UserOutlined />} src={x.avatar} />
{x.name}
</Space>
),
Expand Down Expand Up @@ -46,3 +51,49 @@ const KnowledgeBaseItem = () => {
};

export default KnowledgeBaseItem;

export function KnowledgeBaseFormField() {
const form = useFormContext();
const { t } = useTranslate('chat');

const { list: knowledgeList } = useFetchKnowledgeList(true);

const filteredKnowledgeList = knowledgeList.filter(
(x) => x.parser_id !== DocumentParserType.Tag,
);

const knowledgeOptions = filteredKnowledgeList.map((x) => ({
label: x.name,
value: x.id,
icon: () => (
<Avatar className="size-4 mr-2">
<AvatarImage src={x.avatar} />
<AvatarFallback>
<Book />
</AvatarFallback>
</Avatar>
),
}));

return (
<FormField
control={form.control}
name="kb_ids"
render={({ field }) => (
<FormItem>
<FormLabel>{t('knowledgeBases')}</FormLabel>
<FormControl>
<MultiSelect
options={knowledgeOptions}
onValueChange={field.onChange}
placeholder={t('knowledgeBasesMessage')}
variant="inverted"
maxCount={100}
{...field}
/>
</FormControl>
</FormItem>
)}
/>
);
}
3 changes: 0 additions & 3 deletions web/src/pages/next-chats/chat/app-settings.tsx

This file was deleted.

163 changes: 163 additions & 0 deletions web/src/pages/next-chats/chat/app-settings/chat-basic-settings.tsx
Original file line number Diff line number Diff line change
@@ -0,0 +1,163 @@
'use client';

import { zodResolver } from '@hookform/resolvers/zod';
import { useForm } from 'react-hook-form';
import { z } from 'zod';

import { FileUploader } from '@/components/file-uploader';
import { KnowledgeBaseFormField } from '@/components/knowledge-base-item';
import {
Form,
FormControl,
FormField,
FormItem,
FormLabel,
FormMessage,
} from '@/components/ui/form';
import { Input } from '@/components/ui/input';
import { useTranslate } from '@/hooks/common-hooks';
import { Subhead } from './subhead';
import { SwitchFormField } from './switch-fom-field';

export default function ChatBasicSetting() {
const { t } = useTranslate('chat');

const promptConfigSchema = z.object({
quote: z.boolean(),
keyword: z.boolean(),
tts: z.boolean(),
empty_response: z.string().min(1, {
message: t('emptyResponse'),
}),
prologue: z.string().min(2, {}),
});

const formSchema = z.object({
name: z.string().min(1, { message: t('assistantNameMessage') }),
icon: z.array(z.instanceof(File)),
language: z.string().min(1, {
message: 'Username must be at least 2 characters.',
}),
description: z.string(),
kb_ids: z.array(z.string()).min(0, {
message: 'Username must be at least 1 characters.',
}),
prompt_config: promptConfigSchema,
});

const form = useForm<z.infer<typeof formSchema>>({
resolver: zodResolver(formSchema),
defaultValues: {
name: '',
language: 'English',
prompt_config: {
quote: true,
keyword: false,
tts: false,
},
},
});

function onSubmit(values: z.infer<typeof formSchema>) {
console.log(values);
}

return (
<section>
<Subhead>Basic settings</Subhead>
<Form {...form}>
<form onSubmit={form.handleSubmit(onSubmit)} className="space-y-8">
<FormField
control={form.control}
name="icon"
render={({ field }) => (
<div className="space-y-6">
<FormItem className="w-full">
<FormLabel>{t('assistantAvatar')}</FormLabel>
<FormControl>
<FileUploader
value={field.value}
onValueChange={field.onChange}
maxFileCount={1}
maxSize={4 * 1024 * 1024}
// progresses={progresses}
// pass the onUpload function here for direct upload
// onUpload={uploadFiles}
// disabled={isUploading}
/>
</FormControl>
<FormMessage />
</FormItem>
</div>
)}
/>
<FormField
control={form.control}
name="name"
render={({ field }) => (
<FormItem>
<FormLabel>{t('assistantName')}</FormLabel>
<FormControl>
<Input {...field}></Input>
</FormControl>
<FormMessage />
</FormItem>
)}
/>
<FormField
control={form.control}
name="description"
render={({ field }) => (
<FormItem>
<FormLabel>{t('description')}</FormLabel>
<FormControl>
<Input {...field}></Input>
</FormControl>
<FormMessage />
</FormItem>
)}
/>
<FormField
control={form.control}
name={'prompt_config.empty_response'}
render={({ field }) => (
<FormItem>
<FormLabel>{t('emptyResponse')}</FormLabel>
<FormControl>
<Input {...field}></Input>
</FormControl>
<FormMessage />
</FormItem>
)}
/>
<FormField
control={form.control}
name={'prompt_config.prologue'}
render={({ field }) => (
<FormItem>
<FormLabel>{t('setAnOpener')}</FormLabel>
<FormControl>
<Input {...field}></Input>
</FormControl>
<FormMessage />
</FormItem>
)}
/>
<SwitchFormField
name={'prompt_config.quote'}
label={t('quote')}
></SwitchFormField>
<SwitchFormField
name={'prompt_config.keyword'}
label={t('keyword')}
></SwitchFormField>
<SwitchFormField
name={'prompt_config.tts'}
label={t('tts')}
></SwitchFormField>
<KnowledgeBaseFormField></KnowledgeBaseFormField>
</form>
</Form>
</section>
);
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
import { Subhead } from './subhead';

export function ChatModelSettings() {
return (
<section>
<Subhead>Model Setting</Subhead>
</section>
);
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
import { Subhead } from './subhead';

export function ChatPromptEngine() {
return (
<section>
<Subhead>Prompt Engine</Subhead>
</section>
);
}
16 changes: 16 additions & 0 deletions web/src/pages/next-chats/chat/app-settings/index.tsx
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
import ChatBasicSetting from './chat-basic-settings';
import { ChatModelSettings } from './chat-model-settings';
import { ChatPromptEngine } from './chat-prompt-engine';

export function AppSettings() {
return (
<section className="p-6 w-[500px] max-w-[25%]">
<div className="text-2xl font-bold mb-4 text-colors-text-neutral-strong">
App settings
</div>
<ChatBasicSetting></ChatBasicSetting>
<ChatPromptEngine></ChatPromptEngine>
<ChatModelSettings></ChatModelSettings>
</section>
);
}
9 changes: 9 additions & 0 deletions web/src/pages/next-chats/chat/app-settings/subhead.tsx
Original file line number Diff line number Diff line change
@@ -0,0 +1,9 @@
import { PropsWithChildren } from 'react';

export function Subhead({ children }: PropsWithChildren) {
return (
<div className="text-xl font-bold mb-4 text-colors-text-neutral-strong">
{children}
</div>
);
}
38 changes: 38 additions & 0 deletions web/src/pages/next-chats/chat/app-settings/switch-fom-field.tsx
Original file line number Diff line number Diff line change
@@ -0,0 +1,38 @@
import {
FormControl,
FormField,
FormItem,
FormLabel,
} from '@/components/ui/form';
import { Switch } from '@/components/ui/switch';
import { ReactNode } from 'react';
import { useFormContext } from 'react-hook-form';

interface SwitchFormItemProps {
name: string;
label: ReactNode;
}

export function SwitchFormField({ label, name }: SwitchFormItemProps) {
const form = useFormContext();

return (
<FormField
control={form.control}
name={name}
render={({ field }) => (
<FormItem className="flex justify-between">
<FormLabel className="text-base">{label}</FormLabel>
<FormControl>
<Switch
checked={field.value}
onCheckedChange={field.onChange}
aria-readonly
className="!m-0"
/>
</FormControl>
</FormItem>
)}
/>
);
}

0 comments on commit 521d25d

Please sign in to comment.