fix: default model in embedding page

This commit is contained in:
jialin
2025-03-05 14:05:12 +08:00
parent fd7ce9027a
commit 3e691aa31f
3 changed files with 51 additions and 25 deletions
@@ -111,11 +111,17 @@ const GroundEmbedding: React.FC<MessageProps> = forwardRef((props, ref) => {
paramsRef, paramsRef,
modelMeta, modelMeta,
formFields formFields
} = useInitLLmMeta(props, { } = useInitLLmMeta(
defaultValues: {}, {
defaultParamsConfig: [], modelList,
metaKeys: LLM_METAKEYS isChat: true
}); },
{
defaultValues: {},
defaultParamsConfig: [],
metaKeys: LLM_METAKEYS
}
);
useImperativeHandle(ref, () => { useImperativeHandle(ref, () => {
return { return {
@@ -429,11 +435,17 @@ const GroundEmbedding: React.FC<MessageProps> = forwardRef((props, ref) => {
]; ];
}, [outputHeight, scatterData, embeddingData]); }, [outputHeight, scatterData, embeddingData]);
useEffect(() => { const onValuesChange = useCallback(
setMessageId(); (changeValues: Record<string, any>, allValues: Record<string, any>) => {
setScatterData([]); if (changeValues.model) {
setTokenResult(null); setScatterData([]);
}, [parameters.model]); setTokenResult(null);
} else {
handleOnValuesChange(changeValues, allValues);
}
},
[handleOnValuesChange]
);
useHotkeys( useHotkeys(
HotKeys.SUBMIT, HotKeys.SUBMIT,
@@ -709,7 +721,7 @@ const GroundEmbedding: React.FC<MessageProps> = forwardRef((props, ref) => {
<div className="box"> <div className="box">
<DynamicParams <DynamicParams
ref={formRef} ref={formRef}
onValuesChange={handleOnValuesChange} onValuesChange={onValuesChange}
paramsConfig={paramsConfig} paramsConfig={paramsConfig}
initialValues={initialValues} initialValues={initialValues}
modelList={modelList} modelList={modelList}
@@ -129,11 +129,17 @@ const GroundReranker: React.FC<MessageProps> = forwardRef((props, ref) => {
paramsRef, paramsRef,
modelMeta, modelMeta,
formFields formFields
} = useInitLLmMeta(props, { } = useInitLLmMeta(
defaultValues: { top_n: 3 }, {
defaultParamsConfig: fieldConfig, modelList,
metaKeys: LLM_METAKEYS isChat: true
}); },
{
defaultValues: { top_n: 3 },
defaultParamsConfig: fieldConfig,
metaKeys: LLM_METAKEYS
}
);
useImperativeHandle(ref, () => { useImperativeHandle(ref, () => {
return { return {
@@ -425,6 +431,17 @@ const GroundReranker: React.FC<MessageProps> = forwardRef((props, ref) => {
setIsEmptyText(false); setIsEmptyText(false);
}; };
const onValuesChange = useCallback(
(changedValues: any, allValues: any) => {
if (changedValues.model) {
setTokenResult(null);
} else {
handleOnValuesChange(changedValues, allValues);
}
},
[handleOnValuesChange]
);
useHotkeys( useHotkeys(
HotKeys.SUBMIT, HotKeys.SUBMIT,
(e: any) => { (e: any) => {
@@ -437,11 +454,6 @@ const GroundReranker: React.FC<MessageProps> = forwardRef((props, ref) => {
} }
); );
useEffect(() => {
setMessageId();
setTokenResult(null);
}, [parameters.model]);
useEffect(() => { useEffect(() => {
if (scroller.current) { if (scroller.current) {
initialize(scroller.current); initialize(scroller.current);
@@ -591,7 +603,7 @@ const GroundReranker: React.FC<MessageProps> = forwardRef((props, ref) => {
<div className="box"> <div className="box">
<DynamicParams <DynamicParams
ref={formRef} ref={formRef}
onValuesChange={handleOnValuesChange} onValuesChange={onValuesChange}
paramsConfig={paramsConfig} paramsConfig={paramsConfig}
initialValues={initialValues} initialValues={initialValues}
modelList={modelList} modelList={modelList}
+5 -3
View File
@@ -109,7 +109,7 @@ export const useInitLLmMeta = (
const handleOnModelChange = useCallback( const handleOnModelChange = useCallback(
(val: string) => { (val: string) => {
if (!val) return; if (!val || val === parameters.model) return;
const model = modelList.find((item) => item.value === val); const model = modelList.find((item) => item.value === val);
const { form: initialData, meta } = extractLLMMeta(model?.meta); const { form: initialData, meta } = extractLLMMeta(model?.meta);
@@ -135,7 +135,7 @@ export const useInitLLmMeta = (
}); });
setParamsConfig(config); setParamsConfig(config);
}, },
[modelList, defaultParamsConfig] [modelList, parameters, defaultParamsConfig]
); );
const handleOnValuesChange = useCallback( const handleOnValuesChange = useCallback(
@@ -143,8 +143,10 @@ export const useInitLLmMeta = (
if (changeValues.model) { if (changeValues.model) {
handleOnModelChange(changeValues.model); handleOnModelChange(changeValues.model);
return; return;
} else {
setParams(allValues);
setInitialValues(allValues);
} }
setParams(allValues);
}, },
[handleOnModelChange] [handleOnModelChange]
); );