fix: open model in playground

This commit is contained in:
jialin
2026-02-03 18:21:08 +08:00
parent 134827a0f7
commit 46969702a8
6 changed files with 51 additions and 41 deletions
@@ -0,0 +1,32 @@
import { useNavigate } from '@umijs/max';
import { modelCategoriesMap } from '../../llmodels/config';
import { categoryToPathMap } from '../../llmodels/config/button-actions';
const useOpenPlayground = () => {
const navigate = useNavigate();
const handleOpenPlayGround = (row: any) => {
for (const [category, path] of Object.entries(categoryToPathMap)) {
if (
row.categories?.includes(category) &&
[
modelCategoriesMap.text_to_speech,
modelCategoriesMap.speech_to_text
].includes(category)
) {
navigate(`${path}&model=${row.name}`);
return;
}
if (row.categories?.includes(category)) {
navigate(`${path}?model=${row.name}`);
return;
}
}
navigate(`/playground/chat?model=${row.name}`);
};
return {
handleOpenPlayGround
};
};
export default useOpenPlayground;