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

Basic toolset for tune

module.exports = async function tail(node, args, context) {
  if (!node || (node.type !== 'text' && node.type !== 'llm')) {
    return node;
  }

  const n = parseInt(args, 10);
  const count = (isNaN(n) || n <= 0) ? 20 : n;


  if (node.type === 'text') {
    return {
      ...node,
      read: async () => {
        const text = await node.read();
        return text.split(/\r?\n/).slice(-count -1).join('\n');
      }
    }
  }

  // llm 
  return {
    ...node,
    exec: async (payload, ctx) => {
      let startIndex = Math.max(0, payload.messages.length - count)
      startIndex = payload.messages.findIndex((msg, index) => (index >= startIndex) && (msg.role === 'user' || msg.role ==='assistant'))
      const messages = payload.messages.filter(
        (message, index) => (message.role === "system") || (index >= startIndex))
    
      return node.exec({
        ...payload,
        messages
      }, ctx)
    }
  }
};