wehub-resource-sync
72 lines
1.7 KiB
Bash
Executable File
72 lines
1.7 KiB
Bash
Executable File
#!/usr/bin/env bash
|
|
set -euo pipefail
|
|
|
|
ROOT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")/.." && pwd)"
|
|
|
|
if [[ -f "${ROOT_DIR}/.env" ]]; then
|
|
set -a
|
|
# shellcheck disable=SC1091
|
|
source "${ROOT_DIR}/.env"
|
|
set +a
|
|
fi
|
|
|
|
# shellcheck disable=SC1091
|
|
source "${ROOT_DIR}/scripts/dashscope_env.sh"
|
|
|
|
usage() {
|
|
cat <<'EOF'
|
|
Usage:
|
|
scripts/dashscope_run.sh official-test
|
|
scripts/dashscope_run.sh generate "prompt text"
|
|
|
|
Commands:
|
|
official-test Run a minimal official OpenAI-compatible Chat Completions request.
|
|
generate Run Articraft generation with --provider dashscope.
|
|
EOF
|
|
}
|
|
|
|
cmd="${1:-}"
|
|
case "${cmd}" in
|
|
official-test)
|
|
if [[ -z "${DASHSCOPE_API_KEY:-}" ]]; then
|
|
echo "Missing DASHSCOPE_API_KEY. Put it in .env." >&2
|
|
exit 1
|
|
fi
|
|
uv --cache-dir "${ROOT_DIR}/.uv-cache" run python - <<'PY'
|
|
import os
|
|
from openai import OpenAI
|
|
|
|
client = OpenAI(
|
|
api_key=os.environ["DASHSCOPE_API_KEY"],
|
|
base_url=os.environ["DASHSCOPE_BASE_URL"],
|
|
)
|
|
resp = client.chat.completions.create(
|
|
model=os.environ.get("DASHSCOPE_MODEL", "qwen3.6-flash"),
|
|
messages=[{"role": "user", "content": "只回复: ok"}],
|
|
extra_body={"enable_thinking": True},
|
|
)
|
|
print(resp.choices[0].message.content)
|
|
PY
|
|
;;
|
|
generate)
|
|
if [[ -z "${DASHSCOPE_API_KEY:-}" ]]; then
|
|
echo "Missing DASHSCOPE_API_KEY. Put it in .env." >&2
|
|
exit 1
|
|
fi
|
|
shift
|
|
prompt="${1:-}"
|
|
if [[ -z "${prompt}" ]]; then
|
|
echo "Missing prompt. Usage: scripts/dashscope_run.sh generate \"prompt text\"" >&2
|
|
exit 1
|
|
fi
|
|
uv --cache-dir "${ROOT_DIR}/.uv-cache" run articraft generate \
|
|
--provider dashscope \
|
|
--model "${DASHSCOPE_MODEL}" \
|
|
"${prompt}"
|
|
;;
|
|
*)
|
|
usage
|
|
exit 2
|
|
;;
|
|
esac
|