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

feat: workflow add knowledge-retrieval node

Browse Source
This commit is contained in:
zhaoying
2025-12-29 14:39:57 +08:00
parent 06e5f4f8ff
commit 0fce86f76b
12 changed files with 899 additions and 27 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -32,7 +32,7 @@ export interface UseWorkflowGraphReturn {
deleteEvent: () => boolean | void;
copyEvent: () => boolean | void;
parseEvent: () => boolean | void;
handleSave: (flag?: boolean) => Promise<any>;
handleSave: (flag?: boolean) => Promise<unknown>;
}
const edge_color = '#155EEF';
@@ -83,7 +83,13 @@ export const useWorkflowGraph = ({
if (nodeLibraryConfig?.config) {
Object.keys(nodeLibraryConfig.config).forEach(key => {
if (nodeLibraryConfig.config && nodeLibraryConfig.config[key] && config[key]) {
if (key === 'knowledge_retrieval' && nodeLibraryConfig.config && nodeLibraryConfig.config[key]) {
const { query, ...rest } = config
nodeLibraryConfig.config[key].defaultValue = {
...rest
}
console.log(type, config, nodeLibraryConfig)
} else if (nodeLibraryConfig.config && nodeLibraryConfig.config[key] && config[key]) {
nodeLibraryConfig.config[key].defaultValue = config[key]
}
})
@@ -686,12 +692,17 @@ export const useWorkflowGraph = ({
nodes: nodes.map((node: Node) => {
const data = node.getData();
const position = node.getPosition();
const config: Record<string, any> = {}
let itemConfig: Record<string, any> = {}
if (data.config) {
Object.keys(data.config).forEach(key => {
if (data.config[key] && 'defaultValue' in data.config[key]) {
config[key] = data.config[key].defaultValue
if (data.config[key] && 'defaultValue' in data.config[key] && key !== 'knowledge_retrieval') {
itemConfig[key] = data.config[key].defaultValue
} else if (key === 'knowledge_retrieval' && data.config[key] && 'defaultValue' in data.config[key]) {
itemConfig = {
...itemConfig,
...data.config[key].defaultValue
}
}
})
}
@@ -704,7 +715,7 @@ export const useWorkflowGraph = ({
x: position.x,
y: position.y,
},
config: config
config: itemConfig
};
}),
edges: edges.map((edge: Edge) => {