Skip to content

feat: add CopilotService gRPC service for extension framework #7172

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
wbreza merged 11 commits into from
Mar 19, 2026
Merged

feat: add CopilotService gRPC service for extension framework #7172

Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
11 commits
Select commit Hold shift + click to select a range
89b561a
feat: add CopilotService gRPC service for extension framework
wbreza Mar 17, 2026
5dde879
feat(demo): add copilot chat command demonstrating CopilotService gRP…
wbreza Mar 18, 2026
a422008
fix: eagerly start copilot client in CreateSession to surface errors …
wbreza Mar 18, 2026
8501f0e
refactor: simplify CopilotService API — remove CreateSession/ResumeSe…
wbreza Mar 18, 2026
870122d
fix(demo): remove per-turn usage display and headless mode
wbreza Mar 18, 2026
dd89c45
fix: multiple UX and reliability improvements
wbreza Mar 18, 2026
605b8ef
refactor: extract Agent/AgentFactory interfaces and add copilotServic…
wbreza Mar 18, 2026
6b9695f
fix: add missing CopilotService param to new server_test from main merge
wbreza Mar 18, 2026
4631416
refactor: introduce AgentMetrics type with String() for display
wbreza Mar 18, 2026
5664aa5
fix: address PR review feedback — flaky tests, resource leaks, thread…
wbreza Mar 18, 2026
0eafddc
feat: add GetMessages RPC to expose Copilot session event log
wbreza Mar 19, 2026
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions cli/azd/cmd/container.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -942,6 +942,7 @@ func registerCommonDependencies(container *ioc.NestedContainer) {
container.MustRegisterSingleton(grpcserver.NewServiceTargetService)
container.MustRegisterSingleton(grpcserver.NewFrameworkService)
container.MustRegisterSingleton(grpcserver.NewAiModelService)
container.MustRegisterScoped(grpcserver.NewCopilotService)

// Required for nested actions called from composite actions like 'up'
registerAction[*cmd.ProvisionAction](container, "azd-provision-action")
Expand Down
12 changes: 6 additions & 6 deletions cli/azd/cmd/init.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -139,7 +139,7 @@ type initAction struct {
featuresManager *alpha.FeatureManager
extensionsManager *extensions.Manager
azd workflow.AzdCommandRunner
agentFactory *agent.CopilotAgentFactory
agentFactory agent.AgentFactory
consentManager consent.ConsentManager
configManager config.UserConfigManager
}
Expand All @@ -156,7 +156,7 @@ func newInitAction(
featuresManager *alpha.FeatureManager,
extensionsManager *extensions.Manager,
azd workflow.AzdCommandRunner,
agentFactory *agent.CopilotAgentFactory,
agentFactory agent.AgentFactory,
consentManager consent.ConsentManager,
configManager config.UserConfigManager,
) actions.Action {
Expand Down Expand Up @@ -522,7 +522,7 @@ When complete, provide a brief summary of what was accomplished.`

i.console.Message(ctx, color.MagentaString("Preparing application for Azure deployment..."))

result, err := copilotAgent.SendMessageWithRetry(ctx, prompt, opts...)
_, err = copilotAgent.SendMessageWithRetry(ctx, prompt, opts...)
if err != nil {
return err
}
Expand All @@ -532,10 +532,10 @@ When complete, provide a brief summary of what was accomplished.`
_ = azdCtx.ClearCopilotSession()
}

// Show usage
if usage := result.Usage.Format(); usage != "" {
// Show session metrics (usage + file changes)
if metrics := copilotAgent.GetMetrics().String(); metrics != "" {
i.console.Message(ctx, "")
i.console.Message(ctx, usage)
i.console.Message(ctx, metrics)
}

i.console.Message(ctx, "")
Expand Down
6 changes: 3 additions & 3 deletions cli/azd/cmd/middleware/error.go
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,7 @@ var (
type ErrorMiddleware struct {
options *Options
console input.Console
agentFactory *agent.CopilotAgentFactory
agentFactory agent.AgentFactory
global *internal.GlobalCommandOptions
featuresManager *alpha.FeatureManager
userConfigManager config.UserConfigManager
Expand Down Expand Up @@ -147,7 +147,7 @@ func shouldSkipErrorAnalysis(err error) bool {

func NewErrorMiddleware(
options *Options, console input.Console,
agentFactory *agent.CopilotAgentFactory,
agentFactory agent.AgentFactory,
global *internal.GlobalCommandOptions,
featuresManager *alpha.FeatureManager,
userConfigManager config.UserConfigManager,
Expand Down Expand Up @@ -275,7 +275,7 @@ func (e *ErrorMiddleware) Run(ctx context.Context, next NextFn) (*actions.Action
// Display usage metrics if available
if agentResult != nil && agentResult.Usage.TotalTokens() > 0 {
e.console.Message(ctx, "")
e.console.Message(ctx, agentResult.Usage.Format())
e.console.Message(ctx, agentResult.Usage.String())
}

// Ask user if the agent applied a fix and they want to retry the command
Expand Down
3 changes: 3 additions & 0 deletions cli/azd/extensions/microsoft.azd.demo/extension.yaml
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -36,3 +36,6 @@ examples:
- name: mcp
description: Start MCP server with demo tools.
usage: azd demo mcp start
- name: copilot
description: Interactive Copilot chat loop demonstrating CopilotService gRPC API.
usage: azd demo copilot [--model <model>] [--resume]
Loading
Loading