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

feat: llm node add context

Browse Source
This commit is contained in:
zhaoying
2026-01-04 12:06:24 +08:00
parent 02c8fd0e3f
commit 351be8aaf3
10 changed files with 233 additions and 150 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
web/src/i18n/en.ts
View File

@@ -1589,6 +1589,7 @@ Memory Bear: After the rebellion, regional warlordism intensified for several re
model_id: 'Model',
temperature: 'Temperature',
max_tokens: 'Max Tokens',
context: 'Context',
},
start: {
variables: 'Input Fields',

1
web/src/i18n/zh.ts
View File

@@ -1690,6 +1690,7 @@ export const zh = {
model_id: '模型',
temperature: '温度',
max_tokens: '最大令牌数',
context: '上下文',
},
start: {
variables: '输入字段',

22
web/src/views/Workflow/components/Editor/nodes/VariableNode.tsx
View File

@@ -42,13 +42,21 @@ const VariableComponent: React.FC<{ nodeKey: NodeKey; data: Suggestion }> = ({
})}
contentEditable={false}
>
<img
src={data.nodeData?.icon}
style={{ width: '12px', height: '12px', marginRight: '4px' }}
alt=""
/>
<span className="rb:wrap-break-word rb:line-clamp-1">{data.nodeData?.name}</span>
<span style={{ color: '#DFE4ED', margin: '0 2px' }}>/</span>
{data.isContext ? (
<span style={{ fontSize: '12px', marginRight: '4px' }}>📄</span>
) : (
<img
src={data.nodeData?.icon}
style={{ width: '12px', height: '12px', marginRight: '4px' }}
alt=""
/>
)}
{!data.isContext && (
<>
<span className="rb:wrap-break-word rb:line-clamp-1">{data.nodeData?.name}</span>
<span style={{ color: '#DFE4ED', margin: '0 2px' }}>/</span>
</>
)}
<span className="rb:text-ellipsis rb:overflow-hidden rb:whitespace-nowrap rb:flex-1" style={{ color: '#155EEF' }}>{data.label}</span>
</span>
);

13
web/src/views/Workflow/components/Editor/plugin/AutocompletePlugin.tsx
View File

