@showlotus/opencode-image-vision 1.0.1 → 1.0.2

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@showlotus/opencode-image-vision",
-  "version": "1.0.1",
+  "version": "1.0.2",
   "description": "MCP server that reads images from OpenCode's SQLite database and analyzes them via vision AI providers",
   "type": "module",
   "packageManager": "pnpm@8.11.0",

package/src/concurrency.js

@@ -0,0 +1,19 @@
+// 并发执行任务，控制最大并发数，结果按原顺序返回
+export async function mapWithConcurrency(items, concurrency, worker) {
+  const results = new Array(items.length)
+  let cursor = 0
+
+  // 单个执行器：循环领取下一个未处理的任务，直到全部完成
+  const run = async () => {
+    while (cursor < items.length) {
+      const i = cursor++
+      results[i] = await worker(items[i], i)
+    }
+  }
+
+  // 启动不超过任务数量的并发执行器
+  await Promise.all(
+    Array.from({ length: Math.min(concurrency, items.length) }, run),
+  )
+  return results
+}

package/src/index.js

@@ -5,6 +5,7 @@ import { StdioServerTransport } from '@modelcontextprotocol/sdk/server/stdio.js'
 import { z } from 'zod'
 import { getDatabase, getImages } from './db.js'
 import { createProvider } from './providers/index.js'
+import { mapWithConcurrency } from './concurrency.js'
 
 const DEFAULT_PROMPT =
   process.env.prompt ||
@@ -17,6 +18,9 @@ const DEFAULT_PROMPT =
 const DEFAULT_LIMIT = Number(process.env.limit) || 5
 const MAX_LIMIT = Number(process.env.max_limit) || 20
 
+// 并发分析图片的最大并发数，可通过环境变量 concurrency 覆盖
+const DEFAULT_CONCURRENCY = Number(process.env.concurrency) || 5
+
 let provider
 try {
   provider = createProvider()
@@ -60,16 +64,19 @@ server.tool(
         }
       }
 
-      const results = []
-      for (let i = 0; i < images.length; i++) {
-        const img = images[i]
-        try {
-          const desc = await provider.analyze(img.base64, img.mime, analysisPrompt)
-          results.push(`### Image ${i + 1}: ${img.filename}\n\n${desc}`)
-        } catch (e) {
-          results.push(`### Image ${i + 1}: ${img.filename}\n\n[Analysis failed: ${e.message}]`)
-        }
-      }
+      // 并发分析图片，单张失败不影响其他图片，结果保持原顺序
+      const results = await mapWithConcurrency(
+        images,
+        DEFAULT_CONCURRENCY,
+        async (img, i) => {
+          try {
+            const desc = await provider.analyze(img.base64, img.mime, analysisPrompt)
+            return `### Image ${i + 1}: ${img.filename}\n\n${desc}`
+          } catch (e) {
+            return `### Image ${i + 1}: ${img.filename}\n\n[Analysis failed: ${e.message}]`
+          }
+        },
+      )
 
       return {
         content: [