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