variable-parser.js 3.3 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677
  1. /**
  2. * 统一解析结点入参、出参:将 action 中的变量引用用 variableContext 解析为实际值。
  3. * 规则:{var}、{{var}} 为变量(替换为变量值),"hello" 为字符串字面量,"hello{var}" 为字符串+变量拼接。
  4. */
  5. const setParser = require('./actions/set-parser.js')
  6. const resolveValue = setParser.resolveValue
  7. const replaceVariablesInString = setParser.replaceVariablesInString
  8. const extractVarName = setParser.extractVarName
  9. /** 视为入参的字段(会被解析);inVars 由各结点按需解析(因部分结点将 inVars 某项作为输出变量名)。 */
  10. const INPUT_KEYS = [
  11. 'value', 'target', 'template', 'area', 'savePath', 'condition', 'delay', 'interval',
  12. 'items', 'screenshot', 'region', 'method', 'clear', 'timeout', 'retry',
  13. 'min', 'max', 'avatar1', 'avatar2', 'friendAvatar', 'avatar', 'path', 'filePath',
  14. 'inputDataString', 'textVariable', 'senderVariable', 'appendMode',
  15. 'summaryPrompt', 'historyPrompt', 'model', 'prompt', 'systemPrompt',
  16. 'regionArea', 'saveDir', 'url', 'filename', 'imageUrl',
  17. ]
  18. /**
  19. * 解析单值:先对字符串做 {{var}} 替换,再对整体做 {var} 引用解析
  20. */
  21. function resolveInputValue(val, variableContext) {
  22. if (variableContext == null) return val
  23. if (typeof val === 'string') {
  24. const replaced = replaceVariablesInString(val, variableContext)
  25. let result = resolveValue(replaced, variableContext)
  26. if (result === val && /^[\w-]+$/.test(val) && variableContext[val] !== undefined) result = variableContext[val]
  27. return result
  28. }
  29. if (Array.isArray(val)) return val.map(item => resolveInputValue(item, variableContext))
  30. if (typeof val === 'object' && val !== null) {
  31. const out = {}
  32. for (const k in val) out[k] = resolveInputValue(val[k], variableContext)
  33. return out
  34. }
  35. return val
  36. }
  37. /**
  38. * 解析整条 action 的入参,返回新对象(不修改原 action)
  39. * @param {object} action - 原始或已 parse 的 action
  40. * @param {object} variableContext - 变量表
  41. * @returns {object} 入参解析后的 action 副本
  42. */
  43. function resolveActionInputs(action, variableContext) {
  44. if (!action || typeof action !== 'object') return action
  45. if (!variableContext || typeof variableContext !== 'object') return Object.assign({}, action)
  46. const resolved = Object.assign({}, action)
  47. for (const key of INPUT_KEYS) {
  48. if (key in resolved && resolved[key] !== undefined && resolved[key] !== null) {
  49. resolved[key] = resolveInputValue(resolved[key], variableContext)
  50. }
  51. }
  52. if (resolved.inVars && Array.isArray(resolved.inVars)) {
  53. resolved.inVars = resolved.inVars.map((v) => resolveInputValue(v, variableContext))
  54. }
  55. if (resolved.outVars && Array.isArray(resolved.outVars)) {
  56. resolved.outVars = resolved.outVars.map((v) => (typeof v === 'string' ? extractVarName(v) : v))
  57. }
  58. if (resolved.condition && typeof resolved.condition === 'object' && !Array.isArray(resolved.condition)) {
  59. const c = resolved.condition
  60. if (c.interval != null) resolved.condition = Object.assign({}, c, { interval: resolveInputValue(c.interval, variableContext) })
  61. if (c.repeat != null) resolved.condition = Object.assign({}, resolved.condition, { repeat: resolveInputValue(c.repeat, variableContext) })
  62. }
  63. return resolved
  64. }
  65. module.exports = {
  66. resolveActionInputs,
  67. resolveInputValue,
  68. }