Adicionar um servidor MCP a uma extensão

Uma extensão Azure Developer CLI (azd) pode expor ferramentas a agentes de IA através de um servidor Model Context Protocol (MCP). Quando adiciona a capacidade mcp-server, agentes como o GitHub Copilot podem detetar e invocar as ferramentas da sua extensão. Este artigo explica como adicionar um servidor MCP à extensão de exemplo Contoso Resource Tagger do início rápido Criar uma extensão de exemplo. Podes aplicar o mesmo padrão a qualquer extensão.

Note

azd As extensões estão atualmente em versão beta.

Como funcionam os servidores MCP

O Model Context Protocol é um standard aberto que permite aos agentes de IA descobrir e chamar ferramentas fornecidas por servidores externos. Quando a sua extensão declara a capacidade mcp-server, o azd pode iniciar a sua extensão como um servidor MCP e encaminhar os pedidos de ferramenta de um agente para a sua extensão. A sua extensão implementa ferramentas que o agente recorre para realizar tarefas, como sugerir etiquetas padronizadas para um projeto.

Declarar a capacidade

Adicione a capacidade mcp-server ao ficheiro de manifesto extension.yaml:

capabilities:
  - custom-commands
  - mcp-server

Também podes adicionar uma mcp secção ao manifesto para configurar como azd inicia o servidor MCP. A serve propriedade suporta args e env; azd já conhece o executável de extensão, por isso não especificas um comando:

mcp:
  serve:
    args:
      - mcp
      - start

Adicionar um comando MCP

Adiciona um comando que inicia o servidor MCP. azd invoca este comando para executar a sua extensão como servidor MCP. Use uma biblioteca de servidores MCP para Go, como mcp-go, para gerir o protocolo.

  1. Crie um ficheiro com um nome mcp.go no seu internal/cmd diretório:

    package cmd
    
    import (
        "context"
        "fmt"
    
        "github.com/mark3labs/mcp-go/mcp"
        "github.com/mark3labs/mcp-go/server"
        "github.com/spf13/cobra"
    )
    
    func newMcpCommand() *cobra.Command {
        mcpCmd := &cobra.Command{
            Use:    "mcp",
            Short:  "Model Context Protocol server commands.",
            Hidden: true,
        }
    
        mcpCmd.AddCommand(newMcpStartCommand())
        return mcpCmd
    }
    
    func newMcpStartCommand() *cobra.Command {
        return &cobra.Command{
            Use:   "start",
            Short: "Starts the MCP server.",
            RunE: func(cmd *cobra.Command, args []string) error {
                s := server.NewMCPServer(
                    "Contoso Resource Tagger",
                    "0.1.0",
                )
    
                // Register the suggest_tags tool.
                s.AddTool(newSuggestTagsTool())
    
                // Serve over stdio so azd and agents can connect.
                return server.ServeStdio(s)
            },
        }
    }
    
  2. Regista o mcp comando no teu comando raiz:

    rootCmd.AddCommand(newMcpCommand())
    

Implementar uma ferramenta

Defina a ferramenta que o agente de IA chama. Adicione a seguinte suggest_tags ferramenta ao mesmo internal/cmd/mcp.go ficheiro que criou anteriormente. Os identificadores mcp e server provêm dos pacotes já importados nesse ficheiro (github.com/mark3labs/mcp-go/mcp e github.com/mark3labs/mcp-go/server), e o processador também utiliza as importações context e fmt. Adicionar a função a mcp.go reutiliza essas importações para que o código compile:

func newSuggestTagsTool() (mcp.Tool, server.ToolHandlerFunc) {
    tool := mcp.NewTool(
        "suggest_tags",
        mcp.WithDescription("Suggests standardized Azure resource tags for a project."),
        mcp.WithString("environment",
            mcp.Description("The target environment, such as dev or prod."),
            mcp.Required(),
        ),
    )

    handler := func(ctx context.Context, request mcp.CallToolRequest) (*mcp.CallToolResult, error) {
        environment, err := request.RequireString("environment")
        if err != nil {
            return mcp.NewToolResultError(err.Error()), nil
        }

        suggestion := fmt.Sprintf(
            "Suggested tags: environment=%s, managed-by=azd, cost-center=<required>",
            environment,
        )

        return mcp.NewToolResultText(suggestion), nil
    }

    return tool, handler
}

Testar o servidor MCP

Depois de adicionares o servidor MCP, reconstrói a tua extensão e verifica se o servidor inicia:

  1. Recriar a extensão:

    azd x build
    
  2. Inicie o servidor MCP diretamente para verificar se corre sem erros:

    azd tagger mcp start
    

    O servidor inicia e ouve com entrada e saída padrão. Um cliente MCP, como um agente de IA configurado para usar a sua extensão, liga-se a este servidor para chamar as suas ferramentas.