diff --git a/scripts/agent_poller.py b/scripts/agent_poller.py index f306d44..2fa479e 100644 --- a/scripts/agent_poller.py +++ b/scripts/agent_poller.py @@ -22,10 +22,16 @@ import urllib.error GITEA_URL = os.environ.get("GITEA_URL", "http://localhost:3000") GITEA_REPO = os.environ.get("GITEA_REPO", "pzhang_zywl/document_analyzer") GITEA_TOKEN = os.environ.get("GITEA_API_TOKEN", "") -AGENT_ID = os.environ.get("DEV_AGENT_ID", "da-01") +DEV_AGENT_ID = os.environ.get("DEV_AGENT_ID", "da-01") +QE_AGENT_ID = os.environ.get("QE_AGENT_ID", "") # Signature appended to all comments / PR bodies -AGENT_SIG = f"\n\n---\n[{AGENT_ID}]" +if QE_AGENT_ID: + AGENT_ID = QE_AGENT_ID + AGENT_SIG = f"\n\n---\n[qe-agent: {QE_AGENT_ID}]" +else: + AGENT_ID = DEV_AGENT_ID + AGENT_SIG = f"\n\n---\n[{DEV_AGENT_ID}]" BASE = f"{GITEA_URL}/api/v1/repos/{GITEA_REPO}" diff --git a/scripts/start_qe_agent.sh b/scripts/start_qe_agent.sh index c8bfab7..a2659f4 100644 --- a/scripts/start_qe_agent.sh +++ b/scripts/start_qe_agent.sh @@ -7,6 +7,7 @@ set -e export GITEA_API_TOKEN="59117246ec418d5d87042de073b0d4197d8054bf" export GITEA_URL="http://localhost:3000" export GITEA_REPO="pzhang_zywl/document_analyzer" +export QE_AGENT_ID="qa-01" cd "$(dirname "$0")/.."