Ai config

2026-05-26 13:37:55 +08:00
parent 55452a2d21
commit 0c7ed3498d
42 changed files with 1264 additions and 767 deletions
@@ -0,0 +1,78 @@
+const { query } = require('../config/database')
+const fetch = require('node-fetch')
+const { AI_VL_MODEL, AI_RECOGNIZE_LIMIT } = require('../config/constants')
+
+const AI_API_KEY = process.env.DASHSCOPE_API_KEY
+const AI_API_URL = 'https://dashscope.aliyuncs.com/compatible-mode/v1/chat/completions'
+
+async function getByBarcode(ctx) {
+  try {
+    const { barcode } = ctx.request.body
+    if (!barcode) {
+      ctx.body = { code: 400, message: '请提供条形码' }
+      return
+    }
+    const goods = await query('SELECT * FROM goods WHERE barcode = ? LIMIT 1', [barcode])
+    if (goods.length > 0) {
+      ctx.body = { code: 200, data: goods[0] }
+    } else {
+      ctx.body = { code: 404, message: '未找到该商品' }
+    }
+  } catch (error) {
+    console.error('条码查询失败:', error)
+    ctx.body = { code: 500, message: '查询失败' }
+  }
+}
+
+async function recognizeImage(ctx) {
+  try {
+    const { imageData } = ctx.request.body
+    if (!imageData) {
+      ctx.body = { code: 400, message: '请提供图片数据' }
+      return
+    }
+    if (!AI_API_KEY) {
+      ctx.body = { code: 500, message: 'AI 识别未配置' }
+      return
+    }
+
+    const response = await fetch(AI_API_URL, {
+      method: 'POST',
+      headers: {
+        'Authorization': `Bearer ${AI_API_KEY}`,
+        'Content-Type': 'application/json'
+      },
+      body: JSON.stringify({
+        model: AI_VL_MODEL,
+        messages: [
+          {
+            role: 'user',
+            content: [
+              { type: 'text', text: '请识别这张图片中的商品，返回商品名称。只返回名称，不要其他内容。' },
+              { type: 'image_url', image_url: { url: imageData } }
+            ]
+          }
+        ]
+      })
+    })
+
+    const result = await response.json()
+    const name = result?.choices?.[0]?.message?.content?.trim() || ''
+    const goods = name
+      ? await query('SELECT * FROM goods WHERE name LIKE ? LIMIT ?', [`%${name}%`, AI_RECOGNIZE_LIMIT])
+      : []
+
+    ctx.body = {
+      code: 200,
+      data: {
+        message: name ? `识别到: ${name}` : '未识别到商品',
+        goods: goods || []
+      }
+    }
+  } catch (error) {
+    console.error('图片识别失败:', error)
+    ctx.body = { code: 500, message: '识别失败' }
+  }
+}
+
+module.exports = { getByBarcode, recognizeImage }