root/gpustack-ui
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

fix: default model in embedding page

Browse Source
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
Download Patch File Download Diff File Expand all files Collapse all files
src/pages/playground/components/ground-embedding.tsx
+23 -11
View File
@@ -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}
src/pages/playground/components/ground-reranker.tsx
+23 -11
View File
@@ -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}