gemini-code-completion.el/gemini-code-completion.el at main · shishimaru/gemini-code-completion.el · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
;;; gemini-code-completion.el --- Code completion empowered by Google Gemini -*- lexical-binding: t; -*-

;; Copyright (C) 2024 Hitoshi Uchida <hitoshi.uchida@gmail.com>

;; Author: Hitoshi Uchida <hitoshi.uchida@gmail.com>
;; Version: 1.0
;; Package-Requires: ((google-gemini "0.1.0") (emacs "25.1"))
;; URL: https://github.com/shishimaru/gemini-code-completion.el

;; This program is free software: you can redistribute it and/or modify
;; it under the terms of the GNU General Public License as published by
;; the Free Software Foundation, either version 3 of the License, or
;; (at your option) any later version.

;; This program is distributed in the hope that it will be useful,
;; but WITHOUT ANY WARRANTY; without even the implied warranty of
;; MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
;; GNU General Public License for more details.

;; You should have received a copy of the GNU General Public License
;; along with this program.  If not, see <http://www.gnu.org/licenses/>.

;;; Commentary:
;;
;; gemini-code-completion.el is an Emacs package to get code completion with
;; Google Gemini for a code at current cursor postion or a selected region.

;;; Code:

(require 'google-gemini)

(defvar gemini-code-completion-default-model
  "gemini-pro-latest"
  "The default Gemini model. You can obtain one by running `google-gemini-list-models'.")

(defvar gemini-code-completion-default-prompt
  "Suggest only missing part to work as feature, at most just only 1 or 2 lines.
Must not exceed 2 lines in suggesting.
Take care indentation as well.


" "Default prompt to input into Google Gemini.")

(defun gemini-code-completion-extract-completion (response)
  "Extract and clean completion text from the RESPONSE returned by Google Gemini."
  (let* ((candidates (alist-get 'candidates response))
         (first-candidate (aref candidates 0))
         (content (alist-get 'content first-candidate))
         (parts (alist-get 'parts content))
         (first-part (aref parts 0))
         (text (alist-get 'text first-part)))
    (gemini-code-completion-clean-text text)))

(defun gemini-code-completion-clean-text (text)
  "Remove unwanted backquotes from the TEXT."
  (replace-regexp-in-string "```[a-z]*\n\\|```" "" text))

(defun gemini-code-completion-handler (response)
  "Handle Gemini code completion RESPONSE."

  (let ((current-position (point))
        (completion-text (gemini-code-completion-extract-completion response)))
    (insert
     (if (use-region-p)
         "\n"
       "")
     (propertize completion-text 'face 'shadow))
    (when (use-region-p)
      (deactivate-mark))
    (goto-char current-position))) ; Restore cursor position

;;;###autoload
(defun gemini-code-completion (prefix)
  "Get completion from Google Gemini for current buffer or a selected region.
If called with a PREFIX argument (\\[universal-argument]), prompt for additional
text to customize the completion."
  (interactive "P")

  (let* ((user-prompt (if prefix (concat (read-string "Prompt: ") "\n\n") ""))
         (selected-text
          (if (use-region-p)
              (buffer-substring-no-properties (region-beginning) (region-end))
            (buffer-substring-no-properties (point-min) (point))))
         (end
          (if (use-region-p)
              (region-end)
            (point))))
    (save-excursion
      (goto-char end)
      (google-gemini-content-generate
       (concat
        user-prompt gemini-code-completion-default-prompt selected-text)
       #'gemini-code-completion-handler
       :model gemini-code-completion-default-model))))

(provide 'gemini-code-completion)
;;; gemini-code-completion.el ends here