#!/bin/bash
# Baidu Search CLI wrapper
# Usage: ./search.sh <query> [top_k]

QUERY="${1:-}"
TOP_K="${2:-5}"

if [ -z "$QUERY" ]; then
  echo '{"error":"No search query provided"}'
  exit 1
fi

API_KEY="${BAIDU_API_KEY:-}"
if [ -z "$API_KEY" ]; then
  echo '{"error":"BAIDU_API_KEY not set"}'
  exit 1
fi

REQUEST_DATA=$(cat <<EOF
{
  "messages": [{"content": $(echo "$QUERY" | python3 -c "import json,sys; print(json.dumps(sys.stdin.read().strip()))"), "role": "user"}],
  "search_source": "web",
  "resource_type_filter": [{"type": "web", "top_k": $TOP_K}]
}
EOF
)

curl -s -X POST \
  -H "Authorization: Bearer $API_KEY" \
  -H "X-Appbuilder-From: openclaw" \
  -H "Content-Type: application/json" \
  -d "$REQUEST_DATA" \
  "https://qianfan.baidubce.com/v2/ai_search/web_search"