feat: Add save functionality for system prompt and model settings
This commit is contained in:
parent
77f0cdbb35
commit
5687517238
@ -5,15 +5,17 @@ import { useStoreChatModelSettings } from "@/store/model"
|
|||||||
import { useQuery } from "@tanstack/react-query"
|
import { useQuery } from "@tanstack/react-query"
|
||||||
import {
|
import {
|
||||||
Collapse,
|
Collapse,
|
||||||
|
Divider,
|
||||||
Drawer,
|
Drawer,
|
||||||
Form,
|
Form,
|
||||||
Input,
|
Input,
|
||||||
InputNumber,
|
InputNumber,
|
||||||
Modal,
|
Modal,
|
||||||
Skeleton,
|
Skeleton,
|
||||||
Switch
|
Switch,
|
||||||
|
Button
|
||||||
} from "antd"
|
} from "antd"
|
||||||
import React from "react"
|
import React, { useState, useCallback } from "react"
|
||||||
import { useTranslation } from "react-i18next"
|
import { useTranslation } from "react-i18next"
|
||||||
|
|
||||||
type Props = {
|
type Props = {
|
||||||
@ -31,6 +33,19 @@ export const CurrentChatModelSettings = ({
|
|||||||
const [form] = Form.useForm()
|
const [form] = Form.useForm()
|
||||||
const cUserSettings = useStoreChatModelSettings()
|
const cUserSettings = useStoreChatModelSettings()
|
||||||
const { selectedSystemPrompt } = useMessageOption()
|
const { selectedSystemPrompt } = useMessageOption()
|
||||||
|
|
||||||
|
const savePrompt = useCallback((value: string) => {
|
||||||
|
cUserSettings.setX('systemPrompt', value)
|
||||||
|
}, [cUserSettings])
|
||||||
|
|
||||||
|
const saveSettings = useCallback((values: any) => {
|
||||||
|
Object.entries(values).forEach(([key, value]) => {
|
||||||
|
if (key !== 'systemPrompt') {
|
||||||
|
cUserSettings.setX(key, value)
|
||||||
|
}
|
||||||
|
})
|
||||||
|
}, [cUserSettings])
|
||||||
|
|
||||||
const { isPending: isLoading } = useQuery({
|
const { isPending: isLoading } = useQuery({
|
||||||
queryKey: ["fetchModelConfig2", open],
|
queryKey: ["fetchModelConfig2", open],
|
||||||
queryFn: async () => {
|
queryFn: async () => {
|
||||||
@ -68,19 +83,12 @@ export const CurrentChatModelSettings = ({
|
|||||||
<>
|
<>
|
||||||
{!isLoading ? (
|
{!isLoading ? (
|
||||||
<Form
|
<Form
|
||||||
onFinish={(values: {
|
|
||||||
keepAlive: string
|
|
||||||
temperature: number
|
|
||||||
topK: number
|
|
||||||
topP: number
|
|
||||||
}) => {
|
|
||||||
Object.entries(values).forEach(([key, value]) => {
|
|
||||||
cUserSettings.setX(key, value)
|
|
||||||
setOpen(false)
|
|
||||||
})
|
|
||||||
}}
|
|
||||||
form={form}
|
form={form}
|
||||||
layout="vertical">
|
layout="vertical"
|
||||||
|
onFinish={(values) => {
|
||||||
|
saveSettings(values)
|
||||||
|
setOpen(false)
|
||||||
|
}}>
|
||||||
{useDrawer && (
|
{useDrawer && (
|
||||||
<>
|
<>
|
||||||
<Form.Item
|
<Form.Item
|
||||||
@ -92,8 +100,10 @@ export const CurrentChatModelSettings = ({
|
|||||||
placeholder={t(
|
placeholder={t(
|
||||||
"modelSettings.form.systemPrompt.placeholder"
|
"modelSettings.form.systemPrompt.placeholder"
|
||||||
)}
|
)}
|
||||||
|
onChange={(e) => savePrompt(e.target.value)}
|
||||||
/>
|
/>
|
||||||
</Form.Item>
|
</Form.Item>
|
||||||
|
<Divider />
|
||||||
</>
|
</>
|
||||||
)}
|
)}
|
||||||
<Form.Item
|
<Form.Item
|
||||||
@ -113,6 +123,7 @@ export const CurrentChatModelSettings = ({
|
|||||||
placeholder={t("modelSettings.form.temperature.placeholder")}
|
placeholder={t("modelSettings.form.temperature.placeholder")}
|
||||||
/>
|
/>
|
||||||
</Form.Item>
|
</Form.Item>
|
||||||
|
|
||||||
<Form.Item
|
<Form.Item
|
||||||
name="seed"
|
name="seed"
|
||||||
help={t("modelSettings.form.seed.help")}
|
help={t("modelSettings.form.seed.help")}
|
||||||
@ -122,6 +133,7 @@ export const CurrentChatModelSettings = ({
|
|||||||
placeholder={t("modelSettings.form.seed.placeholder")}
|
placeholder={t("modelSettings.form.seed.placeholder")}
|
||||||
/>
|
/>
|
||||||
</Form.Item>
|
</Form.Item>
|
||||||
|
|
||||||
<Form.Item
|
<Form.Item
|
||||||
name="numCtx"
|
name="numCtx"
|
||||||
label={t("modelSettings.form.numCtx.label")}>
|
label={t("modelSettings.form.numCtx.label")}>
|
||||||
@ -140,6 +152,8 @@ export const CurrentChatModelSettings = ({
|
|||||||
/>
|
/>
|
||||||
</Form.Item>
|
</Form.Item>
|
||||||
|
|
||||||
|
<Divider />
|
||||||
|
|
||||||
<Collapse
|
<Collapse
|
||||||
ghost
|
ghost
|
||||||
className="border-none bg-transparent"
|
className="border-none bg-transparent"
|
||||||
@ -188,12 +202,12 @@ export const CurrentChatModelSettings = ({
|
|||||||
}
|
}
|
||||||
]}
|
]}
|
||||||
/>
|
/>
|
||||||
|
<Button
|
||||||
<button
|
type="primary"
|
||||||
type="submit"
|
htmlType="submit"
|
||||||
className="inline-flex justify-center w-full text-center mt-3 items-center rounded-md border border-transparent bg-black px-2 py-2 text-sm font-medium leading-4 text-white shadow-sm hover:bg-gray-700 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2 dark:bg-white dark:text-gray-800 dark:hover:bg-gray-100 dark:focus:ring-gray-500 dark:focus:ring-offset-gray-100 disabled:opacity-50 ">
|
className="inline-flex justify-center w-full text-center mt-3 items-center rounded-md border border-transparent bg-black px-2 py-2 text-sm font-medium leading-4 text-white shadow-sm hover:bg-gray-700 focus:outline-none focus:ring-2 focus:ring-indigo-500 focus:ring-offset-2 dark:bg-white dark:text-gray-800 dark:hover:bg-gray-100 dark:focus:ring-gray-500 dark:focus:ring-offset-gray-100 disabled:opacity-50 ">
|
||||||
{t("save")}
|
{t("save")}
|
||||||
</button>
|
</Button>
|
||||||
</Form>
|
</Form>
|
||||||
) : (
|
) : (
|
||||||
<Skeleton active />
|
<Skeleton active />
|
||||||
@ -209,7 +223,8 @@ export const CurrentChatModelSettings = ({
|
|||||||
open={open}
|
open={open}
|
||||||
onClose={() => setOpen(false)}
|
onClose={() => setOpen(false)}
|
||||||
width={500}
|
width={500}
|
||||||
title={t("currentChatModelSettings")}>
|
title={t("currentChatModelSettings")}
|
||||||
|
>
|
||||||
{renderBody()}
|
{renderBody()}
|
||||||
</Drawer>
|
</Drawer>
|
||||||
)
|
)
|
||||||
@ -221,7 +236,8 @@ export const CurrentChatModelSettings = ({
|
|||||||
open={open}
|
open={open}
|
||||||
onOk={() => setOpen(false)}
|
onOk={() => setOpen(false)}
|
||||||
onCancel={() => setOpen(false)}
|
onCancel={() => setOpen(false)}
|
||||||
footer={null}>
|
footer={null}
|
||||||
|
>
|
||||||
{renderBody()}
|
{renderBody()}
|
||||||
</Modal>
|
</Modal>
|
||||||
)
|
)
|
||||||
|
Loading…
x
Reference in New Issue
Block a user