Initial Implementation.

<Tab> accepts one word from suggestion.

<S-Tab> accepts all.

Author: abdullah8a0

autoload/copilot.vim

@@ -419,7 +419,7 @@ endfunction
 
 function! copilot#IsMapped() abort
   return get(g:, 'copilot_assume_mapped') ||
-        \ hasmapto('copilot#Accept(', 'i')
+        \ hasmapto('copilot#AcceptOne(', 'i')
 endfunction
 let s:is_mapped = copilot#IsMapped()
 
@@ -465,7 +465,57 @@ function! copilot#TextQueuedForInsertion() abort
   endtry
 endfunction
 
-function! copilot#Accept(...) abort
+let s:sugg_buffer = {'sugg': '', 'count': 0, 'text': '', 'valid': v:false}
+function! copilot#AcceptOne(...) abort
+
+  if !s:sugg_buffer.valid
+    let s = copilot#GetDisplayedSuggestion()
+    if !empty(s)
+      " split the suggestion into words, keep the trailing space
+      let s:sugg_buffer = {'sugg': s, 'count': 0, 'text': split(s.text, '[\t ]\zs') , 'valid': v:true}
+      " if the text is whitespace, invalidate the buffer
+      if (len(s:sugg_buffer.text)==0) || !(s:sugg_buffer.text[0] =~# '\S')
+        let s:sugg_buffer.valid = v:false
+      endif
+    endif
+  endif
+
+  if s:sugg_buffer.valid
+    let s:suggestion_text = s:sugg_buffer.text[s:sugg_buffer.count]
+    let s:sugg_buffer.count += 1
+    if s:sugg_buffer.count >= len(s:sugg_buffer.text)
+      let s:sugg_buffer.valid = v:false
+      unlet! b:_copilot
+      call copilot#Request('notifyAccepted', {'uuid': s:sugg_buffer.sugg.uuid}) 
+      unlet! s:uuid
+      call s:ClearPreview()
+    endif
+    if s:sugg_buffer.valid && s:sugg_buffer.count == 1
+      return repeat("\<Left>\<Del>", s:sugg_buffer.sugg.outdentSize) . repeat("\<Del>", s:sugg_buffer.sugg.deleteSize) .
+            \ "\<C-R>\<C-O>=copilot#TextQueuedForInsertion()\<CR>"
+    endif
+    return "\<C-R>\<C-O>=copilot#TextQueuedForInsertion()\<CR>" 
+  endif
+  
+  let default = get(g:, 'copilot_tab_fallback', pumvisible() ? "\<C-N>" : "\t")
+  if !a:0
+    return default
+  elseif type(a:1) == v:t_string
+    return a:1
+  elseif type(a:1) == v:t_func
+    try
+      return call(a:1, [])
+    catch
+      call copilot#logger#Exception()
+      return default
+    endtry
+  else
+    return default
+  endif
+endfunction
+
+
+function! copilot#AcceptAll(...) abort
   let s = copilot#GetDisplayedSuggestion()
   if !empty(s.text)
     unlet! b:_copilot

plugin/copilot.vim

@@ -20,13 +20,15 @@ function! s:ColorScheme() abort
   hi def link CopilotAnnotation Normal
 endfunction
 
+
 function! s:MapTab() abort
   if get(g:, 'copilot_no_tab_map') || get(g:, 'copilot_no_maps')
     return
   endif
   let tab_map = maparg('<Tab>', 'i', 0, 1)
   if !has_key(tab_map, 'rhs')
-    imap <script><silent><nowait><expr> <Tab> copilot#Accept()
+    imap <script><silent><nowait><expr> <Tab> copilot#AcceptOne()
+    imap <script><silent><nowait><expr> <S-Tab> copilot#AcceptAll()
   elseif tab_map.rhs !~# 'copilot'
     if tab_map.expr
       let tab_fallback = '{ -> ' . tab_map.rhs . ' }'
@@ -35,9 +37,11 @@ function! s:MapTab() abort
     endif
     let tab_fallback = substitute(tab_fallback, '<SID>', '<SNR>' . get(tab_map, 'sid') . '_', 'g')
     if get(tab_map, 'noremap') || get(tab_map, 'script') || mapcheck('<Left>', 'i') || mapcheck('<Del>', 'i')
-      exe 'imap <script><silent><nowait><expr> <Tab> copilot#Accept(' . tab_fallback . ')'
+      exe 'imap <script><silent><nowait><expr> <Tab> copilot#AcceptOne(' . tab_fallback . ')'
+      exe 'imap <script><silent><nowait><expr> <S-Tab> copilot#AcceptAll(' . tab_fallback . ')'
     else
-      exe 'imap <silent><nowait><expr>         <Tab> copilot#Accept(' . tab_fallback . ')'
+      exe 'imap <silent><nowait><expr>         <Tab> copilot#AcceptOne(' . tab_fallback . ')'
+      exe 'imap <silent><nowait><expr>         <S-Tab> copilot#AcceptAll(' . tab_fallback . ')'
     endif
   endif
 endfunction