From 7cae075709dafb6cb4f50a0d6a847e83e516e205 Mon Sep 17 00:00:00 2001 From: Haitao Pan Date: Thu, 5 Feb 2026 00:06:18 +0800 Subject: [PATCH] feat: Allow internal agent ID to be configured via `AGENT_ID` environment variable, defaulting to 'internal-agent'. --- cmd/accountsvc/main.go | 9 +++++++-- 1 file changed, 7 insertions(+), 2 deletions(-) diff --git a/cmd/accountsvc/main.go b/cmd/accountsvc/main.go index 96207c5..aa3ef8f 100644 --- a/cmd/accountsvc/main.go +++ b/cmd/accountsvc/main.go @@ -658,10 +658,15 @@ func runServer(ctx context.Context, cfg *config.Config, logger *slog.Logger) err } } else if token := os.Getenv("INTERNAL_SERVICE_TOKEN"); token != "" { // Fallback: if no credentials configured but we have an internal token, - // register a default internal agent. + // accept any agent that presents this token (ID will be taken from agent's self-reported ID) + // This allows the agent to use its configured ID (e.g., "hk-xhttp.svc.plus") + agentID := strings.TrimSpace(os.Getenv("AGENT_ID")) + if agentID == "" { + agentID = "internal-agent" // fallback ID if not specified + } agentRegistry, err = agentserver.NewRegistry(agentserver.Config{ Credentials: []agentserver.Credential{{ - ID: "internal-agent", + ID: agentID, Name: "Internal Agent", Token: token, Groups: []string{"internal"},