skillfree 0.1.4 → 0.1.5

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "skillfree",
-  "version": "0.1.4",
+  "version": "0.1.5",
   "description": "🦞 一个 API，满足所有龙虾技能需求",
   "main": "bin/skillfree.js",
   "bin": {

package/scripts/commands/pilot.js

@@ -75,8 +75,30 @@ async function pilot(flags) {
 
   // ── IMAGE ─────────────────────────────────────────────────────────────────────
   if (type === 'image') {
+    const imageModel = model || 'gemini-3.1-flash-image-preview'
+
+    // qwen-image-edit-plus：图片编辑，走 /v1/images/edits，multipart/form-data
+    if (imageModel.startsWith('qwen-image')) {
+      if (!flags.file) throw new Error('qwen-image-edit 需要 --file 指定原始图片路径')
+      const imgBuf = fs.readFileSync(path.resolve(flags.file))
+      const form = new FormData()
+      form.append('model', imageModel)
+      form.append('prompt', prompt)
+      form.append('size', flags.size || '1024x1024')
+      form.append('image', new Blob([imgBuf], { type: 'image/png' }), path.basename(flags.file))
+      const res = await request('/images/edits', { method: 'POST', body: form })
+      const data = await res.json()
+      if (data.error) throw new Error(data.error.message || JSON.stringify(data.error))
+      const url = data.data?.[0]?.url
+      if (!url) throw new Error('未返回图像URL: ' + JSON.stringify(data).slice(0, 200))
+      if (output) await downloadAndSave(url, output)
+      else console.log('图像 URL:', url)
+      return
+    }
+
+    // doubao-seedream-5.0-lite / 其他标准图像模型
     const res = await post('/images/generations', {
-      model: model || 'gemini-3.1-flash-image-preview',
+      model: imageModel,
       prompt,
       n: 1,
       size: flags.size || '1024x1024',
@@ -85,12 +107,8 @@ async function pilot(flags) {
     const url = res.data?.[0]?.url || res.data?.[0]?.b64_json
     if (!url) throw new Error('未返回图像数据: ' + JSON.stringify(res).slice(0, 200))
     if (output) {
-      if (url.startsWith('http')) {
-        await downloadAndSave(url, output)
-      } else {
-        fs.writeFileSync(output, Buffer.from(url, 'base64'))
-        console.log(`✅ 已保存到 ${output}`)
-      }
+      if (url.startsWith('http')) await downloadAndSave(url, output)
+      else { fs.writeFileSync(output, Buffer.from(url, 'base64')); console.log(`✅ 已保存到 ${output}`) }
     } else {
       console.log('图像 URL:', url)
     }
@@ -216,8 +234,26 @@ async function pilot(flags) {
       throw new Error('Suno 生成超时（150s），请手动查询 task_id: ' + taskId)
 
     } else {
-      // music-2.5（MiniMax）暂不支持（需要 lyrics 参数，建议直接调 /v1/responses）
-      throw new Error(`music-2.5 需要通过 /v1/responses 直接调用，请参考文档`)
+      // music-2.5（MiniMax），走 /v1/responses，需要 lyrics
+      const res = await request('/responses', {
+        method: 'POST',
+        body: JSON.stringify({
+          model: 'music-2.5',
+          input: prompt,
+          lyrics: flags.lyrics || `[verse]\n${prompt}`,
+          audio_setting: { sample_rate: 44100, bitrate: 256000, format: 'mp3' },
+          output_format: 'url',
+          stream: false,
+        }),
+      })
+      const data = await res.json()
+      if (data.error) throw new Error(data.error.message || JSON.stringify(data.error))
+      const audioUrl = data.output?.[0]?.content?.[0]?.audio
+      if (!audioUrl) throw new Error('未返回音频: ' + JSON.stringify(data).slice(0, 200))
+      const duration = ((data.extra_info?.music_duration || 0) / 1000).toFixed(1)
+      console.log(`✅ music-2.5 生成成功！时长约 ${duration} 秒`)
+      console.log('音频 URL:', audioUrl)
+      if (output) await downloadAndSave(audioUrl, output)
     }
   }
 
@@ -265,7 +301,38 @@ async function pilot(flags) {
     return
   }
 
-  throw new Error(`不支持的类型: ${type}，可选: chat | image | tts | stt | music | ocr | video`)
+  // ── EMBEDDING ─────────────────────────────────────────────────────────────────
+  if (type === 'embedding') {
+    // doubao-embedding-vision：走 /v1/responses（多模态），input 为数组
+    // 简单文本 embedding 也可用
+    const inputData = flags.file
+      ? [{ type: 'image_url', image_url: { url: 'data:image/png;base64,' + fs.readFileSync(path.resolve(flags.file)).toString('base64') } }]
+      : [{ type: 'text', text: prompt }]
+    const res = await request('/embeddings', {
+      method: 'POST',
+      body: JSON.stringify({
+        model: model || 'doubao-embedding-vision-251215',
+        input: inputData,
+        encoding_format: 'float',
+        dimensions: 1024,
+        sparse_embedding: { type: 'disabled' },
+      }),
+    })
+    const data = await res.json()
+    if (data.error) throw new Error(data.error.message || JSON.stringify(data.error))
+    const embedding = data.data?.[0]?.embedding
+    if (!embedding) throw new Error('未返回向量: ' + JSON.stringify(data).slice(0, 200))
+    console.log(`✅ Embedding 成功！维度: ${embedding.length}`)
+    if (output) {
+      fs.writeFileSync(output, JSON.stringify(embedding))
+      console.log(`已保存到 ${output}`)
+    } else {
+      console.log('前5维:', embedding.slice(0, 5))
+    }
+    return
+  }
+
+  throw new Error(`不支持的类型: ${type}，可选: chat | image | tts | stt | music | ocr | video | embedding`)
 }
 
 module.exports = { pilot }