Compare commits
3 Commits
7476e27e01
...
4e9a32f501
Author | SHA1 | Date | |
---|---|---|---|
4e9a32f501 | |||
5fb26e6220 | |||
abd32275ba |
52
aicommit.clj
Executable file
52
aicommit.clj
Executable file
@ -0,0 +1,52 @@
|
|||||||
|
#!/usr/bin/env bb
|
||||||
|
|
||||||
|
(require '[babashka.curl :as curl]
|
||||||
|
'[babashka.process :as p]
|
||||||
|
'[cheshire.core :as json]
|
||||||
|
'[babashka.fs :as fs]
|
||||||
|
'[babashka.cli :as cli])
|
||||||
|
|
||||||
|
(def url "https://api.groq.com/openai/v1/chat/completions")
|
||||||
|
|
||||||
|
(def system-msg "
|
||||||
|
You are an expert programmer that values clear, unambiguous communication and are specialized in generating concise and informative git commit messages.
|
||||||
|
Your task is to generate a concise, informative git commit message based on the following git diff.
|
||||||
|
Be sure that the commit message reflects the entire diff.
|
||||||
|
It is very important that the entire commit is clear and understandable.
|
||||||
|
Only reply with the commit message and nothing else.
|
||||||
|
")
|
||||||
|
|
||||||
|
(def diff (:out (p/check (p/sh "git" "diff" "--cached"))))
|
||||||
|
|
||||||
|
(defn find-git
|
||||||
|
[dir]
|
||||||
|
(loop [d dir]
|
||||||
|
(if (fs/directory? (fs/path d ".git"))
|
||||||
|
(fs/path d ".git")
|
||||||
|
(recur (fs/parent d)))))
|
||||||
|
|
||||||
|
(def commit-message-path (fs/unixify (fs/path (find-git (fs/cwd)) "COMMIT_EDITMSG")))
|
||||||
|
|
||||||
|
(def commit-message (slurp commit-message-path))
|
||||||
|
|
||||||
|
(def message (json/generate-string
|
||||||
|
{:messages [{:role "system" :content system-msg}
|
||||||
|
{:role "user" :content diff}]
|
||||||
|
:model "llama-3.1-70b-versatile"
|
||||||
|
:temperature 1
|
||||||
|
:max_tokens 1024
|
||||||
|
:top_p 1
|
||||||
|
:stream false
|
||||||
|
:stop nil}))
|
||||||
|
|
||||||
|
(def response
|
||||||
|
(curl/post url
|
||||||
|
{:headers {"Content-Type" "application/json"
|
||||||
|
"Authorization" (str "Bearer " (System/getenv "GROQ_API_KEY"))}
|
||||||
|
:body message}))
|
||||||
|
|
||||||
|
(def content (get-in (:choices (json/parse-string (:body response) true)) [0 :message :content]))
|
||||||
|
|
||||||
|
(spit commit-message-path (str content "\n" commit-message))
|
||||||
|
|
||||||
|
(p/shell (System/getenv "EDITOR") (first *command-line-args*))
|
Loading…
Reference in New Issue
Block a user