pinokiod 7.5.16 → 7.5.17

package/kernel/api/hf/index.js

@@ -327,5 +327,36 @@ class HF {
     let res = await shell.run(req, ondata, kernel)
     return res
   }
+  /*
+  {
+    "method": "hf.upload",
+    "params": {
+      path: <cwd>,
+      env: {
+        <env1>: <val1>,
+        <env1>: <val1>,
+        <env1>: <val1>,
+      },
+      _: [<command line args>'],
+      <arg1>: <val1>,
+      <arg2>: <val2>,
+      ...
+    }
+  }
+  */
+  async upload(req, ondata, kernel) {
+    const shell = new Shell()
+    let params = Object.assign({}, req.params)
+    delete params.env
+    delete params.path
+    const positional = Array.isArray(params._) ? params._ : (params._ == null ? [] : [params._])
+    params._ = ["hf", "upload", ...positional]
+    let message = [
+      params
+    ]
+    req.params.message = message
+    let res = await shell.run(req, ondata, kernel)
+    return res
+  }
 }
 module.exports = HF

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "pinokiod",
-  "version": "7.5.16",
+  "version": "7.5.17",
   "description": "",
   "main": "index.js",
   "scripts": {

package/test/hf-api.test.js

@@ -408,6 +408,53 @@ test('hf.logout delegates to the Hugging Face connect provider and returns succe
   assert.deepEqual(calls, [{ method: 'logout', provider: 'huggingface', params: req.params }])
 })
 
+test('hf.upload runs hf upload through shell.run with Hugging Face CLI args', async () => {
+  const hf = new HF()
+  const calls = []
+  const kernel = {
+    platform: 'darwin',
+    shell: {
+      async run(params, options) {
+        calls.push({
+          params,
+          options
+        })
+        return { status: 'success' }
+      }
+    }
+  }
+  const req = {
+    cwd: '/pinokio/api/test',
+    parent: { id: 'test-script' },
+    params: {
+      path: '/pinokio/api/test/output',
+      env: { TEST_ENV: '1' },
+      _: ['cocktailpeanut/privatetest', './output', '.'],
+      'repo-type': 'dataset',
+      private: true,
+      'commit-message': 'initial upload'
+    }
+  }
+
+  const result = await hf.upload(req, () => {}, kernel)
+
+  assert.deepEqual(result, { status: 'success' })
+  assert.equal(calls.length, 1)
+  assert.deepEqual(calls[0].params.message, [
+    {
+      _: ['hf', 'upload', 'cocktailpeanut/privatetest', './output', '.'],
+      'repo-type': 'dataset',
+      private: true,
+      'commit-message': 'initial upload'
+    }
+  ])
+  assert.deepEqual(calls[0].params.env, { TEST_ENV: '1' })
+  assert.equal(calls[0].params.path, '/pinokio/api/test/output')
+  assert.equal(calls[0].params.bluefairy, 'off')
+  assert.equal(calls[0].options.cwd, '/pinokio/api/test')
+  assert.equal(calls[0].options.group, 'test-script')
+})
+
 test('hf.login reports unavailable Hugging Face connect provider', async () => {
   const hf = new HF()