tune-basic-toolset/src/curry.proc.js

Basic toolset for tune

const { parseArgs } = require('./utils.js')

module.exports = async function curry(node, args, ctx) {
  if (!node || node.type !== 'tool' ) {
    throw Error('curry can only modify tool') 
  }

  // parse args string to object
  // supports key=value pairs where value can be quoted string
  // keys can optionally start with $
  const parsedArgs = parseArgs(args.trim())
  const schema = { ...node.schema }
  const curryParams = {} 
  const newNode = { 
    ...node,
    schema,
    exec: async(params, ctx) => node.exec({...params, ...curryParams}, ctx)
  }
  for(const key in parsedArgs) {
    if (key.indexOf("$") === -1)  {
      curryParams[key] = parsedArgs[key]
      if (schema.parameters.properties[key]) {
        // throw Error(`parameter ${key} is not defined in ${node.name}'s schema`)
        delete schema.parameters.properties[key]
      }
      const idx = (schema.parameters.required || []).indexOf(key)
      if (idx !== -1) {
        schema.parameters.required.splice(idx, 1)
      }
    } else {
      const path = key.slice(1).split(".")
      if (path[0] !== "name" && path[0] !== "description") {
        throw Error(`path should start with $name or with $description`)
      }
      newNode[path[0]] = parsedArgs[key]
    }
  }

  return newNode 
}