feat: Add save functionality for system prompt and model settings

This commit is contained in:
n4ze3m 2024-12-01 15:52:52 +05:30
parent 77f0cdbb35
commit 5687517238

View File

@ -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>
) )