feat(ai-chat): 新增豆包API + AI模型配置项支持动态切换
- 后端新增豆包（火山引擎Ark）API集成：DoubaoController、ToolDoubaoServiceImpl， 使用OkHttp3 SSE流式对话，兼容OpenAI Chat Completions格式 - 新增DoubaoConfig配置类，读取doubao.api.*配置 - 在eb_system_config表新增ai_chat_model配置项，支持doubao/coze/gemini三种模型切换 - 新增GET /api/front/doubao/ai-model-config接口供前端读取当前模型配置 - 前端ai-nutritionist.vue的sendToAI按系统配置分发到_sendViaDoubao/_sendViaCoze/_sendViaGemini - 前端models-api.js新增doubaoChatStream/doubaoChat/getAiModelConfig函数 - 附带豆包API测试脚本和数据库初始化SQL Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
This commit is contained in:
msh-agent
144
test-doubao-api.sh
Normal file
144
test-doubao-api.sh
Normal file
@@ -0,0 +1,144 @@
|
||||
#!/bin/bash
|
||||
# 豆包 API 连通性测试脚本
|
||||
# 用法: bash test-doubao-api.sh
|
||||
|
||||
API_KEY="18480c26-ebcd-4263-8a6f-48359b8bd65d"
|
||||
BASE_URL="https://ark.cn-beijing.volces.com/api/v3"
|
||||
MODEL="doubao-seed-2-0-pro-260215"
|
||||
|
||||
echo "============================================"
|
||||
echo " 豆包（火山引擎 Ark）API 测试"
|
||||
echo "============================================"
|
||||
echo ""
|
||||
|
||||
# ---- 测试 1: 非流式调用 ----
|
||||
echo "【测试 1】非流式调用"
|
||||
echo "请求: 你好，请用一句话介绍你自己"
|
||||
echo "---"
|
||||
|
||||
START=$(date +%s%3N 2>/dev/null || python3 -c "import time; print(int(time.time()*1000))")
|
||||
|
||||
RESPONSE=$(curl -s -w "\n---HTTP_CODE:%{http_code}---" \
|
||||
-X POST "${BASE_URL}/chat/completions" \
|
||||
-H "Content-Type: application/json" \
|
||||
-H "Authorization: Bearer ${API_KEY}" \
|
||||
-d "{
|
||||
\"model\": \"${MODEL}\",
|
||||
\"messages\": [{\"role\": \"user\", \"content\": \"你好，请用一句话介绍你自己\"}],
|
||||
\"stream\": false
|
||||
}" 2>&1)
|
||||
|
||||
END=$(date +%s%3N 2>/dev/null || python3 -c "import time; print(int(time.time()*1000))")
|
||||
ELAPSED=$((END - START))
|
||||
|
||||
HTTP_CODE=$(echo "$RESPONSE" | grep -oP '(?<=---HTTP_CODE:)\d+(?=---)')
|
||||
BODY=$(echo "$RESPONSE" | sed 's/---HTTP_CODE:[0-9]*---//')
|
||||
|
||||
echo "HTTP 状态码: ${HTTP_CODE}"
|
||||
echo "耗时: ${ELAPSED} ms"
|
||||
|
||||
if [ "$HTTP_CODE" = "200" ]; then
|
||||
# 提取回复内容
|
||||
CONTENT=$(echo "$BODY" | python3 -c "
|
||||
import sys, json
|
||||
data = json.load(sys.stdin)
|
||||
choices = data.get('choices', [])
|
||||
if choices:
|
||||
print('回复:', choices[0].get('message', {}).get('content', '(empty)'))
|
||||
usage = data.get('usage', {})
|
||||
print(f'Token: prompt={usage.get(\"prompt_tokens\",0)}, completion={usage.get(\"completion_tokens\",0)}, total={usage.get(\"total_tokens\",0)}')
|
||||
print(f'模型: {data.get(\"model\", \"?\")}')
|
||||
" 2>&1)
|
||||
echo "$CONTENT"
|
||||
else
|
||||
echo "错误响应: $BODY"
|
||||
fi
|
||||
|
||||
echo ""
|
||||
echo ""
|
||||
|
||||
# ---- 测试 2: 流式调用 ----
|
||||
echo "【测试 2】流式调用 (SSE)"
|
||||
echo "请求: 简单说一下健康饮食的三个要点"
|
||||
echo "---"
|
||||
|
||||
START2=$(date +%s%3N 2>/dev/null || python3 -c "import time; print(int(time.time()*1000))")
|
||||
|
||||
# 流式请求，逐行输出
|
||||
STREAM_OUTPUT=""
|
||||
FIRST_CHUNK_TIME=""
|
||||
|
||||
curl -s -N \
|
||||
-X POST "${BASE_URL}/chat/completions" \
|
||||
-H "Content-Type: application/json" \
|
||||
-H "Authorization: Bearer ${API_KEY}" \
|
||||
-d "{
|
||||
\"model\": \"${MODEL}\",
|
||||
\"messages\": [{\"role\": \"user\", \"content\": \"简单说一下健康饮食的三个要点，每个要点一句话\"}],
|
||||
\"stream\": true
|
||||
}" 2>/dev/null | while IFS= read -r line; do
|
||||
if [[ "$line" == data:* ]]; then
|
||||
JSON_DATA="${line#data: }"
|
||||
if [ "$JSON_DATA" = "[DONE]" ]; then
|
||||
break
|
||||
fi
|
||||
# 提取 delta content
|
||||
DELTA=$(echo "$JSON_DATA" | python3 -c "
|
||||
import sys, json
|
||||
try:
|
||||
d = json.load(sys.stdin)
|
||||
c = d.get('choices',[{}])[0].get('delta',{}).get('content','')
|
||||
if c: print(c, end='', flush=True)
|
||||
except: pass
|
||||
" 2>/dev/null)
|
||||
if [ -z "$FIRST_CHUNK_TIME" ] && [ -n "$DELTA" ]; then
|
||||
FIRST_CHUNK_TIME=$(date +%s%3N 2>/dev/null || python3 -c "import time; print(int(time.time()*1000))")
|
||||
fi
|
||||
fi
|
||||
done
|
||||
|
||||
END2=$(date +%s%3N 2>/dev/null || python3 -c "import time; print(int(time.time()*1000))")
|
||||
ELAPSED2=$((END2 - START2))
|
||||
|
||||
echo ""
|
||||
echo "---"
|
||||
echo "流式总耗时: ${ELAPSED2} ms"
|
||||
|
||||
echo ""
|
||||
echo ""
|
||||
|
||||
# ---- 测试 3: 通过后端接口调用（需先启动后端服务） ----
|
||||
echo "【测试 3】通过后端接口调用（需后端服务运行中）"
|
||||
echo "请确认后端已启动，默认地址: http://localhost:8081"
|
||||
echo "---"
|
||||
|
||||
BACKEND_URL="http://localhost:8081/api/front/doubao/chat"
|
||||
|
||||
RESP3=$(curl -s -w "\n---HTTP_CODE:%{http_code}---" \
|
||||
-X POST "${BACKEND_URL}" \
|
||||
-H "Content-Type: application/json" \
|
||||
-d "{
|
||||
\"messages\": [{\"role\": \"user\", \"content\": \"你好\"}]
|
||||
}" 2>&1)
|
||||
|
||||
HTTP3=$(echo "$RESP3" | grep -oP '(?<=---HTTP_CODE:)\d+(?=---)')
|
||||
BODY3=$(echo "$RESP3" | sed 's/---HTTP_CODE:[0-9]*---//')
|
||||
|
||||
if [ "$HTTP3" = "200" ]; then
|
||||
echo "后端接口调用成功! HTTP $HTTP3"
|
||||
echo "$BODY3" | python3 -c "
|
||||
import sys, json
|
||||
data = json.load(sys.stdin)
|
||||
choices = data.get('choices', [])
|
||||
if choices:
|
||||
print('回复:', choices[0].get('message', {}).get('content', '(empty)')[:100])
|
||||
" 2>&1
|
||||
else
|
||||
echo "后端接口调用失败: HTTP ${HTTP3:-连接失败}"
|
||||
echo "（如果后端未启动，这是正常的）"
|
||||
fi
|
||||
|
||||
echo ""
|
||||
echo "============================================"
|
||||
echo " 测试完成"
|
||||
echo "============================================"
|
||||
Reference in New Issue
Block a user