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}"
 

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")/.."