mirror of
https://github.com/zhayujie/chatgpt-on-wechat.git
synced 2026-02-25 08:57:51 +08:00
139 lines
4.4 KiB
Bash
Executable File
139 lines
4.4 KiB
Bash
Executable File
#!/usr/bin/env bash
|
||
# LinkAI Agent Caller
|
||
# API Docs: https://api.link-ai.tech/v1/chat/completions
|
||
|
||
set -euo pipefail
|
||
|
||
app_code="${1:-}"
|
||
question="${2:-}"
|
||
model="${3:-}"
|
||
stream="${4:-false}"
|
||
timeout="${5:-120}" # Default 120 seconds for video/image generation
|
||
|
||
if [ -z "$app_code" ]; then
|
||
echo '{"error": "app_code is required", "usage": "bash call.sh <app_code> <question> [model] [stream] [timeout]"}'
|
||
exit 1
|
||
fi
|
||
|
||
if [ -z "$question" ]; then
|
||
echo '{"error": "question is required", "usage": "bash call.sh <app_code> <question> [model] [stream] [timeout]"}'
|
||
exit 1
|
||
fi
|
||
|
||
if [ -z "${LINKAI_API_KEY:-}" ]; then
|
||
echo '{"error": "LINKAI_API_KEY environment variable is not set", "help": "Use env_config to set LINKAI_API_KEY"}'
|
||
exit 1
|
||
fi
|
||
|
||
# API endpoint
|
||
api_url="https://api.link-ai.tech/v1/chat/completions"
|
||
|
||
# Build JSON request body
|
||
if [ -n "$model" ]; then
|
||
request_body=$(cat <<EOF
|
||
{
|
||
"app_code": "$app_code",
|
||
"model": "$model",
|
||
"messages": [
|
||
{
|
||
"role": "user",
|
||
"content": "$question"
|
||
}
|
||
],
|
||
"stream": $stream
|
||
}
|
||
EOF
|
||
)
|
||
else
|
||
request_body=$(cat <<EOF
|
||
{
|
||
"app_code": "$app_code",
|
||
"messages": [
|
||
{
|
||
"role": "user",
|
||
"content": "$question"
|
||
}
|
||
],
|
||
"stream": $stream
|
||
}
|
||
EOF
|
||
)
|
||
fi
|
||
|
||
# Call LinkAI API
|
||
response=$(curl -sS --max-time "$timeout" \
|
||
-X POST \
|
||
-H "Authorization: Bearer $LINKAI_API_KEY" \
|
||
-H "Content-Type: application/json" \
|
||
-d "$request_body" \
|
||
"$api_url" 2>&1)
|
||
|
||
curl_exit_code=$?
|
||
|
||
if [ $curl_exit_code -ne 0 ]; then
|
||
echo "{\"error\": \"Failed to call LinkAI API\", \"details\": \"$response\"}"
|
||
exit 1
|
||
fi
|
||
|
||
# Simple JSON validation
|
||
if [[ ! "$response" =~ ^[[:space:]]*[\{\[] ]]; then
|
||
echo "{\"error\": \"Invalid JSON response from API\", \"response\": \"$response\"}"
|
||
exit 1
|
||
fi
|
||
|
||
# Check for API error (top-level error only, not content_filter_result)
|
||
if echo "$response" | grep -q '^[[:space:]]*{[[:space:]]*"error"[[:space:]]*:' || echo "$response" | grep -q '"error"[[:space:]]*:[[:space:]]*{[^}]*"code"[[:space:]]*:[[:space:]]*"[^"]*"[^}]*"message"'; then
|
||
# Make sure it's not just content_filter_result inside choices
|
||
if ! echo "$response" | grep -q '"choices"[[:space:]]*:[[:space:]]*\['; then
|
||
# Extract error message
|
||
error_msg=$(echo "$response" | grep -o '"message"[[:space:]]*:[[:space:]]*"[^"]*"' | sed 's/"message"[[:space:]]*:[[:space:]]*"\(.*\)"/\1/' | head -1)
|
||
error_code=$(echo "$response" | grep -o '"code"[[:space:]]*:[[:space:]]*"[^"]*"' | sed 's/"code"[[:space:]]*:[[:space:]]*"\(.*\)"/\1/' | head -1)
|
||
|
||
if [ -z "$error_msg" ]; then
|
||
error_msg="Unknown API error"
|
||
fi
|
||
|
||
# Provide friendly error message for content filter
|
||
if [ "$error_code" = "content_filter_error" ] || echo "$error_msg" | grep -qi "content.*filter"; then
|
||
echo "{\"error\": \"内容安全审核\", \"message\": \"您的问题或应用返回的内容触发了LinkAI的安全审核机制，请换一种方式提问或检查应用配置\", \"details\": \"$error_msg\"}"
|
||
else
|
||
echo "{\"error\": \"LinkAI API error\", \"message\": \"$error_msg\", \"code\": \"$error_code\"}"
|
||
fi
|
||
exit 1
|
||
fi
|
||
fi
|
||
|
||
# For non-stream mode, extract and format the response
|
||
if [ "$stream" = "false" ]; then
|
||
# Extract content from response
|
||
content=$(echo "$response" | grep -o '"content"[[:space:]]*:[[:space:]]*"[^"]*"' | sed 's/"content"[[:space:]]*:[[:space:]]*"\(.*\)"/\1/' | head -1)
|
||
|
||
# Extract usage information
|
||
prompt_tokens=$(echo "$response" | grep -o '"prompt_tokens"[[:space:]]*:[[:space:]]*[0-9]*' | grep -o '[0-9]*' | head -1)
|
||
completion_tokens=$(echo "$response" | grep -o '"completion_tokens"[[:space:]]*:[[:space:]]*[0-9]*' | grep -o '[0-9]*' | head -1)
|
||
total_tokens=$(echo "$response" | grep -o '"total_tokens"[[:space:]]*:[[:space:]]*[0-9]*' | grep -o '[0-9]*' | head -1)
|
||
|
||
if [ -n "$content" ]; then
|
||
# Unescape JSON content
|
||
content=$(echo "$content" | sed 's/\\n/\n/g' | sed 's/\\"/"/g')
|
||
|
||
cat <<EOF
|
||
{
|
||
"app_code": "$app_code",
|
||
"content": "$content",
|
||
"usage": {
|
||
"prompt_tokens": ${prompt_tokens:-0},
|
||
"completion_tokens": ${completion_tokens:-0},
|
||
"total_tokens": ${total_tokens:-0}
|
||
}
|
||
}
|
||
EOF
|
||
else
|
||
# Return full response if we can't extract content
|
||
echo "$response"
|
||
fi
|
||
else
|
||
# For stream mode, return raw response (caller needs to handle streaming)
|
||
echo "$response"
|
||
fi
|