riq0h.jp/content/post/test.md

title	date	draft	tags
NeovimでGithub Copilotを試す	2024-09-30T09:55:09+09:00	true	tech

LLMの技術的成果が目に見えて我々に還元されるのを感じる今日この頃、さすがにろくに使い方も知らないのでは困るということでついにGithub Copilotを試すことにした。LLMを応用した検索エンジン（Perprexy）は以前から使っていたが、開発環境と密に統合されるとどう変化するのかは当然気になるところである。

さて、僕の開発環境はNeovimなのでそれにしたがってCopilotを設定していく。Github Copilotは最低月額10ドルからのサブスクサービスなのだが、契約完了時に送られてきたクイックガイドにNeovim向けのものがあったのには少々驚かせられた。大抵Vim向けの設定というのは各々が手弁当で勝手に生やす場合がほとんどだったからだ。なんならNeovimのプラグインも公式で用意されている。

だが、本稿ではあえてこれは使わず有志がLuaで書き直したcopilot.luaの方を用いる。公式のものと比べて処理の効率化が図られている。プラグインマネージャにはlazy.nvimを使うものとする。まずは下記の通りにプラグインを記述して起動時に読み込まれるように設定する。

---lazy
{ "zbirenbaum/copilot.lua", cmd = "Copilot" },

---copilot
require("copilot").setup({
    suggestion = { enabled = false },
    panel = { enabled = false },
})

上記の設定でsuggestionがfalseになっている理由は標準とは異なる形でコード支援を受けるためだ。trueにすると任意のキーでエディタ上にコードの予測が現れる。また、auto_triggerを項目に追加してtrueに設定すると自動的に表示される。しかし、本稿では補完プラグインであるnvim-cmpとの連携を行うため、これらは無効化しておく必要がある。

次に、一旦Copilotの認証を済ませる。プラグインが読み込まれた後に:Copilot authを実行すると、時限制のトークンコードと共にGitHubの認証ページが案内される。ブラウザ上で認証を終えるとスムーズに連携が行われる。自前で作成したファイルにsecretを格納する形式かと思っていたので、さすが今時は違うなと素直に感心した次第だ。

続いてnvim-cmpと連携を行う。標準設定ではエディタ上に薄くハイライトされるが、この設定を通じて補完候補の一つとして表示されるようになる。個人的には納得のいかないサジェストが前面にぶわっと出るよりこっちの方が好ましいと感じている。nvim-cmp本体の設定はすでに実施されているものとして、本稿では割愛させて頂く。

---lazy
{ "zbirenbaum/copilot-cmp", config = true, event = "InsertEnter" },


---nvim-cmp
local cmp = require("cmp")

--他の設定は省略

sources = cmp.config.sources({
    { name = "nvim_lsp", max_item_count = 15, keyword_length = 2 },
    { name = "vsnip", max_item_count = 15, keyword_length = 2 },
    { name = "copilot", max_item_count = 15, keyword_length = 2 }, -- copilotを補完ソースに追加する
    { name = "nvim_lsp_signature_help" },
    { name = "buffer", max_item_count = 15, keyword_length = 2 },
}),
})

local capabilities = require("cmp_nvim_lsp").default_capabilities()

max_item_countは補完候補の最大量でkeyword_lengthは補完が発動する最小のキーワード数を意味する。不要ならこの辺りは削っても構わない。lspkind.nvimを利用している場合はformattingに続く設定の形で補完候補のシンボルに任意の絵文字を設定できる。きれいな絵文字が並んでいるとモチベが上がるのでぜひ設定したい。

local lspkind = require("lspkind")

formatting = {
    format = lspkind.cmp_format({
        mode = "symbol",
        maxwidth = 50,
        ellipsis_char = "...",
        symbol_map = { Copilot = "" },
    }),
},

なお、次候補の選択をTabキーで行っている場合は以下の特殊な設定を追記する必要がある。

