Configure LLM agent URL

code-asher · code-asher · commit f4e06c6a4843 · 2025-06-12T14:18:25.000-08:00
diff --git a/cli/exp_mcp.go b/cli/exp_mcp.go
@@ -126,6 +126,7 @@ func (r *RootCmd) mcpConfigureClaudeCode() *serpent.Command {
 		coderPrompt      string
 		appStatusSlug    string
 		testBinaryName   string
+		llmAgentURL      url.URL
 
 		deprecatedCoderMCPClaudeAPIKey string
 	)
@@ -164,6 +165,9 @@ func (r *RootCmd) mcpConfigureClaudeCode() *serpent.Command {
 			if appStatusSlug != "" {
 				configureClaudeEnv[envAppStatusSlug] = appStatusSlug
 			}
+			if llmAgentURL.String() != "" {
+				configureClaudeEnv[envLLMAgentURL] = llmAgentURL.String()
+			}
 			if deprecatedSystemPromptEnv, ok := os.LookupEnv("SYSTEM_PROMPT"); ok {
 				cliui.Warnf(inv.Stderr, "SYSTEM_PROMPT is deprecated, use CODER_MCP_CLAUDE_SYSTEM_PROMPT instead")
 				systemPrompt = deprecatedSystemPromptEnv
@@ -262,6 +266,12 @@ func (r *RootCmd) mcpConfigureClaudeCode() *serpent.Command {
 				Flag:        "claude-app-status-slug",
 				Value:       serpent.StringOf(&appStatusSlug),
 			},
+			{
+				Flag:        "llm-agent-url",
+				Description: "The URL of the LLM agent API, used to listen for status updates.",
+				Env:         envLLMAgentURL,
+				Value:       serpent.URLOf(&llmAgentURL),
+			},
 			{
 				Name:        "test-binary-name",
 				Description: "Only used for testing.",
@@ -446,7 +456,9 @@ func (r *RootCmd) mcpServer() *serpent.Command {
 			defer srv.queue.Close()
 
 			cliui.Infof(inv.Stderr, "Failed to watch screen events")
-			// Start the reporter, watcher, and server.
+			// Start the reporter, watcher, and server.  These are all tied to the
+			// lifetime of the MCP server, which is itself tied to the lifetime of the
+			// LLM agent.
 			if srv.agentClient != nil && appStatusSlug != "" {
 				srv.startReporter(ctx, inv)
 				if srv.llmClient != nil {
diff --git a/cli/exp_mcp_test.go b/cli/exp_mcp_test.go
@@ -365,7 +365,8 @@ test-system-prompt
 							"env": {
 								"CODER_AGENT_URL": "%s",
 								"CODER_AGENT_TOKEN": "test-agent-token",
-								"CODER_MCP_APP_STATUS_SLUG": "some-app-name"
+								"CODER_MCP_APP_STATUS_SLUG": "some-app-name",
+								"CODER_MCP_LLM_AGENT_URL": "http://localhost:3284"
 							}
 						}
 					}
@@ -390,6 +391,7 @@ test-system-prompt
 			"--claude-test-binary-name=pathtothecoderbinary",
 			"--agent-url", client.URL.String(),
 			"--agent-token", "test-agent-token",
+			"--llm-agent-url", "http://localhost:3284",
 		)
 		clitest.SetupConfig(t, client, root)
 

Original file line number	Diff line number	Diff line change
`@@ -365,7 +365,8 @@ test-system-prompt`
`365`	`365`	`"env": {`
`366`	`366`	`"CODER_AGENT_URL": "%s",`
`367`	`367`	`"CODER_AGENT_TOKEN": "test-agent-token",`
`368`		`- "CODER_MCP_APP_STATUS_SLUG": "some-app-name"`
	`368`	`+ "CODER_MCP_APP_STATUS_SLUG": "some-app-name",`
	`369`	`+ "CODER_MCP_LLM_AGENT_URL": "http://localhost:3284"`
`369`	`370`	`}`
`370`	`371`	`}`
`371`	`372`	`}`
`@@ -390,6 +391,7 @@ test-system-prompt`
`390`	`391`	`"--claude-test-binary-name=pathtothecoderbinary",`
`391`	`392`	`"--agent-url", client.URL.String(),`
`392`	`393`	`"--agent-token", "test-agent-token",`
	`394`	`+ "--llm-agent-url", "http://localhost:3284",`
`393`	`395`	`)`
`394`	`396`	`clitest.SetupConfig(t, client, root)`
`395`	`397`