tune-basic-toolset 0.1.4 → 0.1.6

package/README.md

@@ -34,6 +34,7 @@ Basic toolset for [Tune](https://github.com/iovdin/tune).
   - [tail](#tail) take last N lines of a file or LLM payload
   - [slice](#slice) take lines from <start> to <finish> of a file
   - [random](#random) random selection, sampling, shuffling, uniform ranges
+  - [curry](#curry) change a tool by setting a parameter
 
 
 ## Setup
@@ -386,6 +387,10 @@ pipe filename content to shell command
 @{ a.log | shp tail }
 
 @{ a.log | shp grep pattern }
+
+print screen of one of tmux session
+@{| shp tmux capture-pane -t 0 -p }
+
 ```
 
 ### `init` 
@@ -575,10 +580,46 @@ user:
 @{| random uniform -2.5..7.5 }       # floats
 @{| random uniform 10 20 }           # two-number form
 
-comment:
+```
 Notes:
 - Quotes are respected for tokens with spaces.
 - Files referenced as @file are expanded to non-empty trimmed lines.
 - Integer ranges like a..b can be mixed with discrete values and files; float ranges cannot be mixed in lists.
 - sample and shuffle require a discrete set; float ranges are not supported there.
 - choices and sample output multiple lines (one item per line).
+
+
+### `curry`
+Modify a tool by setting parameter or name or description. 
+Narrow possible usage of a tool so that LLM wont mess up
+
+```chat
+user:
+@{ sh | curry text=ls $name=ls_cwd}
+
+what is in my current directory?
+
+assistant:
+
+tool_call: ls_cwd
+tool_result:
+node_modules
+package.json
+README.md
+src
+
+
+user:
+@{ sqlite | curry filename=db.sqlite format=table}
+
+what tables are outhere?
+
+user: 
+@{ list | curry filename=mylist.todo $name=todo }
+
+create sample todo list
+tool_call: todo
+[] - Create sample todo list
+tool_result:
+list updated
+```

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "tune-basic-toolset",
-  "version": "0.1.4",
+  "version": "0.1.6",
   "description": "Basic toolset for tune",
   "main": "src/index.js",
   "files": [

package/src/curry.proc.js

@@ -0,0 +1,56 @@
+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 = {};
+  // regex to split on spaces but keep quoted substrings together
+  const regex = /([^\s"']+|"[^"]*"|'[^']*')+/g;
+  const tokens = args.match(regex) || [];
+
+  tokens.forEach(token => {
+    const equalIndex = token.indexOf('=');
+    if (equalIndex > 0) {
+      const key = token.substring(0, equalIndex);
+      let value = token.substring(equalIndex + 1);
+
+      // strip double or single quotes from value if present
+      if ((value.startsWith('"') && value.endsWith('"')) || (value.startsWith("'") && value.endsWith("'"))) {
+        value = value.substring(1, value.length - 1);
+      }
+      parsedArgs[key] = value;
+    }
+  });
+  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 
+}

package/src/list.schema.json

@@ -6,7 +6,7 @@
     "properties": {
       "text": {
         "type": "string",
-        "description": "lines in format 'status - item', they'll be used to update list or to insert to the list, to delete item change its status to deleted or cancelled\nThe lines should contain only those items that needs to be changed - inserted/updated/deleted"
+        "description": "text where each line is in format 'status - item', they'll be used to update list or to insert to the list, to delete item change its status to deleted or cancelled\nThe lines should contain only those items that needs to be changed - inserted/updated/deleted"
       },
       "filename": {
         "type": "string",

package/src/resolve.proc.js

@@ -6,5 +6,11 @@ module.exports = async function resolve(node, args, ctx) {
     throw Error(`can resolve only text nodes, got ${node.type}`)
   }
   const filename = await node.read()
-  return ctx.resolve(filename.trim())
+  const result = await ctx.resolve(filename.trim())
+  if (result) return result
+  
+  return {
+    type: "text",
+    read: async () => args || "Not found"
+  }
 }

package/src/rf.tool.js

@@ -16,5 +16,8 @@ module.exports = async function readFile({ filename, linenum }, ctx) {
   if (pathArr.length > 1) {
     return`@{ ${pathArr.join(" | ")} }`;
   }
+  if (relFile.match(/\s+/)) {
+    return `@{${relFile}}`;
+  }
   return `@${relFile}`;
-};
+};