local has_words_before = function()
    if vim.api.nvim_buf_get_option(0, "buftype") == "prompt" then
        return false
    end
    local line, col = unpack(vim.api.nvim_win_get_cursor(0))
    return col ~= 0 and vim.api.nvim_buf_get_text(0, line - 1, 0, line - 1, col, {})[1]:match("^%s*$") == nil
end
cmp.setup({
    mapping = {
        ["<Tab>"] = vim.schedule_wrap(function(fallback)
            if cmp.visible() and has_words_before() then
                cmp.select_next_item({ behavior = cmp.SelectBehavior.Select })
            else
                fallback()
            end
        end),
    },
})

以上の設定でGitHub Copilotの提案が補完候補の一部として表示されるようになる。

さて、これでCopilotの設定は終わりかと思いきや、実はそうではない。LLMの最大と言ってもいい機能は対話による改善や、全体を俯瞰した効率的な提案なのでチャットができないことには魅力半減だ。しかし公式のCopilot.vimにも今回紹介した非公式の方にもそれは備わっていない。そこで、CopiotChat.nvimというさらに別のプラグインを導入する。

CopilotChat.nvimは名前通りCopilotとの対話機能を提供するプラグインである。多くの連携プラグインが用意されており、設定内容も多岐に渡るが本稿では現在のバッファの内容を考慮した上で質問する方法とTelescopeと連携して定型文の質問を行う方法の2種類について記す。Telescopeは導入済みとする。

---lazy
{ "CopilotC-Nvim/CopilotChat.nvim", build = "make tiktoken" },

--CopilotChat

local select = require("CopilotChat.select")

require("CopilotChat").setup({
    debug = true,

    window = {
        layout = "float",
        relative = "editor",
    },
    prompts = {
        Explain = {
            prompt = "/COPILOT_EXPLAIN 選択されたコードの説明を段落をつけて書いてください。",
        },
        Review = {
            prompt = "/COPILOT_REVIEW 選択されたコードをレビューしてください。",
            callback = function(response, source) end,
        },
        Fix = {
            prompt = "/COPILOT_FIX このコードには問題があります。バグを修正したコードに書き直してください。",
        },
        Optimize = {
            prompt = "/COPILOT_REFACTOR 選択されたコードを最適化してパフォーマンスと可読性を向上させてください。",
        },
        Docs = {
            prompt = "/COPILOT_DOCS 選択されたコードに対してドキュメンテーションコメントを追加してください。",
        },
        Tests = {
            prompt = "/COPILOT_TESTS 選択されたコードの詳細な単体テスト関数を書いてください。",
        },
        FixDiagnostic = {
            prompt = "ファイル内の次のような診断上の問題を解決してください:",
            selection = select.diagnostics,
        },
    },
})

function CopilotChatBuffer()
    local input = vim.fn.input("Quick Chat: ")
    if input ~= "" then
        require("CopilotChat").ask(input, { selection = require("CopilotChat.select").buffer })
    end
end

vim.api.nvim_set_keymap("n", "<leader>9", "<cmd>lua CopilotChatBuffer()<cr>", { noremap = true, silent = true })

function ShowCopilotChatActionPrompt()
    local actions = require("CopilotChat.actions")
    require("CopilotChat.integrations.telescope").pick(actions.prompt_actions())
end

vim.api.nvim_set_keymap(
    "n",
    "<leader>0",
    "<cmd>lua ShowCopilotChatActionPrompt()<cr>",
    { noremap = true, silent = true }
)

上記のうちpromptsに続く日本語は標準では英語で用意されている定型句の質問文を書き換えたものとなる。Copilot Chatは質問に用いた言語で回答が返ってくるため、あえて日本語を設定しなおした。英語で差し支えないならこの項目は丸ごとカットしても問題ない。また、function ShowCopilotChatActionPromptでTelescopeを呼び出す関数を定義している。

この機能のなにが嬉しいのかというとコーディング中に頻繁に用いるであろう質問文をわずかなキータイプで入力できること、質問文が固定されているので精度の高い回答が期待できることなどが挙げられる。一方、自由記述の入力ウインドウはfunction CopilotChatBufferによって呼びされる。