adai/MemoryBear
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #768 from SuanmoSuanyangTechnology/hotfix/v0.2.9

Browse Source 
fix(web): knowledge base model api params
This commit is contained in:
Ke Sun
2026-04-02 14:39:43 +08:00
committed by GitHub
parent 18703919a8 9535545947
commit 9ff3a3d5f7
4 changed files with 31 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
web/src/api/knowledgeBase.ts
View File

@@ -68,7 +68,7 @@ export const getModelTypeList = async () => {
return response as any[]; return response as any[];
}; };
// 获取模型列表 // 获取模型列表
export const getModelList = async (types: string[], pageInfo: PageRequest) => { export const getModelList = async (pageInfo: PageRequest, types?: string[]) => {
const response = await request.get(`${apiPrefix}/models`, { ...pageInfo, type: types?.join(','), is_active: true }); const response = await request.get(`${apiPrefix}/models`, { ...pageInfo, type: types?.join(','), is_active: true });
return response as any; return response as any;
}; };

2
web/src/views/KnowledgeBase/components/CreateModal.tsx
View File

@@ -162,7 +162,7 @@ const CreateModal = forwardRef<CreateModalRef, CreateModalRefProps>(({
// If model data hasn't been fetched yet, fetch it once // If model data hasn't been fetched yet, fetch it once
if (!models) { if (!models) {
try { try {
models = await getModelList({ page: 1, pagesize: 100 }); models = await getModelList({ page: 1, pagesize: 100 }, ['llm', 'embedding', 'rerank', 'chat']);
} catch (error) { } catch (error) {
console.error('Failed to fetch models:', error); console.error('Failed to fetch models:', error);
models = { items: [] }; models = { items: [] };

2
web/src/views/KnowledgeBase/index.tsx
View File

@@ -207,7 +207,7 @@ const KnowledgeBaseManagement: FC = () => {
}; };
const fetchModelList = async () => { const fetchModelList = async () => {
try { try {
const response = await getModelList(['llm', 'embedding', 'rerank', 'chat'], { page: 1, pagesize: 100 }); const response = await getModelList({ page: 1, pagesize: 100 }, ['llm', 'embedding', 'rerank', 'chat']);
// 缓存模型列表，建立 id -> name 的映射 // 缓存模型列表，建立 id -> name 的映射
if (response?.items && Array.isArray(response.items)) { if (response?.items && Array.isArray(response.items)) {
const cache: Record<string, string> = {}; const cache: Record<string, string> = {};

52
web/src/views/Workflow/hooks/useWorkflowGraph.ts
View File

@@ -490,32 +490,36 @@ export const useWorkflowGraph = ({
* @param node - Clicked node * @param node - Clicked node
*/ */
const nodeClick = ({ node }: { node: Node }) => { const nodeClick = ({ node }: { node: Node }) => {
// Ignore add-node type node clicks blankClick()
const nodeData = node.getData()
if (nodeData?.type === 'add-node' || nodeData.type === 'break' || nodeData.type === 'cycle-start') {
setSelectedNode(null)
return;
}
const nodes = graphRef.current?.getNodes(); setTimeout(() => {
// Ignore add-node type node clicks
nodes?.forEach(vo => { const nodeData = node.getData()
const data = vo.getData(); if (nodeData?.type === 'add-node' || nodeData.type === 'break' || nodeData.type === 'cycle-start') {
if (data.isSelected) { setSelectedNode(null)
vo.setData({ return;
...data,
isSelected: false,
});
} }
});
node.setData({ const nodes = graphRef.current?.getNodes();
...nodeData,
isSelected: true, nodes?.forEach(vo => {
}); const data = vo.getData();
clearEdgeSelect() if (data.isSelected) {
if (nodeData.type !== 'notes') { vo.setData({
setSelectedNode(node); ...data,
} isSelected: false,
});
}
});
node.setData({
...nodeData,
isSelected: true,
});
clearEdgeSelect()
if (nodeData.type !== 'notes') {
setSelectedNode(node);
}
}, 0)
}; };
/** /**
* Handle edge click event * Handle edge click event