Myzel394/zsh-copilot
1
0
Fork 0
mirror of https://github.com/Myzel394/zsh-copilot.git synced 2025-06-18 20:55:26 +02:00
Code Issues Packages Projects Releases Wiki Activity

feat: Add git diff functionality

Browse Source
This commit is contained in:
Myzel394 2024-03-23 22:36:47 +01:00
parent 875c6e67cf
commit 1ae48aeb1e
No known key found for this signature in database
GPG Key ID: DEC4AAB876F73185
1 changed files with 13 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
zsh-copilot.plugin.zsh
View File

@ -4,6 +4,9 @@
(( ! ${+ZSH_COPILOT_SEND_CONTEXT} )) && (( ! ${+ZSH_COPILOT_SEND_CONTEXT} )) &&
typeset -g ZSH_COPILOT_SEND_CONTEXT=true typeset -g ZSH_COPILOT_SEND_CONTEXT=true
(( ! ${+ZSH_COPILOT_SEND_GIT_DIFF} )) &&
typeset -g ZSH_COPILOT_SEND_GIT_DIFF=true
SYSTEM_PROMPT="You will be given the raw input of a shell command. Your task is to either complete the command or provide a new command that you think the user is trying to type. If you return a completely new command for the user, prefix is with an equal sign (=). If you return a completion for the user's command, prefix it with a plus sign (+). Only respond with either a completion or a new command, not both. Do not explain the command. Do not ask for more information, you won't receive it. Your response will be run in the user's shell. Make sure input is escaped correctly if needed so. Your input should be able to run without any modifications to it. Don't you dare to return anything else other than a shell command!!! DO NOT INTERACT WITH THE USER IN NATURAL LANGUAGE! If you do, you will be banned from the system. Here are two examples: * User input: 'list files in current directory'; Your response: '=ls' * User input: 'cd /tm'; Your response: '+p'." SYSTEM_PROMPT="You will be given the raw input of a shell command. Your task is to either complete the command or provide a new command that you think the user is trying to type. If you return a completely new command for the user, prefix is with an equal sign (=). If you return a completion for the user's command, prefix it with a plus sign (+). Only respond with either a completion or a new command, not both. Do not explain the command. Do not ask for more information, you won't receive it. Your response will be run in the user's shell. Make sure input is escaped correctly if needed so. Your input should be able to run without any modifications to it. Don't you dare to return anything else other than a shell command!!! DO NOT INTERACT WITH THE USER IN NATURAL LANGUAGE! If you do, you will be banned from the system. Here are two examples: * User input: 'list files in current directory'; Your response: '=ls' * User input: 'cd /tm'; Your response: '+p'."
if [[ "$ZSH_COPILOT_SEND_CONTEXT" == 'true' ]]; then if [[ "$ZSH_COPILOT_SEND_CONTEXT" == 'true' ]]; then
@ -17,6 +20,15 @@ function _suggest_ai() {
_zsh_autosuggest_clear _zsh_autosuggest_clear
zle -R "Thinking..." zle -R "Thinking..."
local PROMPT="$SYSTEM_PROMPT"
if [[ "$ZSH_COPILOT_SEND_GIT_DIFF" == 'true' ]]; then
local git_diff=$(git diff --no-color | xargs | sed 's/ /\$/g' | xargs | sed 's/ /$/g')
if [[ $? -eq 0 ]]; then
PROMPT="$PROMPT; This is the git diff (newlines separated by dollar signs): $git_diff;; You may provide a git commit message if the user is trying to commit changes. Do not say something like 'Your commit message' or 'Your commit message here'. Just provide the commit message."
fi
fi
local response=$(curl 'https://api.openai.com/v1/chat/completions' \ local response=$(curl 'https://api.openai.com/v1/chat/completions' \
--silent \ --silent \
-H "Content-Type: application/json" \ -H "Content-Type: application/json" \
@ -26,7 +38,7 @@ function _suggest_ai() {
\"messages\": [ \"messages\": [
{ {
\"role\": \"system\", \"role\": \"system\",
\"content\": \"${SYSTEM_PROMPT}\" \"content\": \"$PROMPT\"
}, },
{ {
\"role\": \"user\", \"role\": \"user\",