ai-workspace-services/accounts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat: Allow internal agent ID to be configured via AGENT_ID environment variable, defaulting to 'internal-agent'.

Browse Source
This commit is contained in:
Haitao Pan 2026-02-05 00:06:18 +08:00
parent 15c07826ae
commit 7cae075709
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
cmd/accountsvc/main.go
View File

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