fix: default model in embedding page

main
jialin 12 months ago
parent fd7ce9027a
commit 3e691aa31f

@ -111,11 +111,17 @@ const GroundEmbedding: React.FC<MessageProps> = forwardRef((props, ref) => {
paramsRef,
modelMeta,
formFields
} = useInitLLmMeta(props, {
defaultValues: {},
defaultParamsConfig: [],
metaKeys: LLM_METAKEYS
});
} = useInitLLmMeta(
{
modelList,
isChat: true
},
{
defaultValues: {},
defaultParamsConfig: [],
metaKeys: LLM_METAKEYS
}
);
useImperativeHandle(ref, () => {
return {
@ -429,11 +435,17 @@ const GroundEmbedding: React.FC<MessageProps> = forwardRef((props, ref) => {
];
}, [outputHeight, scatterData, embeddingData]);
useEffect(() => {
setMessageId();
setScatterData([]);
setTokenResult(null);
}, [parameters.model]);
const onValuesChange = useCallback(
(changeValues: Record<string, any>, allValues: Record<string, any>) => {
if (changeValues.model) {
setScatterData([]);
setTokenResult(null);
} else {
handleOnValuesChange(changeValues, allValues);
}
},
[handleOnValuesChange]
);
useHotkeys(
HotKeys.SUBMIT,
@ -709,7 +721,7 @@ const GroundEmbedding: React.FC<MessageProps> = forwardRef((props, ref) => {
<div className="box">
<DynamicParams
ref={formRef}
onValuesChange={handleOnValuesChange}
onValuesChange={onValuesChange}
paramsConfig={paramsConfig}
initialValues={initialValues}
modelList={modelList}

@ -129,11 +129,17 @@ const GroundReranker: React.FC<MessageProps> = forwardRef((props, ref) => {
paramsRef,
modelMeta,
formFields
} = useInitLLmMeta(props, {
defaultValues: { top_n: 3 },
defaultParamsConfig: fieldConfig,
metaKeys: LLM_METAKEYS
});
} = useInitLLmMeta(
{
modelList,
isChat: true
},
{
defaultValues: { top_n: 3 },
defaultParamsConfig: fieldConfig,
metaKeys: LLM_METAKEYS
}
);
useImperativeHandle(ref, () => {
return {
@ -425,6 +431,17 @@ const GroundReranker: React.FC<MessageProps> = forwardRef((props, ref) => {
setIsEmptyText(false);
};
const onValuesChange = useCallback(
(changedValues: any, allValues: any) => {
if (changedValues.model) {
setTokenResult(null);
} else {
handleOnValuesChange(changedValues, allValues);
}
},
[handleOnValuesChange]
);
useHotkeys(
HotKeys.SUBMIT,
(e: any) => {
@ -437,11 +454,6 @@ const GroundReranker: React.FC<MessageProps> = forwardRef((props, ref) => {
}
);
useEffect(() => {
setMessageId();
setTokenResult(null);
}, [parameters.model]);
useEffect(() => {
if (scroller.current) {
initialize(scroller.current);
@ -591,7 +603,7 @@ const GroundReranker: React.FC<MessageProps> = forwardRef((props, ref) => {
<div className="box">
<DynamicParams
ref={formRef}
onValuesChange={handleOnValuesChange}
onValuesChange={onValuesChange}
paramsConfig={paramsConfig}
initialValues={initialValues}
modelList={modelList}

@ -109,7 +109,7 @@ export const useInitLLmMeta = (
const handleOnModelChange = useCallback(
(val: string) => {
if (!val) return;
if (!val || val === parameters.model) return;
const model = modelList.find((item) => item.value === val);
const { form: initialData, meta } = extractLLMMeta(model?.meta);
@ -135,7 +135,7 @@ export const useInitLLmMeta = (
});
setParamsConfig(config);
},
[modelList, defaultParamsConfig]
[modelList, parameters, defaultParamsConfig]
);
const handleOnValuesChange = useCallback(
@ -143,8 +143,10 @@ export const useInitLLmMeta = (
if (changeValues.model) {
handleOnModelChange(changeValues.model);
return;
} else {
setParams(allValues);
setInitialValues(allValues);
}
setParams(allValues);
},
[handleOnModelChange]
);

Loading…
Cancel
Save