@@ -11,7 +11,9 @@ export interface Suggestion {
type: string;
dataType: string;
value: string;
nodeData: NodeProperties
nodeData: NodeProperties;
isContext?: boolean; // 标记是否为context变量
disabled?: boolean; // 标记是否禁用
}
const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
@@ -131,19 +133,20 @@ const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
key={option.key}
style={{
padding: '8px 12px',
cursor: 'pointer',
cursor: option.disabled ? 'not-allowed' : 'pointer',
background: selectedIndex === globalIndex ? '#f0f8ff' : 'white',
display: 'flex',
alignItems: 'center',
justifyContent: 'space-between',
opacity: option.disabled ? 0.5 : 1,
}}
onClick={() => insertMention(option)}
onClick={() => !option.disabled && insertMention(option)}
onMouseEnter={() => setSelectedIndex(globalIndex)}
>
<div style={{ display: 'flex', alignItems: 'center', gap: '8px' }}>
<span
style={{
background: option.type === 'context' ? '#722ed1' :
background: option.isContext ? '#722ed1' :
option.type === 'system' ? '#1890ff' : '#52c41a',
color: 'white',
padding: '2px 6px',
@@ -153,7 +156,7 @@ const AutocompletePlugin: FC<{ options: Suggestion[] }> = ({ options }) => {
textAlign: 'center',
}}
>
{option.type === 'context' ? '📄' :
{option.isContext ? '📄' :
option.type === 'system' ? 'x' : 'x'}
</span>
<span style={{ fontSize: '14px' }}>{option.label}</span>

13
web/src/views/Workflow/components/Editor/plugin/InitialValuePlugin.tsx
View File

@@ -25,7 +25,20 @@ const InitialValuePlugin: React.FC<InitialValuePluginProps> = ({ value, options
parts.forEach(part => {
const match = part.match(/^\{\{([^.]+)\.([^}]+)\}\}$/);
const contextMatch = part.match(/^\{\{context\}\}$/);
// 匹配{{context}}格式
if (contextMatch) {
const contextSuggestion = options.find(s => s.isContext && s.label === 'context');
if (contextSuggestion) {
paragraph.append($createVariableNode(contextSuggestion));
} else {
paragraph.append($createTextNode(part));
}
return
}
// 匹配普通变量{{nodeId.label}}格式
if (match) {
const [_, nodeId, label] = match;

6
web/src/views/Workflow/components/Properties/MappingList/index.tsx
View File

@@ -1,6 +1,6 @@
import React from 'react';
import { useTranslation } from 'react-i18next'
import { MinusCircleOutlined, PlusOutlined } from '@ant-design/icons';
import { MinusCircleOutlined } from '@ant-design/icons';
import { Button, Form, Input, Space } from 'antd';
interface MappingListProps {
@@ -33,8 +33,8 @@ const MappingList: React.FC<MappingListProps> = ({ name }) => {
</Space>
))}
<Form.Item>
<Button type="dashed" onClick={() => add()} block icon={<PlusOutlined />}>
Add field
<Button type="dashed" onClick={() => add()} block>
+ {t('common.add')}
</Button>
</Form.Item>
</>

24
web/src/views/Workflow/components/Properties/MessageEditor.tsx
View File

@@ -1,4 +1,4 @@
import { type FC } from 'react';
import { type FC, useMemo } from 'react';
import { useTranslation } from 'react-i18next'
import { Input, Form, Space, Button, Row, Col, Select, type FormListOperation } from 'antd';
import { MinusCircleOutlined } from '@ant-design/icons';
@@ -31,6 +31,24 @@ const MessageEditor: FC<TextareaProps> = ({
const form = Form.useFormInstance();
const values = form.getFieldsValue()
// 检查是否已经使用了context变量，将已使用的context设置为disabled
const processedOptions = useMemo(() => {
if (!isArray || !values[parentName]) return options;
// 获取所有消息内容
const allContents = values[parentName]
.map((msg: any) => msg.content || '')
.join(' ');
// 将已使用的context变量标记为disabled
return options.map(opt => {
if (opt.isContext && allContents.includes(opt.value)) {
return { ...opt, disabled: true };
}
return opt;
});
}, [options, values, parentName, isArray]);
const handleAdd = (add: FormListOperation['add']) => {
const list = values[parentName];
const lastRole = list[list.length - 1].role
@@ -80,7 +98,7 @@ const MessageEditor: FC<TextareaProps> = ({
name={[name, 'content']}
noStyle
>
<Editor placeholder={placeholder} options={options} />
<Editor placeholder={placeholder} options={processedOptions} />
</Form.Item>
</Space>
)
@@ -104,7 +122,7 @@ const MessageEditor: FC<TextareaProps> = ({
name={parentName}
noStyle
>
<Editor placeholder={placeholder} options={options} />
<Editor placeholder={placeholder} options={processedOptions} />
</Form.Item>
</Space>
}

23
web/src/views/Workflow/components/Properties/index.tsx
View File

@@ -330,9 +330,30 @@ const Properties: FC<PropertiesProps> = ({
}
if (selectedNode?.data?.type === 'llm' && key === 'messages' && config.type === 'define') {
// 为llm节点且isArray=true时添加context变量支持
let contextVariableList = [...variableList];
const isArrayMode = config.isArray !== false; // 默认为true
if (isArrayMode) {
const contextKey = `${selectedNode.id}_context`;
const hasContextVariable = contextVariableList.some(v => v.key === contextKey);
if (!hasContextVariable) {
contextVariableList.unshift({
key: contextKey,
label: 'context',
type: 'variable',
dataType: 'String',
value: `{{context}}`,
nodeData: selectedNode.getData(),
isContext: true,
});
}
}
return (
<Form.Item key={key} name={key}>
<MessageEditor options={variableList} parentName={key} />
<MessageEditor options={contextVariableList} parentName={key} />
</Form.Item>
)
}

263
web/src/views/Workflow/constant.ts
View File

@@ -117,6 +117,9 @@ export const nodeLibrary: NodeLibrary[] = [
step: 1,
defaultValue: 2000
},
context: {
type: 'variableList',
},
messages: {
type: 'define',
defaultValue: [
@@ -142,27 +145,27 @@ export const nodeLibrary: NodeLibrary[] = [
}
},
// { type: "classification", icon: classificationIcon },
// { type: "parameter-extractor", icon: parameterExtractionIcon,
// config: {
// model_id: {
// type: 'customSelect',
// url: getModelListUrl,
// params: { type: 'llm,chat' }, // llm/chat
// valueKey: 'id',
// labelKey: 'name',
// },
// text: {
// type: 'variableList',
// },
// params: {
// type: 'paramList',
// },
// prompt: {
// type: 'messageEditor',
// isArray: false,
// },
// }
// }
{ type: "parameter-extractor", icon: parameterExtractionIcon,
config: {
model_id: {
type: 'customSelect',
url: getModelListUrl,
params: { type: 'llm,chat' }, // llm/chat
valueKey: 'id',
labelKey: 'name',
},
text: {
type: 'variableList',
},
params: {
type: 'paramList',
},
prompt: {
type: 'messageEditor',
isArray: false,
},
}
}
]
},
// {
@@ -182,115 +185,115 @@ export const nodeLibrary: NodeLibrary[] = [
// { type: "agent_arbitration", icon: agentArbitrationIcon }
// ]
// },
// {
// category: "flowControl",
// nodes: [
// { type: "if-else", icon: conditionIcon,
// config: {
// cases: {
// type: 'caseList',
// defaultValue: [
// {
// logical_operator: 'and',
// expressions: []
// }
// ]
// }
// }
// },
// // { type: "iteration", icon: iterationIcon },
// { type: "loop", icon: loopIcon },
// // { type: "parallel", icon: parallelIcon },
// { type: "var-aggregator", icon: aggregatorIcon,
// config: {
// group: {
// type: 'switch',
// defaultValue: false
// },
// group_names: {
// type: 'groupVariableList',
// defaultValue: [{ key: 'Group1', value: []}]
// }
// }
// }
// ]
// },
// {
// category: "externalInteraction",
// nodes: [
// { type: "http-request", icon: httpRequestIcon,
// config: {
// method: {
// type: 'select',
// options: [
// { label: 'GET', value: 'GET' },
// { label: 'POST', value: 'POST' },
// { label: 'HEAD', value: 'HEAD' },
// { label: 'PATCH', value: 'PATCH' },
// { label: 'PUT', value: 'PUT' },
// { label: 'DELETE', value: 'DELETE' },
// ],
// defaultValue: 'GET'
// },
// url: {
// type: 'messageEditor',
// isArray: false,
// },
// auth: {
// type: 'define',
// defaultValue: {
// auth_type: 'none'
// }
// },
// headers: {
// type: 'define',
// defaultValue: {}
// },
// params: {
// type: 'define',
// defaultValue: {}
// },
// body: {
// type: 'define',
// defaultValue: {
// 'content_type': 'none'
// }
// },
// verify_ssl: {
// type: 'switch',
// defaultValue: false
// },
// timeouts: {
// type: 'define',
// defaultValue: {}
// },
// retry: {
// type: 'define',
// },
// error_handle: {
// type: 'define',
// defaultValue: {
// method: 'default'
// }
// }
// }
// },
// // { type: "tools", icon: toolsIcon },
// // { type: "code_execution", icon: codeExecutionIcon },
// { type: "jinja-render", icon: templateRenderingIcon,
// config: {
// mapping: {
// type: 'mappingList',
// defaultValue: []
// },
// template: {
// type: 'messageEditor',
// isArray: false,
// },
// }
// }
// ]
// },
{
category: "flowControl",
nodes: [
{ type: "if-else", icon: conditionIcon,
config: {
cases: {
type: 'caseList',
defaultValue: [
{
logical_operator: 'and',
expressions: []
}
]
}
}
},
// { type: "iteration", icon: iterationIcon },
// { type: "loop", icon: loopIcon },
// { type: "parallel", icon: parallelIcon },
{ type: "var-aggregator", icon: aggregatorIcon,
config: {
group: {
type: 'switch',
defaultValue: false
},
group_names: {
type: 'groupVariableList',
defaultValue: [{ key: 'Group1', value: []}]
}
}
}
]
},
{
category: "externalInteraction",
nodes: [
{ type: "http-request", icon: httpRequestIcon,
config: {
method: {
type: 'select',
options: [
{ label: 'GET', value: 'GET' },
{ label: 'POST', value: 'POST' },
{ label: 'HEAD', value: 'HEAD' },
{ label: 'PATCH', value: 'PATCH' },
{ label: 'PUT', value: 'PUT' },
{ label: 'DELETE', value: 'DELETE' },
],
defaultValue: 'GET'
},
url: {
type: 'messageEditor',
isArray: false,
},
auth: {
type: 'define',
defaultValue: {
auth_type: 'none'
}
},
headers: {
type: 'define',
defaultValue: {}
},
params: {
type: 'define',
defaultValue: {}
},
body: {
type: 'define',
defaultValue: {
'content_type': 'none'
}
},
verify_ssl: {
type: 'switch',
defaultValue: false
},
timeouts: {
type: 'define',
defaultValue: {}
},
retry: {
type: 'define',
},
error_handle: {
type: 'define',
defaultValue: {
method: 'default'
}
}
}
},
// { type: "tools", icon: toolsIcon },
// { type: "code_execution", icon: codeExecutionIcon },
{ type: "jinja-render", icon: templateRenderingIcon,
config: {
mapping: {
type: 'mappingList',
defaultValue: []
},
template: {
type: 'messageEditor',
isArray: false,
},
}
}
]
},
// {
// category: "safetyAndCompliance",
// nodes: [

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

@@ -160,7 +160,12 @@ export const useWorkflowGraph = ({
graphRef.current?.addNodes(nodeList)
}
if (edges.length) {
const edgeList = edges.map(edge => {
// 去重处理：相同节点之间的连线仅连一次
const uniqueEdges = edges.filter((edge, index, arr) => {
return arr.findIndex(e => e.source === edge.source && e.target === edge.target) === index;
});
const edgeList = uniqueEdges.map(edge => {
const { source, target, label } = edge
const sourceCell = graphRef.current?.getCellById(source)
const targetCell = graphRef.current?.getCellById(target)
@@ -788,6 +793,11 @@ export const useWorkflowGraph = ({
const targetCell = graphRef.current?.getCellById(edge.getTargetCellId());
const sourcePortId = edge.getSourcePortId();
// 过滤无效连线：源节点或目标节点不存在
if (!sourceCell?.getData()?.id || !targetCell?.getData()?.id) {
return null;
}
// 如果是if-else节点的右侧端口连线，添加label
if (sourceCell?.getData()?.type === 'if-else' && sourcePortId?.startsWith('CASE')) {
return {
@@ -801,6 +811,11 @@ export const useWorkflowGraph = ({
source: sourceCell?.getData().id,
target: targetCell?.getData().id,
};
})
.filter(edge => edge !== null)
.filter((edge, index, arr) => {
// 去重：相同节点之间的连线仅保留一次
return arr.findIndex(e => e && e.source === edge?.source && e.target === edge?.target) === index;
}),
}
saveWorkflowConfig(config.app_id, params as WorkflowConfig)