#!/bin/bash
# 构建三个美股 Agent 镜像并推送到 ACR
# 使用方法: ./build_stock_agents.sh

set -e

echo "=========================================="
echo "构建美股 Agent 镜像并推送到 ACR"
echo "=========================================="

# 登录 ACR
echo ""
echo "=== 步骤 1: 登录 ACR ==="
az acr login --name agnettaiji

cd /home/taiji/tools/agent-manager

# 构建 stock-quote-agent
echo ""
echo "=== 步骤 2: 构建 stock-quote-agent (ARM64) ==="
docker buildx build --platform linux/arm64 \
  -t agnettaiji.azurecr.io/ai-agents/stock-quote-agent:latest \
  -f agent_templates/agents/stock_quote_agent/stock_quote_agent.Dockerfile \
  agent_templates/ --push
echo "✅ stock-quote-agent 推送完成"

# 构建 stock-news-agent
echo ""
echo "=== 步骤 3: 构建 stock-news-agent (ARM64) ==="
docker buildx build --platform linux/arm64 \
  -t agnettaiji.azurecr.io/ai-agents/stock-news-agent:latest \
  -f agent_templates/agents/stock_news_agent/stock_news_agent.Dockerfile \
  agent_templates/ --push
echo "✅ stock-news-agent 推送完成"

# 构建 stock-analysis-agent
echo ""
echo "=== 步骤 4: 构建 stock-analysis-agent (ARM64) ==="
docker buildx build --platform linux/arm64 \
  -t agnettaiji.azurecr.io/ai-agents/stock-analysis-agent:latest \
  -f agent_templates/agents/stock_analysis_agent/stock_analysis_agent.Dockerfile \
  agent_templates/ --push
echo "✅ stock-analysis-agent 推送完成"

echo ""
echo "=========================================="
echo "✅ 所有镜像构建并推送完成！"
echo "=========================================="
echo ""
echo "镜像列表:"
echo "  - agnettaiji.azurecr.io/ai-agents/stock-quote-agent:latest"
echo "  - agnettaiji.azurecr.io/ai-agents/stock-news-agent:latest"
echo "  - agnettaiji.azurecr.io/ai-agents/stock-analysis-agent:latest"
echo ""
echo "下一步: 重启 agent-manager 以加载新模板"
echo "  kubectl rollout restart deployment/agent-manager -n agent-manager"