fix: default model in embedding page
This commit is contained in:
@@ -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]
|
||||
);
|
||||
|
||||
Reference in New Issue
Block a user