#!/bin/bash
# WeMM-Embedding 服务启动脚本

set -e

SCRIPT_DIR="$(cd "$(dirname "${BASH_SOURCE[0]}")" && pwd)"
source "$SCRIPT_DIR/../s2s-env/bin/activate"

export WEMM_MODEL="${WEMM_MODEL:-tencent/WeMM-Embedding-2B}"
export PORT="${PORT:-8765}"
export MAX_DIMENSION="${MAX_DIMENSION:-2048}"

echo "=========================================="
echo "  WeMM-Embedding 服务"
echo "=========================================="
echo "  模型: $WEMM_MODEL"
echo "  端口: $PORT"
echo "  设备: $(python3 -c 'import torch; print(\"cuda\" if torch.cuda.is_available() else \"cpu\")')"
echo "=========================================="

cd "$SCRIPT_DIR"
python3 wemm_service.py