github/DB-GPT
1
0
Fork 0
mirror of https://github.com/csunny/DB-GPT.git synced 2025-09-11 22:09:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat(core): Support max output tokens parameter (#2218)

Browse Source
This commit is contained in:
Fangyin Cheng
2024-12-18 17:19:18 +08:00
committed by GitHub
parent b05febbf77
commit 7f4b5e79cf
121 changed files with 333 additions and 219 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
web/pages/construct/app/extra/components/NativeApp.tsx
View File

@@ -21,6 +21,7 @@ interface FormProps {
bind_value?: string;
model?: string;
temperature?: number;
max_new_tokens?: number;
}
const NativeApp: React.FC<{
@@ -34,6 +35,7 @@ const NativeApp: React.FC<{
const bindValue = Form.useWatch('bind_value', form);
const model = Form.useWatch('model', form);
const temperature = Form.useWatch('temperature', form);
const max_new_tokens = Form.useWatch('max_new_tokens', form);
const { team_context, param_need } = initValue || {};
@@ -47,6 +49,7 @@ const NativeApp: React.FC<{
form.setFieldValue('chat_scene', team_context?.chat_scene);
form.setFieldValue('model', param_need?.find(param => param.type === 'model')?.value);
form.setFieldValue('temperature', param_need?.find(param => param.type === 'temperature')?.value);
form.setFieldValue('max_new_tokens', param_need?.find(param => param.type === 'max_new_tokens')?.value);
await run(param_need?.find(param => param.type === 'resource')?.value || '');
return [types, models] ?? [];
});
@@ -112,6 +115,7 @@ const NativeApp: React.FC<{
[
{ type: 'model', value: rawVal.model },
{ type: 'temperature', value: rawVal.temperature },
{ type: 'max_new_tokens', value: rawVal.max_new_tokens },
{
type: 'resource',
value: appTypeOptions
@@ -122,7 +126,7 @@ const NativeApp: React.FC<{
],
],
]);
}, [form, chatScene, bindValue, model, temperature, updateData, appTypeOptions, loading]);
}, [form, chatScene, bindValue, model, temperature, max_new_tokens, updateData, appTypeOptions, loading]);
useEffect(() => {
const type = (data?.[0]?.[1]?.find((type: any) => type.chat_scene === chatScene) as any)?.param_need?.find(
@@ -173,6 +177,9 @@ const NativeApp: React.FC<{
<Form.Item label={t('temperature')} tooltip name='temperature'>
<InputNumber className='w-1/5 h-8' max={1} min={0} step={0.1} placeholder={t('please_input_temperature')} />
</Form.Item>
<Form.Item label={t('max_new_tokens')} tooltip name='max_new_tokens'>
<InputNumber className='w-1/5 h-8' min={1} placeholder={t('please_input_max_new_tokens')} />
</Form.Item>
</Form>
</div>
);