feat(git): add set_upstream option to push command

- introduce set_upstream parameter for git push to enable automatic remote tracking
- default set_upstream to true for push operations when not specified
- update type definitions and documentation for push function
- improve code comments for operation execution safety

Author: jinzhongjia

lua/codecompanion/_extensions/gitcommit/init.lua

@@ -482,6 +482,18 @@ return {
         return GitTool.merge(branch)
       end,
 
+      ---Push changes to remote repository
+      ---@param remote? string The remote to push to (e.g., origin)
+      ---@param branch? string The branch to push (defaults to current branch)
+      ---@param force? boolean Force push (DANGEROUS)
+      ---@param set_upstream? boolean Set upstream branch (default: true for auto-linking)
+      ---@param tags? boolean Push all tags
+      ---@param tag_name? string Single tag to push
+      push = function(remote, branch, force, set_upstream, tags, tag_name)
+        local GitTool = require("codecompanion._extensions.gitcommit.tools.git").GitTool
+        return GitTool.push(remote, branch, force, set_upstream, tags, tag_name)
+      end,
+
       ---Generate release notes between tags
       ---@param from_tag? string Starting tag (if not provided, uses second latest tag)
       ---@param to_tag? string Ending tag (if not provided, uses latest tag)

lua/codecompanion/_extensions/gitcommit/tools/git.lua

@@ -489,14 +489,18 @@ end
 ---@param remote? string The name of the remote to push to (e.g., origin)
 ---@param branch? string The name of the branch to push (defaults to current branch)
 ---@param force? boolean Force push (DANGEROUS: overwrites remote history)
+---@param set_upstream? boolean Set the upstream branch for the current local branch
 ---@param tags? boolean Push all tags
----@param tag_name? string The name of a single tag to push
+---@param tag_name? string The name of a single tag to push (takes priority over tags parameter)
 ---@return boolean success, string output
-function GitTool.push(remote, branch, force, tags, tag_name)
+function GitTool.push(remote, branch, force, set_upstream, tags, tag_name)
   local cmd = "git push"
   if force then
     cmd = cmd .. " --force"
   end
+  if set_upstream then
+    cmd = cmd .. " --set-upstream"
+  end
 
   -- Handle tag pushing - single tag takes priority over all tags
   if tag_name and vim.trim(tag_name) ~= "" then

lua/codecompanion/_extensions/gitcommit/tools/git_edit.lua

@@ -207,6 +207,10 @@ Available write-access Git operations:
     end
 
     if operation == "push" then
+      -- If set_upstream is not explicitly specified, default to true for automatic remote tracking
+      if op_args.set_upstream == nil then
+        op_args.set_upstream = true
+      end
       return GitTool.push_async(
         op_args.remote,
         op_args.branch,
@@ -218,7 +222,7 @@ Available write-access Git operations:
       )
     end
 
-    -- 通过 pcall 安全执行操作，确保始终有响应
+    -- Safely execute operations through pcall to ensure there's always a response
     local ok, result = pcall(function()
       local success, output
 

lua/codecompanion/_extensions/gitcommit/tools/git_read.lua

@@ -155,7 +155,7 @@ GitRead.cmds = {
 
     local success, output, user_msg, llm_msg
 
-    -- 通过 pcall 安全执行操作，确保始终有响应
+    -- Safely execute operations through pcall to ensure there's always a response
     local ok, result = pcall(function()
       if operation == "status" then
         success, output, user_msg, llm_msg = GitTool.get_status()

lua/codecompanion/_extensions/gitcommit/types.lua

@@ -1,6 +1,5 @@
----@field use_commit_history? boolean Enable using commit history as context (default: true)
+---@field push fun(remote?: string, branch?: string, force?: boolean, set_upstream?: boolean, tags?: boolean, tag_name?: string): boolean, string
 
----@class CodeCompanion.GitCommit.ExtensionOpts.Buffer
 ---@field enabled boolean Enable buffer-specific keymap for git commit
 ---@field keymap string Keymap for generating commit message in git commit buffer
 ---@field auto_generate? boolean Automatically generate commit message on entering gitcommit buffer