feat: claude provider

Author: boltlessengineer

lua/zeta/provider/claude.lua

@@ -0,0 +1,71 @@
+local log = require("zeta.log")
+local curl = require("plenary.curl")
+
+local M = {}
+
+local CLAUDE_API_URL = "https://api.anthropic.com/v1/messages"
+local CLAUDE_API_TOKEN = vim.env.CLAUDE_API_TOKEN
+local SYSTEM_PROMPT_PATH = vim.api.nvim_get_runtime_file("prompt/claude/system-prompt.md", false)[1]
+local USER_PROMPT_PATH = vim.api.nvim_get_runtime_file("prompt/claude/user-prompt.md", false)[1]
+
+---@param body zeta.PredictEditRequestBody
+---@param callback fun(res: zeta.PredictEditResponse)
+function M.request(body, callback)
+    local system_prompt = (function()
+        local file = assert(io.open(SYSTEM_PROMPT_PATH))
+        local content = file:read("*a")
+        file:close()
+        return content
+    end)()
+    local user_prompt_template = (function()
+        local file = assert(io.open(USER_PROMPT_PATH))
+        local content = file:read("*a")
+        file:close()
+        return content
+    end)()
+    local user_prompt = user_prompt_template
+        :gsub("<events>", body.input_events)
+        :gsub("<excerpt>", body.input_excerpt)
+    local req_body = {
+        model="claude-3-5-haiku-20241022",
+        max_tokens=1000,
+        temperature=0,
+        system = system_prompt,
+        messages = {
+            {
+                role = "user",
+                content = {
+                    { type = "text", text = user_prompt },
+                },
+            },
+            {
+                role = "assistant",
+                content = {
+                    { type = "text", text = "### Predicted Excerpt:\n\nHere is a modified excerpt including users predicted next edits." },
+                },
+            },
+        },
+    }
+    log.debug("claude request body:", req_body)
+    curl.post(CLAUDE_API_URL, {
+        body = vim.json.encode(req_body),
+        headers = {
+            ["Content-Type"] = "application/json",
+            ["x-api-key"] = CLAUDE_API_TOKEN,
+            ["anthropic-version"] = "2023-06-01",
+        },
+        callback = function(resp)
+            if resp.status ~= 200 then
+                return
+            end
+            local _ok, resp_body = pcall(vim.json.decode, resp.body)
+            -- TODO: validate resp_body signature
+            callback({
+                request_id = "",
+                output_excerpt = resp_body.content[1].text,
+            })
+        end
+    })
+end
+
+return M

prompt/claude/system-prompt.md

@@ -0,0 +1 @@
+You are a code completion assistant and your task is to analyze user edits and then rewrite an excerpt that the user provides, suggesting the appropriate edits within the excerpt, taking into account the cursor location. You will be given recent diff created by user and excerpt of codebase around user's cursor position (this is not the full file content). Guess what user will edit later and apply those edits into your response. Note that your job is to do predicted edits inside user excerpt and not completing the incomplete excerpt code snippet. Preserve given `<|editable_region_start|>` and `<|editable_region_end|>` tokens. Do not include the explanation in your response.

prompt/claude/user-prompt.md

@@ -0,0 +1,7 @@
+### User Edits:
+
+<events>
+
+### User Excerpt:
+
+<excerpt>